From 66e08dba71a10bc3c00b8501edee7aa36cf7541d Mon Sep 17 00:00:00 2001 From: Pavel Iakubovskii Date: Thu, 10 Oct 2024 12:04:06 +0100 Subject: [PATCH] Fix pipelines tests (#34049) * Fix wrong skip annotation * Remove error raise --- tests/test_pipeline_mixin.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) 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 @@ def run_task_tests(self, task, torch_dtype="float32"): 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 @@ def run_task_tests(self, task, torch_dtype="float32"): 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 "