From 85d1a3809917375d3e1763766ba51c923819b45f Mon Sep 17 00:00:00 2001
From: Arne Binder <arne.binder@dfki.de>
Date: Tue, 14 Feb 2023 11:39:00 +0100
Subject: [PATCH] add missing result_document_type

---
 src/pytorch_ie/pipeline.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/pytorch_ie/pipeline.py b/src/pytorch_ie/pipeline.py
index 9bc4f222..64db8019 100644
--- 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: