phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_least_busy_routing.py): fix test init

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-30 12:39:02 +05:30
parent 535a547b66
commit 2acd086596
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_least_busy_routing.py
View file

@ -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 = {