From f16bbf1475cfb84124fec36b0f2188538e90d26a Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Thu, 29 Sep 2022 12:25:15 -0400 Subject: [PATCH] Skip pipeline tests (#19248) --- tests/pipelines/test_pipelines_image_segmentation.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/pipelines/test_pipelines_image_segmentation.py b/tests/pipelines/test_pipelines_image_segmentation.py index 611e9d2ed1..f35c6640c5 100644 --- a/tests/pipelines/test_pipelines_image_segmentation.py +++ b/tests/pipelines/test_pipelines_image_segmentation.py @@ -74,6 +74,9 @@ class ImageSegmentationPipelineTests(unittest.TestCase, metaclass=PipelineTestCa } def get_test_pipeline(self, model, tokenizer, feature_extractor): + # Fix me Alara + if model.__class__.__name__ == "DetrForSegmentation": + return None, None image_segmenter = ImageSegmentationPipeline(model=model, feature_extractor=feature_extractor) return image_segmenter, [ "./tests/fixtures/tests_samples/COCO/000000039769.png", @@ -147,6 +150,7 @@ class ImageSegmentationPipelineTests(unittest.TestCase, metaclass=PipelineTestCa pass @require_torch + @unittest.skip("Fix me Alara!") def test_small_model_pt(self): model_id = "hf-internal-testing/tiny-detr-mobilenetsv3-panoptic"