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]