From 2acd0865965ac939a5fda64ecbf11522128fe55a Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 30 Dec 2023 12:39:02 +0530 Subject: [PATCH] test(test_least_busy_routing.py): fix test init --- litellm/tests/test_least_busy_routing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_least_busy_routing.py b/litellm/tests/test_least_busy_routing.py index c404683a9..0d0f6973a 100644 --- a/litellm/tests/test_least_busy_routing.py +++ b/litellm/tests/test_least_busy_routing.py @@ -22,7 +22,7 @@ from litellm.caching import DualCache def test_model_added(): test_cache = DualCache() - least_busy_logger = LeastBusyLoggingHandler(router_cache=test_cache) + least_busy_logger = LeastBusyLoggingHandler(router_cache=test_cache, model_list=[]) kwargs = { "litellm_params": { "metadata": { @@ -39,7 +39,7 @@ def test_model_added(): def test_get_available_deployments(): test_cache = DualCache() - least_busy_logger = LeastBusyLoggingHandler(router_cache=test_cache) + least_busy_logger = LeastBusyLoggingHandler(router_cache=test_cache, model_list=[]) model_group = "gpt-3.5-turbo" deployment = "azure/chatgpt-v-2" kwargs = {