From 9450d2cf778a232a252b5ece5d432d6e3efe9753 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 22 Jun 2024 22:43:56 -0700 Subject: [PATCH] fix(test_dynamic_rate_limit_handler.py): cleanup --- litellm/tests/test_dynamic_rate_limit_handler.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/litellm/tests/test_dynamic_rate_limit_handler.py b/litellm/tests/test_dynamic_rate_limit_handler.py index c3fcca6a6be..6e1b55d186d 100644 --- a/litellm/tests/test_dynamic_rate_limit_handler.py +++ b/litellm/tests/test_dynamic_rate_limit_handler.py @@ -214,23 +214,23 @@ async def test_base_case(dynamic_rate_limit_handler, mock_response): prev_availability: Optional[int] = None allowed_fails = 1 - for _ in range(5): + for _ in range(2): try: # check availability availability, _, _ = await dynamic_rate_limit_handler.check_available_tpm( model=model ) - ## assert availability updated - if prev_availability is not None and availability is not None: - assert availability == prev_availability - 10 - print( "prev_availability={}, availability={}".format( prev_availability, availability ) ) + ## assert availability updated + if prev_availability is not None and availability is not None: + assert availability == prev_availability - 10 + prev_availability = availability # make call