reset commits b8128ca and 99176e0

Author: ArneBinder

src/pytorch_ie/data/__init__.py

@@ -1,8 +1,13 @@
+from typing import Dict, Union
+
+from datasets import Split
+
 from .builder import GeneratorBasedBuilder
 from .dataset import Dataset, IterableDataset
-from .dataset_dict import DatasetDict
 from .dataset_formatter import DocumentFormatter
 
+DatasetDict = Dict[Union[str, Split], Dataset]
+
 __all__ = [
     "GeneratorBasedBuilder",
     "Dataset",