Skip to content

GitLab

Unverified Commit 62f28bc1 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `ImageToTextPipelineTests.test_small_model_tf` (#19565) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent e82c1cb7
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 20 deletions
tests/pipelines/test_pipelines_image_to_text.py
View file @ 62f28bc1
...@@ -64,12 +64,7 @@ class ImageToTextPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta ...@@ -64,12 +64,7 @@ class ImageToTextPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta
outputs, outputs,
[ [
{ {
"generated_text": ( "generated_text": "growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO"
" intermedi intermedi intermedi intermedi intermedi "
"explorer explorer explorer explorer explorer explorer "
"explorer medicine medicine medicine medicine medicine "
"medicine medicine"
)
}, },
], ],
) )
...@@ -80,23 +75,13 @@ class ImageToTextPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta ...@@ -80,23 +75,13 @@ class ImageToTextPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta
[ [
[ [
{ {
"generated_text": ( "generated_text": "growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO"
" intermedi intermedi intermedi intermedi intermedi " }
"explorer explorer explorer explorer explorer explorer "
"explorer medicine medicine medicine medicine medicine "
"medicine medicine"
)
},
], ],
[ [
{ {
"generated_text": ( "generated_text": "growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO"
" intermedi intermedi intermedi intermedi intermedi " }
"explorer explorer explorer explorer explorer explorer "
"explorer medicine medicine medicine medicine medicine "
"medicine medicine"
)
},
], ],
], ],
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment