From bed2edb99f380d860e041fdfaedcdcdd1a585365 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 19 Oct 2022 16:24:03 +0200 Subject: [PATCH] Specify TF framework explicitly in more pipeline tests (#19748) Co-authored-by: ydshieh --- tests/pipelines/test_pipelines_image_classification.py | 2 +- tests/pipelines/test_pipelines_image_to_text.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/test_pipelines_image_classification.py b/tests/pipelines/test_pipelines_image_classification.py index 59a58ff44d..8c83888ba0 100644 --- a/tests/pipelines/test_pipelines_image_classification.py +++ b/tests/pipelines/test_pipelines_image_classification.py @@ -140,7 +140,7 @@ class ImageClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTest @require_tf def test_small_model_tf(self): small_model = "hf-internal-testing/tiny-random-vit" - image_classifier = pipeline("image-classification", model=small_model) + image_classifier = pipeline("image-classification", model=small_model, framework="tf") outputs = image_classifier("http://images.cocodataset.org/val2017/000000039769.jpg") self.assertEqual( diff --git a/tests/pipelines/test_pipelines_image_to_text.py b/tests/pipelines/test_pipelines_image_to_text.py index 16069971ce..2da17a1d22 100644 --- a/tests/pipelines/test_pipelines_image_to_text.py +++ b/tests/pipelines/test_pipelines_image_to_text.py @@ -154,7 +154,7 @@ class ImageToTextPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta @slow @require_tf def test_large_model_tf(self): - pipe = pipeline("image-to-text", model="ydshieh/vit-gpt2-coco-en") + pipe = pipeline("image-to-text", model="ydshieh/vit-gpt2-coco-en", framework="tf") image = "./tests/fixtures/tests_samples/COCO/000000039769.png" outputs = pipe(image)