fix

huggingface · Apr 18, 2024 · 191d780 · 191d780
1 parent 661fc88
commit 191d780
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/pipelines/test_pipelines_document_question_answering.py b/tests/pipelines/test_pipelines_document_question_answering.py
@@ -103,7 +103,9 @@ def run_pipeline_test(self, dqa_pipeline, examples):
     @require_detectron2
     @require_pytesseract
     def test_small_model_pt(self):
-        dqa_pipeline = pipeline("document-question-answering", model="hf-internal-testing/tiny-random-layoutlmv2-for-dqa-test")
+        dqa_pipeline = pipeline(
+            "document-question-answering", model="hf-internal-testing/tiny-random-layoutlmv2-for-dqa-test"
+        )
         image = INVOICE_URL
         question = "How many cats are there?"