diff --git a/libs/core/tests/unit_tests/language_models/chat_models/test_rate_limiting.py b/libs/core/tests/unit_tests/language_models/chat_models/test_rate_limiting.py index b15b202f484..2d251aadb92 100644 --- a/libs/core/tests/unit_tests/language_models/chat_models/test_rate_limiting.py +++ b/libs/core/tests/unit_tests/language_models/chat_models/test_rate_limiting.py @@ -84,7 +84,7 @@ def test_rate_limit_batch() -> None: toc = time.time() # Should be larger than check every n seconds since the token bucket starts # with 0 tokens. - assert time_to_fill < toc - tic < time_to_fill + 0.01 + assert time_to_fill < toc - tic < time_to_fill + 0.03 async def test_rate_limit_abatch() -> None: @@ -102,7 +102,7 @@ async def test_rate_limit_abatch() -> None: toc = time.time() # Should be larger than check every n seconds since the token bucket starts # with 0 tokens. - assert time_to_fill < toc - tic < time_to_fill + 0.01 + assert time_to_fill < toc - tic < time_to_fill + 0.03 def test_rate_limit_stream() -> None: