fix test

modelscope · Oct 8, 2024 · 05d85af · 05d85af
1 parent 987c19a
commit 05d85af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/rpc_agent_test.py b/tests/rpc_agent_test.py
@@ -917,7 +917,7 @@ def test_retry_strategy(self) -> None:
         st = time.time()
         result = fix_retry.retry(mock_func)
         et = time.time()
-        self.assertTrue(et - st > delay * 0.5 + 0.1)
+        self.assertTrue(et - st > delay * 0.5)
         self.assertTrue(et - st < delay * 1.5 + 0.1)
         self.assertEqual(result, "Success")
         mock_func = MagicMock(
@@ -926,6 +926,6 @@ def test_retry_strategy(self) -> None:
         st = time.time()
         result = exp_retry.retry(mock_func)
         et = time.time()
-        self.assertTrue(et - st > delay * 0.5 + 0.1)
+        self.assertTrue(et - st > delay * 0.5)
         self.assertTrue(et - st < delay * 1.5 + 0.1)
         self.assertEqual(result, "Success")