Skip to content

Commit 566e8a8

Browse files
committed
Revert "fix lazy loading with typing"
This reverts commit 0baabfe.
1 parent 0adcc1a commit 566e8a8

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

google/generativeai/vision_models/_vision_models.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,8 @@
4242

4343
try:
4444
from PIL import Image as PIL_Image
45-
from PIL.Image import Image as PILImageClass
4645
except ImportError:
4746
PIL_Image = None
48-
PILImageClass = None
4947

5048

5149
# This is to get around https://github.com/googleapis/proto-plus-python/issues/488
@@ -113,7 +111,7 @@ class Image:
113111
__module__ = "vertexai.vision_models"
114112

115113
_loaded_bytes: Optional[bytes] = None
116-
_loaded_image: Optional["PILImageClass"] = None
114+
_loaded_image: Optional["PIL_Image.Image"] = None
117115

118116
def __init__(
119117
self,
@@ -151,7 +149,7 @@ def _image_bytes(self, value: bytes):
151149
self._loaded_bytes = value
152150

153151
@property
154-
def _pil_image(self) -> "PILImageClass":
152+
def _pil_image(self) -> "PIL_Image.Image": # type: ignore
155153
if self._loaded_image is None:
156154
if not PIL_Image:
157155
raise RuntimeError(

0 commit comments

Comments
 (0)