Merge pull request #31 from ArneBinder/remove_pytorch_ie_dataset_left…

…overs remove leftovers from datasets in pytorch-ie
ArneBinder · Nov 8, 2023 · 8d1e9dc · 8d1e9dc
2 parents 1461e98 + 4f5b878
commit 8d1e9dc
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/dataset_builders/pie/conll2003/conll2003.py b/dataset_builders/pie/conll2003/conll2003.py
@@ -1,7 +1,6 @@
 from dataclasses import dataclass
 
 import datasets
-import pytorch_ie.data.builder
 from pytorch_ie.annotations import LabeledSpan
 from pytorch_ie.core import AnnotationList, annotation_field
 from pytorch_ie.documents import TextDocument, TextDocumentWithLabeledSpans

diff --git a/src/pie_datasets/document/processing/regex_partitioner.py b/src/pie_datasets/document/processing/regex_partitioner.py
@@ -6,11 +6,10 @@
 import statistics
 from typing import Any, Callable, Iterable, Iterator, Match, TypeVar
 
-from pytorch_ie import Dataset, IterableDataset
 from pytorch_ie.annotations import LabeledSpan
 from pytorch_ie.documents import TextBasedDocument
 
-from pie_datasets import EnterDatasetMixin, ExitDatasetMixin
+from pie_datasets import Dataset, EnterDatasetMixin, ExitDatasetMixin, IterableDataset
 
 logger = logging.getLogger(__name__)