test_image_generation_azure_dall_e_3

tests/image_gen_tests/test_image_generation.py

@@ -187,46 +187,6 @@ class TestAzureOpenAIDalle3(BaseImageGenTest):
 
 
 
-@pytest.mark.flaky(retries=3, delay=1)
-def test_image_generation_azure_dall_e_3():
-    try:
-        litellm.set_verbose = True
-        response = litellm.image_generation(
-            prompt="A cute baby sea otter",
-            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"),
-            metadata={
-                "model_info": {
-                    "base_model": "azure/dall-e-3",
-                }
-            },
-        )
-        print(f"response: {response}")
-
-        print("response", response._hidden_params)
-        assert len(response.data) > 0
-    except litellm.InternalServerError as e:
-        pass
-    except litellm.ContentPolicyViolationError:
-        pass  # OpenAI randomly raises these errors - skip when they occur
-    except litellm.InternalServerError:
-        pass
-    except litellm.RateLimitError as e:
-        pass
-    except Exception as e:
-        if "Your task failed as a result of our safety system." in str(e):
-            pass
-        if "Connection error" in str(e):
-            pass
-        else:
-            pytest.fail(f"An exception occurred - {str(e)}")
-
-
-# asyncio.run(test_async_image_generation_openai())
-
-
 @pytest.mark.skip(reason="model EOL")
 @pytest.mark.asyncio
 async def test_aimage_generation_bedrock_with_optional_params():