diff --git a/tests/local_testing/test_router.py b/tests/local_testing/test_router.py index 165f0a5f2a..e6032f2172 100644 --- a/tests/local_testing/test_router.py +++ b/tests/local_testing/test_router.py @@ -1154,16 +1154,7 @@ def test_img_gen_on_router(): "litellm_params": { "model": "dall-e-3", }, - }, - { - "model_name": "dall-e-3", - "litellm_params": { - "model": "azure/dall-e-3-test", - "api_version": "2023-12-01-preview", - "api_base": os.getenv("AZURE_SWEDEN_API_BASE"), - "api_key": os.getenv("AZURE_SWEDEN_API_KEY"), - }, - }, + } ] router = Router(model_list=model_list) response = router.image_generation(