Merge 2d72246 into 4871a9a

pytorch · Jul 26, 2023 · 23cdfd7 · 23cdfd7
2 parents 4871a9a + 2d72246
commit 23cdfd7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/optim/test_core.py b/test/optim/test_core.py
@@ -81,7 +81,7 @@ def test_basic(self):
     def test_timeout(self):
         x = Parameter(torch.tensor(1.0))
         # adding a small delay here to combat some timing issues on windows
-        closure = partial(norm_squared, x, delay=1e-3)
+        closure = partial(norm_squared, x, delay=1e-2)
         result = scipy_minimize(closure, {"x": x}, timeout_sec=1e-4)
         self.assertEqual(result.status, OptimizationStatus.STOPPED)
         self.assertTrue("Optimization timed out after" in result.message)

diff --git a/test/optim/utils/test_timeout.py b/test/optim/utils/test_timeout.py
@@ -41,7 +41,7 @@ def f_and_g(x: np.ndarray, sleep_sec: float = 0.0):
             self.assertEqual(res.nit, 2)  # quadratic approx. is exact
 
         with self.subTest("test w/ binding timeout"):
-            res = minimize_with_timeout(**base_kwargs, args=(1e-3,), timeout_sec=1e-4)
+            res = minimize_with_timeout(**base_kwargs, args=(1e-2,), timeout_sec=1e-4)
             self.assertFalse(res.success)
             self.assertEqual(res.nit, 1)  # only one call to the callback is made