Change parameter for torch2.6

Default parameter for torch.load has changed.
Gurobi · Feb 7, 2025 · 37c91b0 · 37c91b0
1 parent 61cbd2d
commit 37c91b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_pytorch/test_pytorch_formulations.py b/tests/test_pytorch/test_pytorch_formulations.py
@@ -16,7 +16,7 @@ def test_diabetes_pytorch(self):
         X = load(os.path.join(self.basedir, "examples_diabetes.joblib"))
 
         filename = os.path.join(self.basedir, "diabetes__pytorch.pt")
-        regressor = torch.load(filename)
+        regressor = torch.load(filename, weights_only=False)
         onecase = {"predictor": regressor, "nonconvex": 0}
         self.do_one_case(onecase, X, 5, "all")
         self.do_one_case(onecase, X, 6, "pairs")