Skip tests properly (#31308)
* Skip tests properly * [test_all] * Add 'reason' as kwarg for skipTest * [test_all] Fix up * [test_all]
This commit is contained in:
@@ -176,15 +176,15 @@ class FeatureExtractionPipelineTests(unittest.TestCase):
|
||||
|
||||
def get_test_pipeline(self, model, tokenizer, processor):
|
||||
if tokenizer is None:
|
||||
self.skipTest("No tokenizer")
|
||||
return
|
||||
self.skipTest(reason="No tokenizer")
|
||||
elif (
|
||||
type(model.config) in FEATURE_EXTRACTOR_MAPPING
|
||||
or isinstance(model.config, LxmertConfig)
|
||||
or type(model.config) in IMAGE_PROCESSOR_MAPPING
|
||||
):
|
||||
self.skipTest("This is a bimodal model, we need to find a more consistent way to switch on those models.")
|
||||
return
|
||||
self.skipTest(
|
||||
reason="This is a bimodal model, we need to find a more consistent way to switch on those models."
|
||||
)
|
||||
elif model.config.is_encoder_decoder:
|
||||
self.skipTest(
|
||||
"""encoder_decoder models are trickier for this pipeline.
|
||||
@@ -193,8 +193,6 @@ class FeatureExtractionPipelineTests(unittest.TestCase):
|
||||
For now ignore those.
|
||||
"""
|
||||
)
|
||||
|
||||
return
|
||||
feature_extractor = FeatureExtractionPipeline(model=model, tokenizer=tokenizer, feature_extractor=processor)
|
||||
return feature_extractor, ["This is a test", "This is another test"]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user