diff --git a/litellm/images/main.py b/litellm/images/main.py
index 6bdccbfe705..03c0e36ad93 100644
--- a/litellm/images/main.py
+++ b/litellm/images/main.py
@@ -671,7 +671,7 @@ def image_variation(
 
 
 @client
-def image_edit(
+def image_edit(  # noqa: PLR0915
     image: Union[FileTypes, List[FileTypes]],
     prompt: str,
     model: Optional[str] = None,
diff --git a/litellm/llms/stability/image_edit/transformations.py b/litellm/llms/stability/image_edit/transformations.py
index 6c9389d3ac4..5daccefbf72 100644
--- a/litellm/llms/stability/image_edit/transformations.py
+++ b/litellm/llms/stability/image_edit/transformations.py
@@ -43,7 +43,7 @@ class StabilityImageEditConfig(BaseImageEditConfig):
 
     def get_supported_openai_params(
         self, model: str
-    ) -> list:
+    ) -> List[str]:
         """
         Return list of OpenAI params supported by Stability AI.