format

Change-Id: Ifb93c70f8ad48dce0fd0921de0e9e117819dc55c
google-gemini · Aug 23, 2024 · e7b802c · e7b802c
1 parent 9fa575d
commit e7b802c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/google/generativeai/types/content_types.py b/google/generativeai/types/content_types.py
@@ -100,7 +100,7 @@ def pil_to_blob(img):
         mime_type = "image/gif"
     else:
         if img.mode != "RGB":
-            img = img.convert('RGB')
+            img = img.convert("RGB")
         img.save(bytesio, format="JPEG")
         mime_type = "image/jpeg"
     bytesio.seek(0)

diff --git a/tests/test_content.py b/tests/test_content.py
@@ -39,6 +39,7 @@
 TEST_GIF_URL = "https://storage.googleapis.com/generativeai-downloads/data/test_img.gif"
 TEST_GIF_DATA = TEST_GIF_PATH.read_bytes()
 
+
 # simple test function
 def datetime():
     "Returns the current UTC date and time."
@@ -93,7 +94,7 @@ def test_jpg_to_blob(self, image):
 
     @parameterized.named_parameters(
         ["PIL", PIL.Image.open(TEST_GIF_PATH)],
-        ["P", PIL.Image.fromarray(np.zeros([6, 6, 3], dtype=np.uint8)).convert('P')],
+        ["P", PIL.Image.fromarray(np.zeros([6, 6, 3], dtype=np.uint8)).convert("P")],
         ["IPython", IPython.display.Image(filename=TEST_GIF_PATH)],
     )
     def test_gif_to_blob(self, image):