Skip to content

Commit fb033db

Browse files
authored
Add PngImagePlugin to imports. (#438)
1 parent 85ee8c1 commit fb033db

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

google/generativeai/types/content_types.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,13 +30,15 @@
3030

3131
if typing.TYPE_CHECKING:
3232
import PIL.Image
33+
import PIL.PngImagePlugin
3334
import IPython.display
3435

3536
IMAGE_TYPES = (PIL.Image.Image, IPython.display.Image)
3637
else:
3738
IMAGE_TYPES = ()
3839
try:
3940
import PIL.Image
41+
import PIL.PngImagePlugin
4042

4143
IMAGE_TYPES = IMAGE_TYPES + (PIL.Image.Image,)
4244
except ImportError:

0 commit comments

Comments
 (0)