From ab0fc0a30de724587b329fa0683c314342631ac4 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 25 Oct 2025 15:11:21 -0700 Subject: [PATCH] test_aimg_gen_on_router --- tests/local_testing/test_router.py | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) diff --git a/tests/local_testing/test_router.py b/tests/local_testing/test_router.py index fc60207bb28..165f0a5f2af 100644 --- a/tests/local_testing/test_router.py +++ b/tests/local_testing/test_router.py @@ -1119,16 +1119,7 @@ async def test_aimg_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, num_retries=3) response = await router.aimage_generation(