fix method call

ArneBinder · Sep 17, 2024 · b099a1d · b099a1d
1 parent 2173844
commit b099a1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/pie_modules/taskmodules/common/taskmodule_with_document_converter.py b/src/pie_modules/taskmodules/common/taskmodule_with_document_converter.py
@@ -89,8 +89,8 @@ def decode(self, **kwargs) -> Sequence[DocumentType]:
         result = []
         for doc in decoded_documents:
             original_document = doc.metadata["original_document"]
-            self._integrate_predictions_into_original_document(
-                document=doc, original_document=original_document
+            self._integrate_predictions_from_converted_document(
+                converted_document=doc, document=original_document
             )
             result.append(original_document)
         return result