diff --git a/tests/image_gen_tests/test_image_generation.py b/tests/image_gen_tests/test_image_generation.py index 7a803daf5d..61843d3151 100644 --- a/tests/image_gen_tests/test_image_generation.py +++ b/tests/image_gen_tests/test_image_generation.py @@ -198,17 +198,6 @@ class TestAzureOpenAIDalle3(BaseImageGenTest): } -class TestAzureFoundryFlux(BaseImageGenTest): - def get_base_image_generation_call_args(self) -> dict: - litellm.set_verbose = True - return { - "model": "azure_ai/FLUX.1-Kontext-pro", - "api_base": os.getenv("AZURE_FLUX_API_BASE"), - "api_key": os.getenv("AZURE_GPT5_API_KEY"), - "n": 1, - "quality": "standard", - } - @pytest.mark.flaky(retries=3, delay=1) def test_image_generation_azure_dall_e_3():