tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-24 10:27:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(test_least_busy_routing.py): fix test init

Browse Source
This commit is contained in:
Krrish Dholakia
2023-12-30 12:39:13 +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
litellm/tests/test_least_busy_routing.py
+2 -2
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 = {