Skip to content

Commit 1dfd59d

Browse files
committed
Don't depend on packaging module
1 parent 425a923 commit 1dfd59d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchvision/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88

99
import numpy as np
1010
import torch
11-
from packaging import version
1211
from PIL import Image, ImageColor, ImageDraw, ImageFont
1312

1413

@@ -184,7 +183,8 @@ def _Image_fromarray(
184183
mode paramter. See:
185184
https://pillow.readthedocs.io/en/stable/releasenotes/11.3.0.html#image-fromarray-mode-parameter
186185
"""
187-
if version.parse(Image.__version__) >= version.parse("11.3.0"):
186+
PILLOW_VERSION = tuple(int(x) for x in PIL.__version__.split("."))
187+
if PILLOW_VERSION >= (11, 3):
188188
return Image.fromarray(obj)
189189
else:
190190
return Image.fromarray(obj, mode)

0 commit comments

Comments
 (0)