(test) custom logger proxy

This commit is contained in:
ishaan-jaff 2023-12-11 14:37:23 -08:00
parent 5cb3a51361
commit e5df3733cb
3 changed files with 18 additions and 6 deletions

View file

@ -326,7 +326,7 @@ def test_model_group_aliases():
selected_model = router.get_available_deployment("gpt-4")
print("\n selected model", selected_model)
selected_model_name = selected_model.get("model_name")
if selected_model_name is not "gpt-3.5-turbo":
if selected_model_name != "gpt-3.5-turbo":
pytest.fail(f"Selected model {selected_model_name} is not gpt-3.5-turbo")
router.reset()