changes:

 - `core` just re-exports everything from `pie_core`
 - `metrics` just re-exports everything from `pie_documents.metrics`
 - `annotations` just re-exports everything from `pie_documents.annotations`
 - `documents` just re-exports everything from `pie_documents.documents`
 - remove `utils.hydra`
 - remove `auto`

Author: ArneBinder

src/pytorch_ie/__init__.py

@@ -1,7 +1,7 @@
 # flake8: noqa
 
-from pytorch_ie.auto import AutoModel, AutoTaskModule
 from pytorch_ie.core import *
 from pytorch_ie.datamodule import PieDataModule
 from pytorch_ie.dataset import IterableTaskEncodingDataset, TaskEncodingDataset
+from pytorch_ie.model import PyTorchIEModel
 from pytorch_ie.pipeline import PyTorchIEPipeline

src/pytorch_ie/annotations.py

@@ -1,13 +1 @@
-# backward compatibility
-from pie_documents.annotations import (
-    BinaryRelation,
-    Label,
-    LabeledMultiSpan,
-    LabeledSpan,
-    MultiLabel,
-    MultiLabeledBinaryRelation,
-    MultiLabeledSpan,
-    MultiSpan,
-    NaryRelation,
-    Span,
-)
+from pie_documents.annotations import *

src/pytorch_ie/auto.py

@@ -1,35 +1 @@
-import sys
-
-import pie_core
-from pie_core import taskmodule
-from pie_core.document import Annotation, AnnotationLayer, Document, annotation_field
-from pie_core.metric import DocumentMetric
-from pie_core.module_mixins import (
-    EnterDatasetDictMixin,
-    EnterDatasetMixin,
-    ExitDatasetDictMixin,
-    ExitDatasetMixin,
-    WithDocumentTypeMixin,
-)
-from pie_core.preparable import PreparableMixin
-from pie_core.statistic import DocumentStatistic
-from pie_core.taskencoding import TaskEncoding, TaskEncodingSequence
-from pie_core.taskmodule import TaskModule
-
-from pytorch_ie import model
-from pytorch_ie.dataset import IterableTaskEncodingDataset, TaskEncodingDataset
-from pytorch_ie.model import PyTorchIEModel
-
-submodules = ["document", "taskmodule", "metric", "statistic"]
-for sub in submodules:
-    module = getattr(pie_core, sub)
-    sys.modules[f"{__name__}.{sub}"] = module
-
-sys.modules[f"{__name__}.model"] = model
-
-taskmodule.TaskEncodingDataset = TaskEncodingDataset
-taskmodule.IterableTaskEncodingDataset = IterableTaskEncodingDataset
-
-# backwards compatibility
-AnnotationList = AnnotationLayer
-RequiresDocumentTypeMixin = WithDocumentTypeMixin
+from pie_core import *

src/pytorch_ie/core/__init__.py

@@ -1,31 +1 @@
-# backwards compatibility
-from pie_documents.documents import (
-    DocumentWithLabel,
-    DocumentWithMultiLabel,
-    TextBasedDocument,
-    TextDocumentWithLabel,
-    TextDocumentWithLabeledMultiSpans,
-    TextDocumentWithLabeledMultiSpansAndBinaryRelations,
-    TextDocumentWithLabeledMultiSpansAndLabeledPartitions,
-    TextDocumentWithLabeledMultiSpansBinaryRelationsAndLabeledPartitions,
-    TextDocumentWithLabeledPartitions,
-    TextDocumentWithLabeledSpans,
-    TextDocumentWithLabeledSpansAndBinaryRelations,
-    TextDocumentWithLabeledSpansAndLabeledPartitions,
-    TextDocumentWithLabeledSpansAndSentences,
-    TextDocumentWithLabeledSpansBinaryRelationsAndLabeledPartitions,
-    TextDocumentWithMultiLabel,
-    TextDocumentWithSentences,
-    TextDocumentWithSpans,
-    TextDocumentWithSpansAndBinaryRelations,
-    TextDocumentWithSpansAndLabeledPartitions,
-    TextDocumentWithSpansBinaryRelationsAndLabeledPartitions,
-    TokenBasedDocument,
-    WithMetadata,
-    WithText,
-    WithTokens,
-)
-from typing_extensions import TypeAlias
-
-# backwards compatibility
-TextDocument: TypeAlias = TextBasedDocument
+from pie_documents.documents import *

src/pytorch_ie/documents.py

@@ -1,2 +1 @@
-# backwards compatibility
-from pie_documents.metrics import ConfusionMatrix, F1Metric, SQuADF1
+from pie_documents.metrics import *

src/pytorch_ie/metrics/__init__.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict
 
 import torch
-from pie_core import Auto, Model
+from pie_core import Model
 from pytorch_lightning import LightningModule