Fix some pipeline tests (#21401)

* fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-02-02 19:03:31 +01:00
committed by GitHub
parent 145bf41c13
commit a6d8a149a8
20 changed files with 69 additions and 37 deletions

View File

@@ -61,7 +61,7 @@ class DocumentQuestionAnsweringPipelineTests(unittest.TestCase, metaclass=Pipeli
@require_vision
def get_test_pipeline(self, model, tokenizer, processor):
dqa_pipeline = pipeline(
"document-question-answering", model=model, tokenizer=tokenizer, feature_extractor=processor
"document-question-answering", model=model, tokenizer=tokenizer, image_processor=processor
)
image = INVOICE_URL
@@ -81,11 +81,6 @@ class DocumentQuestionAnsweringPipelineTests(unittest.TestCase, metaclass=Pipeli
"question": question,
"word_boxes": word_boxes,
},
{
"image": None,
"question": question,
"word_boxes": word_boxes,
},
]
return dqa_pipeline, examples
@@ -99,7 +94,7 @@ class DocumentQuestionAnsweringPipelineTests(unittest.TestCase, metaclass=Pipeli
{"score": ANY(float), "answer": ANY(str), "start": ANY(int), "end": ANY(int)},
]
]
* 4,
* 3,
)
@require_torch