fix pytype

Change-Id: I431c66e45e7582218b5de7a90eeeee01b80df664
google-gemini · Oct 24, 2024 · 3dc69a0 · 3dc69a0
1 parent 48932f9
commit 3dc69a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/google/generativeai/types/image_types/_image_types.py b/google/generativeai/types/image_types/_image_types.py
@@ -169,7 +169,7 @@ class Image:
     __module__ = "vertexai.vision_models"
 
     _loaded_bytes: Optional[bytes] = None
-    _loaded_image: Optional["PIL_Image.Image"] = None
+    _loaded_image: Optional["PIL.Image.Image"] = None
 
     def __init__(
         self,
@@ -207,7 +207,7 @@ def _image_bytes(self, value: bytes):
         self._loaded_bytes = value
 
     @property
-    def _pil_image(self) -> "PIL_Image.Image":  # type: ignore
+    def _pil_image(self) -> "PIL.Image.Image":  # type: ignore
         if self._loaded_image is None:
             if not PIL:
                 raise RuntimeError(