From 6ad853c768b0841823d2c8122ead0f271b32ccd6 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 13 Apr 2024 10:51:11 -0700 Subject: [PATCH] fix - test add models + call /health --- tests/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index 59b0fb02ef2..6f9a6cc6966 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -280,7 +280,7 @@ async def test_add_model_run_health(): print("calling /health?model=", model_name) _health_info = await get_model_health( - session=session, key=key, model_name="azure/chatgpt-v-2" + session=session, key=key, model_name=model_name ) _healthy_endpooint = _health_info["healthy_endpoints"][0]