Skip 2 tests in VisionTextDualEncoderProcessorTest (#20098)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2022-11-07 14:51:05 +01:00 committed by GitHub
parent b77406bcb2
commit 4ab6e9e2f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -76,6 +76,8 @@ class VisionTextDualEncoderProcessorTest(unittest.TestCase):
return image_inputs
# TODO (Amy): fix me
@unittest.skip("An issue introduced in PR #19796 will be fixed by `AutoImageProcessor`")
def test_save_load_pretrained_default(self):
tokenizer = self.get_tokenizer()
feature_extractor = self.get_feature_extractor()
@ -91,6 +93,8 @@ class VisionTextDualEncoderProcessorTest(unittest.TestCase):
self.assertEqual(processor.feature_extractor.to_json_string(), feature_extractor.to_json_string())
self.assertIsInstance(processor.feature_extractor, ViTFeatureExtractor)
# TODO (Amy): fix me
@unittest.skip("An issue introduced in PR #19796 will be fixed by `AutoImageProcessor`")
def test_save_load_pretrained_additional_features(self):
processor = VisionTextDualEncoderProcessor(
tokenizer=self.get_tokenizer(), feature_extractor=self.get_feature_extractor()