From 62f28bc152020d8c19f58e9c6a988bc7084a4bd6 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 14 Oct 2022 16:29:54 +0200 Subject: [PATCH] Fix `ImageToTextPipelineTests.test_small_model_tf` (#19565) Co-authored-by: ydshieh --- .../pipelines/test_pipelines_image_to_text.py | 25 ++++--------------- 1 file changed, 5 insertions(+), 20 deletions(-) diff --git a/tests/pipelines/test_pipelines_image_to_text.py b/tests/pipelines/test_pipelines_image_to_text.py index 32503033003..6c0b37c8a11 100644 --- a/tests/pipelines/test_pipelines_image_to_text.py +++ b/tests/pipelines/test_pipelines_image_to_text.py @@ -64,12 +64,7 @@ class ImageToTextPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta outputs, [ { - "generated_text": ( - " intermedi intermedi intermedi intermedi intermedi " - "explorer explorer explorer explorer explorer explorer " - "explorer medicine medicine medicine medicine medicine " - "medicine medicine" - ) + "generated_text": "growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO" }, ], ) @@ -80,23 +75,13 @@ class ImageToTextPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta [ [ { - "generated_text": ( - " intermedi intermedi intermedi intermedi intermedi " - "explorer explorer explorer explorer explorer explorer " - "explorer medicine medicine medicine medicine medicine " - "medicine medicine" - ) - }, + "generated_text": "growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO" + } ], [ { - "generated_text": ( - " intermedi intermedi intermedi intermedi intermedi " - "explorer explorer explorer explorer explorer explorer " - "explorer medicine medicine medicine medicine medicine " - "medicine medicine" - ) - }, + "generated_text": "growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO" + } ], ], )