scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-12 03:48:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 57

fix test_single_deployment_no_cooldowns_test_prod_mock_completion_calls

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-14 11:49:18 -07:00
parent 687749a41b
commit 8687ae800d
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_router_cooldowns.py
View file

@ -21,6 +21,7 @@ import openai
import litellm
from litellm import Router
from litellm.integrations.custom_logger import CustomLogger
from litellm.router_utils.cooldown_handlers import _async_get_cooldown_deployments
from litellm.types.router import DeploymentTypedDict, LiteLLMParamsTypedDict
@ -239,7 +240,9 @@ async def test_single_deployment_no_cooldowns_test_prod_mock_completion_calls():
except litellm.RateLimitError:
pass
cooldown_list = await router._async_get_cooldown_deployments()
cooldown_list = await _async_get_cooldown_deployments(
litellm_router_instance=router
)
assert len(cooldown_list) == 0
healthy_deployments, _ = await router._async_get_healthy_deployments(