SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix type - update any PIL.Image.Resampling (#20172)

Browse Source
This commit is contained in:
amyeroberts
2022-11-11 16:55:59 +00:00
committed by GitHub
parent cbbeca3d17
commit 6cc06d1739
8 changed files with 23 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/models/flava/test_feature_extraction_flava.py
View File

@@ -31,6 +31,7 @@ if is_vision_available():
import PIL
from transformers import FlavaFeatureExtractor
from transformers.image_utils import PILImageResampling
from transformers.models.flava.image_processing_flava import (
FLAVA_CODEBOOK_MEAN,
FLAVA_CODEBOOK_STD,
@@ -89,7 +90,7 @@ class FlavaFeatureExtractionTester(unittest.TestCase):
self.min_resolution = min_resolution
self.max_resolution = max_resolution
self.size = size
self.resample = resample if resample is not None else PIL.Image.Resampling.BICUBIC
self.resample = resample if resample is not None else PILImageResampling.BICUBIC
self.do_normalize = do_normalize
self.image_mean = image_mean
self.image_std = image_std
@@ -105,7 +106,7 @@ class FlavaFeatureExtractionTester(unittest.TestCase):
self.codebook_do_resize = codebook_do_resize
self.codebook_size = codebook_size
self.codebook_resample = codebook_resample if codebook_resample is not None else PIL.Image.Resampling.LANCZOS
self.codebook_resample = codebook_resample if codebook_resample is not None else PILImageResampling.LANCZOS
self.codebook_do_center_crop = codebook_do_center_crop
self.codebook_crop_size = codebook_crop_size
self.codebook_do_map_pixels = codebook_do_map_pixels