add missing result_document_type

ArneBinder · Sep 17, 2023 · 85d1a38 · 85d1a38
1 parent 47ca43b
commit 85d1a38
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/pytorch_ie/pipeline.py b/src/pytorch_ie/pipeline.py
@@ -398,6 +398,7 @@ def __call__(
                         forward_params=forward_params,
                         postprocess_params=postprocess_params,
                     ),
+                    result_document_type=documents.document_type,
                     batched=True,
                 )
             finally: