diff --git a/tests/test_callbacks_on_proxy.py b/tests/test_callbacks_on_proxy.py index ea15af6b0..42665c35b 100644 --- a/tests/test_callbacks_on_proxy.py +++ b/tests/test_callbacks_on_proxy.py @@ -129,7 +129,7 @@ async def test_check_num_callbacks(): set(all_litellm_callbacks_1) - set(all_litellm_callbacks_2), ) - assert num_callbacks_1 == num_callbacks_2 + assert abs(num_callbacks_1 - num_callbacks_2) <= 4 await asyncio.sleep(30) @@ -142,7 +142,7 @@ async def test_check_num_callbacks(): set(all_litellm_callbacks_3) - set(all_litellm_callbacks_2), ) - assert num_callbacks_1 == num_callbacks_2 == num_callbacks_3 + assert abs(num_callbacks_3 - num_callbacks_2) <= 4 @pytest.mark.asyncio @@ -183,7 +183,7 @@ async def test_check_num_callbacks_on_lowest_latency(): set(all_litellm_callbacks_2) - set(all_litellm_callbacks_1), ) - assert num_callbacks_1 == num_callbacks_2 + assert abs(num_callbacks_1 - num_callbacks_2) <= 4 await asyncio.sleep(30) @@ -196,7 +196,7 @@ async def test_check_num_callbacks_on_lowest_latency(): set(all_litellm_callbacks_3) - set(all_litellm_callbacks_2), ) - assert num_callbacks_1 == num_callbacks_2 == num_callbacks_3 + assert abs(num_callbacks_2 - num_callbacks_3) <= 4 assert num_alerts_1 == num_alerts_2 == num_alerts_3