diff --git a/tests/test_litellm/llms/gemini/test_gemini_image_generation.py b/tests/test_litellm/llms/gemini/test_gemini_image_generation.py index f5658da5df2..62467924701 100644 --- a/tests/test_litellm/llms/gemini/test_gemini_image_generation.py +++ b/tests/test_litellm/llms/gemini/test_gemini_image_generation.py @@ -1,5 +1,4 @@ import pytest -import litellm from litellm.llms.gemini.image_generation.transformation import GoogleImageGenConfig def test_gemini_flash_image_generation_extra_body():