Manual parsing of PIL version

Author: scotts

test/common_utils.py

@@ -12,6 +12,7 @@
 from subprocess import CalledProcessError, check_output, STDOUT
 
 import numpy as np
+import PIL
 import pytest
 import torch
 import torch.testing

torchvision/utils.py

@@ -8,10 +8,12 @@
 
 import numpy as np
 import torch
-from PIL import Image, ImageColor, ImageDraw, ImageFont
+from PIL import __version__ as PILLOW_VERSION, Image, ImageColor, ImageDraw, ImageFont
 
+PILLOW_VERSION = tuple(int(x) for x in PILLOW_VERSION.split("."))
 
 __all__ = [
+    "_Image_fromarray",
     "make_grid",
     "save_image",
     "draw_bounding_boxes",
@@ -183,7 +185,6 @@ def _Image_fromarray(
     mode paramter. See:
       https://pillow.readthedocs.io/en/stable/releasenotes/11.3.0.html#image-fromarray-mode-parameter
     """
-    PILLOW_VERSION = tuple(int(x) for x in PIL.__version__.split("."))
     if PILLOW_VERSION >= (11, 3):
         return Image.fromarray(obj)
     else: