Safer import

huggingface · May 1, 2024 · 3268821 · 3268821
1 parent b5be4e9
commit 3268821
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/transformers/image_processing_utils.py b/src/transformers/image_processing_utils.py
@@ -26,7 +26,7 @@
 from .dynamic_module_utils import custom_object_save
 from .feature_extraction_utils import BatchFeature as BaseBatchFeature
 from .image_transforms import center_crop, normalize, rescale
-from .image_utils import ChannelDimension, PILImageResampling, is_scaled_image, to_numpy_array, valid_images
+from .image_utils import ChannelDimension, is_scaled_image, to_numpy_array, valid_images
 from .utils import (
  IMAGE_PROCESSOR_NAME,
  PushToHubMixin,
@@ -44,6 +44,8 @@
 if is_vision_available():
  from PIL import Image
 
+ from .image_utils import PILImageResampling
+
 logger = logging.get_logger(__name__)