From 9aedce99b0c5843c51530bd4958806b3384627f8 Mon Sep 17 00:00:00 2001 From: amyeroberts <22614925+amyeroberts@users.noreply.github.com> Date: Wed, 2 Nov 2022 15:19:33 +0000 Subject: [PATCH] Update auto processor to check image processor created (#20021) --- tests/models/auto/test_processor_auto.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/auto/test_processor_auto.py b/tests/models/auto/test_processor_auto.py index fe57078a61..6cddfc1376 100644 --- a/tests/models/auto/test_processor_auto.py +++ b/tests/models/auto/test_processor_auto.py @@ -206,9 +206,9 @@ class AutoFeatureExtractorTest(unittest.TestCase): processor = AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-bert") 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") - self.assertEqual(processor.__class__.__name__, "ConvNextFeatureExtractor") + self.assertEqual(processor.__class__.__name__, "ConvNextImageProcessor") @is_staging_test