diff --git a/litellm/llms/dashscope/image_generation/__init__.py b/litellm/llms/dashscope/image_generation/__init__.py
index 9fdb46586e..aa5724b4d8 100644
--- a/litellm/llms/dashscope/image_generation/__init__.py
+++ b/litellm/llms/dashscope/image_generation/__init__.py
@@ -1,4 +1,6 @@
-from litellm.llms.base_llm.image_generation.transformation import BaseImageGenerationConfig
+from litellm.llms.base_llm.image_generation.transformation import (
+    BaseImageGenerationConfig,
+)
 
 from .transformation import DashScopeImageGenerationConfig
 
diff --git a/litellm/llms/dashscope/image_generation/transformation.py b/litellm/llms/dashscope/image_generation/transformation.py
index feac811df8..152c4791bf 100644
--- a/litellm/llms/dashscope/image_generation/transformation.py
+++ b/litellm/llms/dashscope/image_generation/transformation.py
@@ -27,9 +27,14 @@ from typing import TYPE_CHECKING, Any, List, Optional
 
 import httpx
 
-from litellm.llms.base_llm.image_generation.transformation import BaseImageGenerationConfig
+from litellm.llms.base_llm.image_generation.transformation import (
+    BaseImageGenerationConfig,
+)
 from litellm.secret_managers.main import get_secret_str
-from litellm.types.llms.openai import AllMessageValues, OpenAIImageGenerationOptionalParams
+from litellm.types.llms.openai import (
+    AllMessageValues,
+    OpenAIImageGenerationOptionalParams,
+)
 from litellm.types.utils import ImageObject, ImageResponse
 
 if TYPE_CHECKING:
@@ -93,9 +98,7 @@ class DashScopeImageGenerationConfig(BaseImageGenerationConfig):
         stream: Optional[bool] = None,
     ) -> str:
         return (
-            api_base
-            or get_secret_str("DASHSCOPE_API_BASE_IMAGE")
-            or DEFAULT_API_BASE
+            api_base or get_secret_str("DASHSCOPE_API_BASE_IMAGE") or DEFAULT_API_BASE
         )
 
     def validate_environment(
@@ -176,9 +179,7 @@ class DashScopeImageGenerationConfig(BaseImageGenerationConfig):
 
         choices = response_data.get("output", {}).get("choices", [])
         for choice in choices:
-            content_list = (
-                choice.get("message", {}).get("content", [])
-            )
+            content_list = choice.get("message", {}).get("content", [])
             for content_item in content_list:
                 image_url = content_item.get("image")
                 if image_url: