From 3aaabaa214736d3db8f0f928ed1aa943b45a7505 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 21 Oct 2022 14:35:20 +0200 Subject: [PATCH] Update `ImageToTextPipelineTests.test_small_model_tf` (#19785) * update expected values for the correct TF checkpoint * Run test * Clean up * fix 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 2da17a1d225..652c140ae51 100644 --- a/tests/pipelines/test_pipelines_image_to_text.py +++ b/tests/pipelines/test_pipelines_image_to_text.py @@ -64,12 +64,7 @@ def test_small_model_tf(self): 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 @@ def test_small_model_tf(self): [ [ { - "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" + } ], ], )