Skip to content

Commit 57f2593

Browse files
committed
fix typing?
Change-Id: I5005dd38b7d3a03edc1c54d42f3a6c8f959d7274
1 parent d2f1a4e commit 57f2593

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

google/generativeai/vision_models/_vision_models.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def _image_bytes(self, value: bytes):
149149
self._loaded_bytes = value
150150

151151
@property
152-
def _pil_image(self) -> "PIL_Image.Image":
152+
def _pil_image(self) -> "PIL_Image.Image": # type: ignore
153153
if self._loaded_image is None:
154154
if not PIL_Image:
155155
raise RuntimeError(
@@ -198,7 +198,7 @@ def _as_base64_string(self) -> str:
198198
return base64.b64encode(self._image_bytes).decode("ascii")
199199

200200
def _repr_png_(self):
201-
return self._pil_image._repr_png_() # type:ignore
201+
return self._pil_image._repr_png_() # type:ignore
202202

203203

204204
class ImageGenerationModel:
@@ -371,9 +371,7 @@ def generate_images(
371371
aspect_ratio: Optional[AspectRatio] = None,
372372
guidance_scale: Optional[float] = None,
373373
language: Optional[str] = None,
374-
safety_filter_level: Optional[
375-
SafetyFilterLevel
376-
] = None,
374+
safety_filter_level: Optional[SafetyFilterLevel] = None,
377375
person_generation: Optional[PersonGeneration] = None,
378376
) -> "ImageGenerationResponse":
379377
"""Generates images from text prompt.

0 commit comments

Comments
 (0)