Fix pipelines tests (#34049)

* Fix wrong skip annotation

* Remove error raise
This commit is contained in:
Pavel Iakubovskii 2024-10-10 12:04:06 +01:00 committed by GitHub
parent a84c413773
commit 66e08dba71
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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 "