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

Update auto processor to check image processor created (#20021)

Browse Source
This commit is contained in:
amyeroberts
2022-11-02 15:19:33 +00:00
committed by GitHub
parent 49b77b89ea
commit 9aedce99b0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/auto/test_processor_auto.py
View File

@@ -206,9 +206,9 @@ class AutoFeatureExtractorTest(unittest.TestCase):
processor = AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-bert") processor = AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-bert")
self.assertEqual(processor.__class__.__name__, "BertTokenizerFast") self.assertEqual(processor.__class__.__name__, "BertTokenizerFast")
def test_auto_processor_creates_feature_extractor(self): def test_auto_processor_creates_image_processor(self):
processor = AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-convnext") processor = AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-convnext")
self.assertEqual(processor.__class__.__name__, "ConvNextFeatureExtractor") self.assertEqual(processor.__class__.__name__, "ConvNextImageProcessor")
@is_staging_test @is_staging_test