From 4b47000962f04c8cb96878c41dd9774486861466 Mon Sep 17 00:00:00 2001
From: Arne Binder <arne.binder@dfki.de>
Date: Sun, 21 Sep 2025 18:03:50 +0200
Subject: [PATCH] remove (Iterable)TaskEncodingDataset from
 TaskModuleWithDocumentConverter

---
 .../taskmodules/common/taskmodule_with_document_converter.py  | 4 ----
 1 file changed, 4 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
index 69f8e7668..e7b706201 100644
--- a/src/pie_modules/taskmodules/common/taskmodule_with_document_converter.py
+++ b/src/pie_modules/taskmodules/common/taskmodule_with_document_converter.py
@@ -3,9 +3,7 @@
 
 from pie_core import (
     Document,
-    IterableTaskEncodingDataset,
     TaskEncoding,
-    TaskEncodingDataset,
     TaskEncodingSequence,
     TaskModule,
 )
@@ -80,8 +78,6 @@ def encode(self, documents: Union[DocumentType, Iterable[DocumentType]], **kwarg
         Sequence[TaskEncodingType],
         TaskEncodingSequence[TaskEncodingType, DocumentType],
         Iterator[TaskEncodingType],
-        TaskEncodingDataset[TaskEncodingType],
-        IterableTaskEncodingDataset[TaskEncodingType],
     ]:
         converted_documents: Union[DocumentType, Iterable[DocumentType]]
         if isinstance(documents, Document):