diff --git a/tests/test_pipeline_mixin.py b/tests/test_pipeline_mixin.py index 7d030893606..19f503dc6e5 100644 --- a/tests/test_pipeline_mixin.py +++ b/tests/test_pipeline_mixin.py @@ -214,8 +214,6 @@ class PipelineTesterMixin: image_processor_names.append(cls_name) elif "FeatureExtractor" in cls_name: feature_extractor_names.append(cls_name) - else: - raise ValueError(f"Unknown processor class: {cls_name}") # Processor classes are not in tiny models JSON file, so extract them from the mapping # processors are mapped to instance, e.g. "XxxProcessor" @@ -242,12 +240,12 @@ class PipelineTesterMixin: commit=commit, torch_dtype=torch_dtype, ) + at_least_one_model_is_tested = True + if task in task_to_pipeline_and_spec_mapping: pipeline, hub_spec = task_to_pipeline_and_spec_mapping[task] compare_pipeline_args_to_hub_spec(pipeline, hub_spec) - at_least_one_model_is_tested = True - if not at_least_one_model_is_tested: self.skipTest( f"{self.__class__.__name__}::test_pipeline_{task.replace('-', '_')}_{torch_dtype} is skipped: Could not find any "