diff --git a/litellm/tests/test_router_get_deployments.py b/litellm/tests/test_router_get_deployments.py index 712867a512..3e6bbd2b6e 100644 --- a/litellm/tests/test_router_get_deployments.py +++ b/litellm/tests/test_router_get_deployments.py @@ -322,7 +322,6 @@ def test_model_group_aliases(): router = Router( model_list=model_list, ) - litellm.model_group_alias_map = {} for _ in range(20): selected_model = router.get_available_deployment("gpt-4") print("\n selected model", selected_model) @@ -331,6 +330,7 @@ def test_model_group_aliases(): pytest.fail(f"Selected model {selected_model_name} is not gpt-3.5-turbo") router.reset() + litellm.model_group_alias_map = {} except Exception as e: traceback.print_exc() pytest.fail(f"Error occurred: {e}")