File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
google/generativeai/vision_models Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change 42
42
43
43
try :
44
44
from PIL import Image as PIL_Image
45
- from PIL .Image import Image as PILImageClass
46
45
except ImportError :
47
46
PIL_Image = None
48
- PILImageClass = None
49
47
50
48
51
49
# This is to get around https://github.com/googleapis/proto-plus-python/issues/488
@@ -113,7 +111,7 @@ class Image:
113
111
__module__ = "vertexai.vision_models"
114
112
115
113
_loaded_bytes : Optional [bytes ] = None
116
- _loaded_image : Optional ["PILImageClass " ] = None
114
+ _loaded_image : Optional ["PIL_Image.Image " ] = None
117
115
118
116
def __init__ (
119
117
self ,
@@ -151,7 +149,7 @@ def _image_bytes(self, value: bytes):
151
149
self ._loaded_bytes = value
152
150
153
151
@property
154
- def _pil_image (self ) -> "PILImageClass" :
152
+ def _pil_image (self ) -> "PIL_Image.Image" : # type: ignore
155
153
if self ._loaded_image is None :
156
154
if not PIL_Image :
157
155
raise RuntimeError (
You can’t perform that action at this time.
0 commit comments