fix import

ArneBinder · Nov 8, 2023 · fe751a8 · fe751a8
1 parent 14d7f38
commit fe751a8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/pie_datasets/document/processing/regex_partitioner.py b/src/pie_datasets/document/processing/regex_partitioner.py
@@ -10,6 +10,7 @@
 from pytorch_ie.documents import TextBasedDocument
 
 from pie_datasets import Dataset, IterableDataset
+from pie_datasets.dataset_dict import EnterDatasetMixin, ExitDatasetMixin
 
 logger = logging.getLogger(__name__)