From 6d832123cbf4ed01cdc11c28688801e247e65a6d Mon Sep 17 00:00:00 2001 From: Arne Binder Date: Thu, 17 Jul 2025 15:59:49 +0200 Subject: [PATCH] import from pie_core directly --- src/pie_modules/annotations.py | 5 +++-- src/pie_modules/document/processing/merge_multi_spans.py | 2 +- .../document/processing/merge_spans_via_relation.py | 2 +- src/pie_modules/document/processing/regex_partitioner.py | 2 +- .../document/processing/relation_argument_sorter.py | 2 +- src/pie_modules/document/processing/text_span_trimmer.py | 2 +- src/pie_modules/document/processing/tokenization.py | 2 +- src/pie_modules/documents.py | 2 +- src/pie_modules/metrics/confusion_matrix.py | 4 ++-- src/pie_modules/metrics/f1.py | 4 ++-- .../metrics/relation_argument_distance_collector.py | 4 ++-- src/pie_modules/metrics/span_coverage_collector.py | 2 +- src/pie_modules/metrics/span_length_collector.py | 2 +- src/pie_modules/metrics/squad_f1.py | 2 +- src/pie_modules/metrics/statistics.py | 2 +- src/pie_modules/models/common/has_taskmodule.py | 3 +-- src/pie_modules/taskmodules/common/interfaces.py | 2 +- src/pie_modules/taskmodules/common/mixins.py | 2 +- .../common/taskmodule_with_document_converter.py | 5 +++-- src/pie_modules/taskmodules/cross_text_binary_coref.py | 2 +- .../taskmodules/extractive_question_answering.py | 2 +- .../labeled_span_extraction_by_token_classification.py | 2 +- .../precision_recall_and_f1_for_labeled_annotations.py | 2 +- .../taskmodules/pointer_network_for_end2end_re.py | 6 +++--- src/pie_modules/taskmodules/re_span_pair_classification.py | 2 +- .../taskmodules/re_text_classification_with_indices.py | 2 +- src/pie_modules/taskmodules/text_to_text.py | 6 +++--- src/pie_modules/utils/hydra.py | 4 ++-- tests/conftest.py | 2 +- tests/document/processing/test_regex_partitioner.py | 2 +- tests/document/processing/test_relation_argument_sorter.py | 2 +- tests/document/processing/test_text_span_trimmer.py | 2 +- tests/document/processing/test_tokenization.py | 2 +- tests/metrics/test_confusion_matrix.py | 2 +- tests/metrics/test_f1.py | 2 +- tests/metrics/test_relation_argument_distance_collector.py | 2 +- tests/metrics/test_span_coverage_collector.py | 2 +- tests/metrics/test_span_length_collector.py | 2 +- tests/taskmodules/common/test_mixins.py | 2 +- .../common/test_taskmodule_with_document_converter.py | 2 +- tests/taskmodules/test_extractive_question_answering.py | 2 +- .../test_labeled_span_extraction_by_token_classification.py | 2 +- tests/taskmodules/test_pointer_network_for_end2end_re.py | 2 +- tests/taskmodules/test_re_span_pair_classification.py | 2 +- .../taskmodules/test_re_text_classification_with_indices.py | 2 +- tests/taskmodules/test_text2text.py | 2 +- tests/taskmodules/test_text2text_with_guidance.py | 2 +- tests/test_annotations.py | 2 +- tests/utils/test_hydra.py | 2 +- 49 files changed, 61 insertions(+), 60 deletions(-) diff --git a/src/pie_modules/annotations.py b/src/pie_modules/annotations.py index 208c2d1b9..d7710595f 100644 --- a/src/pie_modules/annotations.py +++ b/src/pie_modules/annotations.py @@ -1,5 +1,7 @@ import dataclasses -from typing import Optional, Tuple +from typing import Optional + +from pie_core import Annotation # re-export all annotations from pytorch_ie to have a single entry point from pytorch_ie.annotations import ( @@ -15,7 +17,6 @@ Span, _post_init_single_label, ) -from pytorch_ie.core import Annotation @dataclasses.dataclass(eq=True, frozen=True) diff --git a/src/pie_modules/document/processing/merge_multi_spans.py b/src/pie_modules/document/processing/merge_multi_spans.py index 1ce2313f2..e7a4d58dc 100644 --- a/src/pie_modules/document/processing/merge_multi_spans.py +++ b/src/pie_modules/document/processing/merge_multi_spans.py @@ -2,7 +2,7 @@ from typing import Generic, TypeVar, get_args -from pytorch_ie.core import Annotation, Document +from pie_core import Annotation, Document from pie_modules.annotations import MultiSpan, Span diff --git a/src/pie_modules/document/processing/merge_spans_via_relation.py b/src/pie_modules/document/processing/merge_spans_via_relation.py index 6be13e88c..46e874af1 100644 --- a/src/pie_modules/document/processing/merge_spans_via_relation.py +++ b/src/pie_modules/document/processing/merge_spans_via_relation.py @@ -1,7 +1,7 @@ import logging from typing import Optional, Sequence, Set, Tuple, TypeVar, Union -from pytorch_ie.core import AnnotationLayer, Document +from pie_core import AnnotationLayer, Document from pie_modules.annotations import BinaryRelation, LabeledMultiSpan, LabeledSpan from pie_modules.utils import resolve_type diff --git a/src/pie_modules/document/processing/regex_partitioner.py b/src/pie_modules/document/processing/regex_partitioner.py index dadb116fa..8d4c23907 100644 --- a/src/pie_modules/document/processing/regex_partitioner.py +++ b/src/pie_modules/document/processing/regex_partitioner.py @@ -6,7 +6,7 @@ import statistics from typing import Any, Callable, Iterable, Iterator, Match, TypeVar -from pytorch_ie.core import EnterDatasetMixin, ExitDatasetMixin +from pie_core import EnterDatasetMixin, ExitDatasetMixin from pie_modules.annotations import LabeledSpan from pie_modules.documents import TextBasedDocument diff --git a/src/pie_modules/document/processing/relation_argument_sorter.py b/src/pie_modules/document/processing/relation_argument_sorter.py index be7af2e6f..bc4ead748 100644 --- a/src/pie_modules/document/processing/relation_argument_sorter.py +++ b/src/pie_modules/document/processing/relation_argument_sorter.py @@ -3,7 +3,7 @@ import logging from typing import TypeVar -from pytorch_ie.core import Annotation, AnnotationLayer, Document +from pie_core import Annotation, AnnotationLayer, Document from pie_modules.annotations import BinaryRelation, LabeledMultiSpan, LabeledSpan diff --git a/src/pie_modules/document/processing/text_span_trimmer.py b/src/pie_modules/document/processing/text_span_trimmer.py index fe492c836..412e01110 100644 --- a/src/pie_modules/document/processing/text_span_trimmer.py +++ b/src/pie_modules/document/processing/text_span_trimmer.py @@ -3,7 +3,7 @@ import logging from typing import TypeVar -from pytorch_ie.core import AnnotationLayer, Document +from pie_core import AnnotationLayer, Document from pie_modules.annotations import LabeledMultiSpan, LabeledSpan, Span diff --git a/src/pie_modules/document/processing/tokenization.py b/src/pie_modules/document/processing/tokenization.py index 544f41b96..bce5a2add 100644 --- a/src/pie_modules/document/processing/tokenization.py +++ b/src/pie_modules/document/processing/tokenization.py @@ -16,7 +16,7 @@ Union, ) -from pytorch_ie.core import Annotation +from pie_core import Annotation from transformers import PreTrainedTokenizer from pie_modules.annotations import MultiSpan, Span diff --git a/src/pie_modules/documents.py b/src/pie_modules/documents.py index 5264d0c69..bf07de932 100644 --- a/src/pie_modules/documents.py +++ b/src/pie_modules/documents.py @@ -1,6 +1,6 @@ import dataclasses -from pytorch_ie.core import AnnotationLayer, annotation_field +from pie_core import AnnotationLayer, annotation_field # re-export all documents from pytorch_ie to have a single entry point from pytorch_ie.documents import ( diff --git a/src/pie_modules/metrics/confusion_matrix.py b/src/pie_modules/metrics/confusion_matrix.py index 16f962426..42bd82fd7 100644 --- a/src/pie_modules/metrics/confusion_matrix.py +++ b/src/pie_modules/metrics/confusion_matrix.py @@ -3,8 +3,8 @@ from typing import Callable, Dict, Optional, Tuple, Union import pandas as pd -from pytorch_ie.core import Annotation, Document, DocumentMetric -from pytorch_ie.utils.hydra import resolve_target +from pie_core import Annotation, Document, DocumentMetric +from pie_core.utils.hydra import resolve_target logger = logging.getLogger(__name__) diff --git a/src/pie_modules/metrics/f1.py b/src/pie_modules/metrics/f1.py index 76f68f4cf..4abec1db5 100644 --- a/src/pie_modules/metrics/f1.py +++ b/src/pie_modules/metrics/f1.py @@ -4,8 +4,8 @@ from typing import Callable, Collection, Dict, Hashable, Optional, Tuple, Union import pandas as pd -from pytorch_ie.core import Annotation, Document, DocumentMetric -from pytorch_ie.utils.hydra import resolve_target +from pie_core import Annotation, Document, DocumentMetric +from pie_core.utils.hydra import resolve_target logger = logging.getLogger(__name__) diff --git a/src/pie_modules/metrics/relation_argument_distance_collector.py b/src/pie_modules/metrics/relation_argument_distance_collector.py index 9356d0975..935c11cbc 100644 --- a/src/pie_modules/metrics/relation_argument_distance_collector.py +++ b/src/pie_modules/metrics/relation_argument_distance_collector.py @@ -1,8 +1,8 @@ from collections import defaultdict from typing import Any, Dict, List, Optional, Type, Union -from pytorch_ie.core import Document, DocumentStatistic -from pytorch_ie.utils.hydra import resolve_target +from pie_core import Document, DocumentStatistic +from pie_core.utils.hydra import resolve_target from transformers import AutoTokenizer, PreTrainedTokenizer from pie_modules.annotations import BinaryRelation, NaryRelation, Span diff --git a/src/pie_modules/metrics/span_coverage_collector.py b/src/pie_modules/metrics/span_coverage_collector.py index 40b47b5d4..9417384d8 100644 --- a/src/pie_modules/metrics/span_coverage_collector.py +++ b/src/pie_modules/metrics/span_coverage_collector.py @@ -1,7 +1,7 @@ import logging from typing import Any, Dict, List, Optional, Set, Type, Union -from pytorch_ie.core import Document, DocumentStatistic +from pie_core import Document, DocumentStatistic from transformers import AutoTokenizer, PreTrainedTokenizer from pie_modules.annotations import LabeledMultiSpan, Span diff --git a/src/pie_modules/metrics/span_length_collector.py b/src/pie_modules/metrics/span_length_collector.py index 1c457e731..a13d1d7d9 100644 --- a/src/pie_modules/metrics/span_length_collector.py +++ b/src/pie_modules/metrics/span_length_collector.py @@ -2,7 +2,7 @@ from collections import defaultdict from typing import Any, Callable, Dict, List, Optional, Type, Union -from pytorch_ie.core import Document, DocumentStatistic +from pie_core import Document, DocumentStatistic from transformers import AutoTokenizer, PreTrainedTokenizer from pie_modules.annotations import Span diff --git a/src/pie_modules/metrics/squad_f1.py b/src/pie_modules/metrics/squad_f1.py index 7ba474027..2149644e6 100644 --- a/src/pie_modules/metrics/squad_f1.py +++ b/src/pie_modules/metrics/squad_f1.py @@ -6,7 +6,7 @@ from typing import Dict, List import pandas as pd -from pytorch_ie.core import DocumentMetric +from pie_core import DocumentMetric from pie_modules.documents import TextDocumentWithQuestionsAndExtractiveAnswers diff --git a/src/pie_modules/metrics/statistics.py b/src/pie_modules/metrics/statistics.py index 97a5c3fb3..16b9d41d3 100644 --- a/src/pie_modules/metrics/statistics.py +++ b/src/pie_modules/metrics/statistics.py @@ -2,7 +2,7 @@ from collections import defaultdict from typing import Any, Callable, Dict, List, Optional, Type, Union -from pytorch_ie.core import Document, DocumentStatistic +from pie_core import Document, DocumentStatistic from transformers import AutoTokenizer, PreTrainedTokenizer from pie_modules.documents import TextBasedDocument diff --git a/src/pie_modules/models/common/has_taskmodule.py b/src/pie_modules/models/common/has_taskmodule.py index 79e2a9e9e..03cd8154c 100644 --- a/src/pie_modules/models/common/has_taskmodule.py +++ b/src/pie_modules/models/common/has_taskmodule.py @@ -1,7 +1,6 @@ from typing import Any, Dict, Optional -from pytorch_ie.auto import AutoTaskModule -from pytorch_ie.core import TaskModule +from pie_core import AutoTaskModule, TaskModule from pie_modules.models.interface import RequiresTaskmoduleConfig diff --git a/src/pie_modules/taskmodules/common/interfaces.py b/src/pie_modules/taskmodules/common/interfaces.py index b40182bf3..388b60ae9 100644 --- a/src/pie_modules/taskmodules/common/interfaces.py +++ b/src/pie_modules/taskmodules/common/interfaces.py @@ -1,7 +1,7 @@ import abc from typing import Any, Dict, Generic, List, Optional, Tuple, TypeVar -from pytorch_ie.core import Annotation +from pie_core import Annotation # Annotation Encoding type: encoding for a single annotation AE = TypeVar("AE") diff --git a/src/pie_modules/taskmodules/common/mixins.py b/src/pie_modules/taskmodules/common/mixins.py index 1e5d68822..0e2909893 100644 --- a/src/pie_modules/taskmodules/common/mixins.py +++ b/src/pie_modules/taskmodules/common/mixins.py @@ -7,7 +7,7 @@ import pandas as pd import torch import torch.nn.functional as F -from pytorch_ie.core import Annotation +from pie_core import Annotation from torch import Tensor logger = logging.getLogger(__name__) 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 76c832507..69f8e7668 100644 --- a/src/pie_modules/taskmodules/common/taskmodule_with_document_converter.py +++ b/src/pie_modules/taskmodules/common/taskmodule_with_document_converter.py @@ -1,12 +1,13 @@ from abc import ABC, abstractmethod from typing import Generic, Iterable, Iterator, Optional, Sequence, Type, TypeVar, Union -from pytorch_ie.core import Document, TaskModule -from pytorch_ie.core.taskmodule import ( +from pie_core import ( + Document, IterableTaskEncodingDataset, TaskEncoding, TaskEncodingDataset, TaskEncodingSequence, + TaskModule, ) from typing_extensions import TypeAlias diff --git a/src/pie_modules/taskmodules/cross_text_binary_coref.py b/src/pie_modules/taskmodules/cross_text_binary_coref.py index 45e87febf..6b42c8c5b 100644 --- a/src/pie_modules/taskmodules/cross_text_binary_coref.py +++ b/src/pie_modules/taskmodules/cross_text_binary_coref.py @@ -15,7 +15,7 @@ ) import torch -from pytorch_ie.core import Annotation, TaskEncoding, TaskModule +from pie_core import Annotation, TaskEncoding, TaskModule from pytorch_ie.utils.window import get_window_around_slice from torchmetrics import MetricCollection from torchmetrics.classification import ( diff --git a/src/pie_modules/taskmodules/extractive_question_answering.py b/src/pie_modules/taskmodules/extractive_question_answering.py index c1f3a9cf9..d774cf18b 100644 --- a/src/pie_modules/taskmodules/extractive_question_answering.py +++ b/src/pie_modules/taskmodules/extractive_question_answering.py @@ -4,7 +4,7 @@ import numpy as np import torch -from pytorch_ie.core import Annotation, AnnotationLayer, TaskEncoding, TaskModule +from pie_core import Annotation, AnnotationLayer, TaskEncoding, TaskModule from tokenizers import Encoding from transformers import AutoTokenizer, BatchEncoding, PreTrainedTokenizer from transformers.modeling_outputs import QuestionAnsweringModelOutput diff --git a/src/pie_modules/taskmodules/labeled_span_extraction_by_token_classification.py b/src/pie_modules/taskmodules/labeled_span_extraction_by_token_classification.py index 6f2d4f3f5..e3dfacb1c 100644 --- a/src/pie_modules/taskmodules/labeled_span_extraction_by_token_classification.py +++ b/src/pie_modules/taskmodules/labeled_span_extraction_by_token_classification.py @@ -24,7 +24,7 @@ ) import torch -from pytorch_ie.core import AnnotationLayer, TaskEncoding, TaskModule +from pie_core import AnnotationLayer, TaskEncoding, TaskModule from tokenizers import Encoding from torchmetrics import F1Score, Metric, MetricCollection, Precision, Recall from transformers import AutoTokenizer diff --git a/src/pie_modules/taskmodules/metrics/precision_recall_and_f1_for_labeled_annotations.py b/src/pie_modules/taskmodules/metrics/precision_recall_and_f1_for_labeled_annotations.py index 64e4de3fc..ca11d7f41 100644 --- a/src/pie_modules/taskmodules/metrics/precision_recall_and_f1_for_labeled_annotations.py +++ b/src/pie_modules/taskmodules/metrics/precision_recall_and_f1_for_labeled_annotations.py @@ -3,7 +3,7 @@ from typing import Any, Collection, Dict, Iterable, Optional, Union import torch -from pytorch_ie.core import Annotation +from pie_core import Annotation from torch import LongTensor from pie_modules.utils import flatten_dict diff --git a/src/pie_modules/taskmodules/pointer_network_for_end2end_re.py b/src/pie_modules/taskmodules/pointer_network_for_end2end_re.py index 1979145b8..5235bd1dc 100644 --- a/src/pie_modules/taskmodules/pointer_network_for_end2end_re.py +++ b/src/pie_modules/taskmodules/pointer_network_for_end2end_re.py @@ -18,18 +18,18 @@ ) import torch -from pytorch_ie.core import ( +from pie_core import ( Annotation, AnnotationLayer, Document, + TaskEncoding, TaskModule, ) -from pytorch_ie.core.taskmodule import ( +from pie_core.taskmodule import ( InputEncoding, ModelBatchOutput, TargetEncoding, TaskBatchEncoding, - TaskEncoding, ) from torchmetrics import Metric from transformers import AutoTokenizer, LogitsProcessorList, PreTrainedTokenizer diff --git a/src/pie_modules/taskmodules/re_span_pair_classification.py b/src/pie_modules/taskmodules/re_span_pair_classification.py index 1db76ae8a..ad259bc3d 100644 --- a/src/pie_modules/taskmodules/re_span_pair_classification.py +++ b/src/pie_modules/taskmodules/re_span_pair_classification.py @@ -27,7 +27,7 @@ import pandas as pd import torch -from pytorch_ie.core import ( +from pie_core import ( Annotation, AnnotationLayer, Document, diff --git a/src/pie_modules/taskmodules/re_text_classification_with_indices.py b/src/pie_modules/taskmodules/re_text_classification_with_indices.py index 990cadfbb..61fa41f25 100644 --- a/src/pie_modules/taskmodules/re_text_classification_with_indices.py +++ b/src/pie_modules/taskmodules/re_text_classification_with_indices.py @@ -27,7 +27,7 @@ import numpy as np import torch -from pytorch_ie.core import ( +from pie_core import ( Annotation, AnnotationLayer, Document, diff --git a/src/pie_modules/taskmodules/text_to_text.py b/src/pie_modules/taskmodules/text_to_text.py index fe61d4063..958b6118b 100644 --- a/src/pie_modules/taskmodules/text_to_text.py +++ b/src/pie_modules/taskmodules/text_to_text.py @@ -15,18 +15,18 @@ ) import torch -from pytorch_ie.core import ( +from pie_core import ( Annotation, AnnotationLayer, Document, + TaskEncoding, TaskModule, ) -from pytorch_ie.core.taskmodule import ( +from pie_core.taskmodule import ( InputEncoding, ModelBatchOutput, TargetEncoding, TaskBatchEncoding, - TaskEncoding, ) from torchmetrics import Metric from transformers import AutoTokenizer, PreTrainedTokenizer diff --git a/src/pie_modules/utils/hydra.py b/src/pie_modules/utils/hydra.py index ea98a4f45..395cd1266 100644 --- a/src/pie_modules/utils/hydra.py +++ b/src/pie_modules/utils/hydra.py @@ -1,7 +1,7 @@ from typing import Optional, Type, TypeVar, Union -from pytorch_ie.core import Document -from pytorch_ie.utils.hydra import resolve_target +from pie_core import Document +from pie_core.utils.hydra import resolve_target T = TypeVar("T", bound=Document) T_super = TypeVar("T_super", bound=Document) diff --git a/tests/conftest.py b/tests/conftest.py index 959fcf317..5a85cba64 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -3,7 +3,7 @@ import pkg_resources import pytest -from pytorch_ie.core import AnnotationLayer, annotation_field +from pie_core import AnnotationLayer, annotation_field from pie_modules.annotations import BinaryRelation, LabeledSpan, Span from pie_modules.documents import TextBasedDocument diff --git a/tests/document/processing/test_regex_partitioner.py b/tests/document/processing/test_regex_partitioner.py index a914b955c..4d9d7897e 100644 --- a/tests/document/processing/test_regex_partitioner.py +++ b/tests/document/processing/test_regex_partitioner.py @@ -4,7 +4,7 @@ from typing import Tuple import pytest -from pytorch_ie.core import AnnotationLayer, annotation_field +from pie_core import AnnotationLayer, annotation_field from pie_modules.annotations import LabeledSpan from pie_modules.document.processing import RegexPartitioner diff --git a/tests/document/processing/test_relation_argument_sorter.py b/tests/document/processing/test_relation_argument_sorter.py index 47104cc03..fec8e1200 100644 --- a/tests/document/processing/test_relation_argument_sorter.py +++ b/tests/document/processing/test_relation_argument_sorter.py @@ -2,7 +2,7 @@ import logging import pytest -from pytorch_ie.core import Annotation, AnnotationLayer, annotation_field +from pie_core import Annotation, AnnotationLayer, annotation_field from pie_modules.annotations import ( BinaryRelation, diff --git a/tests/document/processing/test_text_span_trimmer.py b/tests/document/processing/test_text_span_trimmer.py index 42a8506a3..10272d264 100644 --- a/tests/document/processing/test_text_span_trimmer.py +++ b/tests/document/processing/test_text_span_trimmer.py @@ -1,7 +1,7 @@ import dataclasses import pytest -from pytorch_ie.core import AnnotationLayer, annotation_field +from pie_core import AnnotationLayer, annotation_field from pie_modules.annotations import BinaryRelation, LabeledMultiSpan, LabeledSpan from pie_modules.document.processing import TextSpanTrimmer diff --git a/tests/document/processing/test_tokenization.py b/tests/document/processing/test_tokenization.py index 8466825aa..c940be177 100644 --- a/tests/document/processing/test_tokenization.py +++ b/tests/document/processing/test_tokenization.py @@ -3,7 +3,7 @@ from typing import Dict import pytest -from pytorch_ie.core import Annotation, AnnotationLayer, Document, annotation_field +from pie_core import Annotation, AnnotationLayer, Document, annotation_field from transformers import AutoTokenizer, PreTrainedTokenizer from pie_modules.annotations import ( diff --git a/tests/metrics/test_confusion_matrix.py b/tests/metrics/test_confusion_matrix.py index 2dba247d0..3122b8f26 100644 --- a/tests/metrics/test_confusion_matrix.py +++ b/tests/metrics/test_confusion_matrix.py @@ -4,7 +4,7 @@ from typing import Dict import pytest -from pytorch_ie.core import Annotation, AnnotationLayer, annotation_field +from pie_core import Annotation, AnnotationLayer, annotation_field from pie_modules.annotations import LabeledSpan from pie_modules.documents import TextBasedDocument diff --git a/tests/metrics/test_f1.py b/tests/metrics/test_f1.py index ebbf7f898..697edb317 100644 --- a/tests/metrics/test_f1.py +++ b/tests/metrics/test_f1.py @@ -1,7 +1,7 @@ from dataclasses import dataclass import pytest -from pytorch_ie.core import AnnotationLayer, annotation_field +from pie_core import AnnotationLayer, annotation_field from pie_modules.annotations import LabeledSpan from pie_modules.documents import TextBasedDocument diff --git a/tests/metrics/test_relation_argument_distance_collector.py b/tests/metrics/test_relation_argument_distance_collector.py index 85e6ef1a9..f51d6e819 100644 --- a/tests/metrics/test_relation_argument_distance_collector.py +++ b/tests/metrics/test_relation_argument_distance_collector.py @@ -1,7 +1,7 @@ import dataclasses import pytest -from pytorch_ie.core import Annotation, AnnotationLayer, Document, annotation_field +from pie_core import Annotation, AnnotationLayer, Document, annotation_field from pie_modules.annotations import BinaryRelation, LabeledSpan, NaryRelation from pie_modules.documents import TextBasedDocument, TokenBasedDocument diff --git a/tests/metrics/test_span_coverage_collector.py b/tests/metrics/test_span_coverage_collector.py index ff1e7d813..026ec28b7 100644 --- a/tests/metrics/test_span_coverage_collector.py +++ b/tests/metrics/test_span_coverage_collector.py @@ -1,7 +1,7 @@ import dataclasses import pytest -from pytorch_ie.core import Annotation, AnnotationLayer, Document, annotation_field +from pie_core import Annotation, AnnotationLayer, Document, annotation_field from pie_modules.annotations import LabeledMultiSpan, LabeledSpan from pie_modules.documents import TextBasedDocument, TokenBasedDocument diff --git a/tests/metrics/test_span_length_collector.py b/tests/metrics/test_span_length_collector.py index c8a5cd5d0..dcd69af01 100644 --- a/tests/metrics/test_span_length_collector.py +++ b/tests/metrics/test_span_length_collector.py @@ -1,7 +1,7 @@ import dataclasses import pytest -from pytorch_ie.core import AnnotationLayer, Document, annotation_field +from pie_core import AnnotationLayer, Document, annotation_field from pie_modules.annotations import Label, LabeledSpan from pie_modules.documents import TextBasedDocument, TokenBasedDocument diff --git a/tests/taskmodules/common/test_mixins.py b/tests/taskmodules/common/test_mixins.py index 715717bd0..b6b921bb2 100644 --- a/tests/taskmodules/common/test_mixins.py +++ b/tests/taskmodules/common/test_mixins.py @@ -3,7 +3,7 @@ from typing import List import torch -from pytorch_ie import Annotation +from pie_core import Annotation from pie_modules.taskmodules.common import BatchableMixin from pie_modules.taskmodules.common.mixins import RelationStatisticsMixin diff --git a/tests/taskmodules/common/test_taskmodule_with_document_converter.py b/tests/taskmodules/common/test_taskmodule_with_document_converter.py index ae0cf8fa9..8540b5a3f 100644 --- a/tests/taskmodules/common/test_taskmodule_with_document_converter.py +++ b/tests/taskmodules/common/test_taskmodule_with_document_converter.py @@ -1,7 +1,7 @@ from typing import Optional, Type import pytest -from pytorch_ie.core import Document +from pie_core import Document from typing_extensions import TypeAlias from pie_modules.documents import TextDocumentWithLabeledSpansAndBinaryRelations diff --git a/tests/taskmodules/test_extractive_question_answering.py b/tests/taskmodules/test_extractive_question_answering.py index a2ea392bf..d38e532d6 100644 --- a/tests/taskmodules/test_extractive_question_answering.py +++ b/tests/taskmodules/test_extractive_question_answering.py @@ -1,7 +1,7 @@ import pytest import torch import transformers -from pytorch_ie.core import AnnotationLayer +from pie_core import AnnotationLayer from pie_modules.annotations import ExtractiveAnswer, Question from pie_modules.documents import TextDocumentWithQuestionsAndExtractiveAnswers diff --git a/tests/taskmodules/test_labeled_span_extraction_by_token_classification.py b/tests/taskmodules/test_labeled_span_extraction_by_token_classification.py index 18218b4bf..b05374a9b 100644 --- a/tests/taskmodules/test_labeled_span_extraction_by_token_classification.py +++ b/tests/taskmodules/test_labeled_span_extraction_by_token_classification.py @@ -6,7 +6,7 @@ import pytest import torch -from pytorch_ie.core import AnnotationLayer, annotation_field +from pie_core import AnnotationLayer, annotation_field from torch import tensor from transformers import BatchEncoding diff --git a/tests/taskmodules/test_pointer_network_for_end2end_re.py b/tests/taskmodules/test_pointer_network_for_end2end_re.py index 8a274d5e1..5251b8ba8 100644 --- a/tests/taskmodules/test_pointer_network_for_end2end_re.py +++ b/tests/taskmodules/test_pointer_network_for_end2end_re.py @@ -5,7 +5,7 @@ import pytest import torch -from pytorch_ie.core import AnnotationLayer, Document, annotation_field +from pie_core import AnnotationLayer, Document, annotation_field from transformers import LogitsProcessorList from pie_modules.annotations import BinaryRelation, LabeledSpan diff --git a/tests/taskmodules/test_re_span_pair_classification.py b/tests/taskmodules/test_re_span_pair_classification.py index 4ff178390..de8c5e478 100644 --- a/tests/taskmodules/test_re_span_pair_classification.py +++ b/tests/taskmodules/test_re_span_pair_classification.py @@ -4,7 +4,7 @@ import pytest import torch -from pytorch_ie.core import AnnotationLayer, annotation_field +from pie_core import AnnotationLayer, annotation_field from torch import tensor from torchmetrics import Metric, MetricCollection diff --git a/tests/taskmodules/test_re_text_classification_with_indices.py b/tests/taskmodules/test_re_text_classification_with_indices.py index 2741c47d5..16bcfe83d 100644 --- a/tests/taskmodules/test_re_text_classification_with_indices.py +++ b/tests/taskmodules/test_re_text_classification_with_indices.py @@ -7,7 +7,7 @@ import pytest import torch -from pytorch_ie.core import ( +from pie_core import ( Annotation, AnnotationLayer, Document, diff --git a/tests/taskmodules/test_text2text.py b/tests/taskmodules/test_text2text.py index c37eff3d7..7dc779acb 100644 --- a/tests/taskmodules/test_text2text.py +++ b/tests/taskmodules/test_text2text.py @@ -3,7 +3,7 @@ import pytest import torch -from pytorch_ie.core import Annotation, TaskEncoding +from pie_core import Annotation, TaskEncoding from pie_modules.annotations import AbstractiveSummary from pie_modules.documents import ( diff --git a/tests/taskmodules/test_text2text_with_guidance.py b/tests/taskmodules/test_text2text_with_guidance.py index 0a4ae7a03..c66711610 100644 --- a/tests/taskmodules/test_text2text_with_guidance.py +++ b/tests/taskmodules/test_text2text_with_guidance.py @@ -2,7 +2,7 @@ import pytest import torch -from pytorch_ie.core import Annotation, TaskEncoding +from pie_core import Annotation, TaskEncoding from pie_modules.annotations import GenerativeAnswer, Question from pie_modules.documents import ( diff --git a/tests/test_annotations.py b/tests/test_annotations.py index ad365a64d..5ea2a0b85 100644 --- a/tests/test_annotations.py +++ b/tests/test_annotations.py @@ -2,7 +2,7 @@ from typing import Dict, Optional import pytest -from pytorch_ie.core import Annotation +from pie_core import Annotation from pie_modules.annotations import LabeledMultiSpan diff --git a/tests/utils/test_hydra.py b/tests/utils/test_hydra.py index 58bbff8c0..1d226a944 100644 --- a/tests/utils/test_hydra.py +++ b/tests/utils/test_hydra.py @@ -1,7 +1,7 @@ import dataclasses import pytest -from pytorch_ie.core import AnnotationLayer, annotation_field +from pie_core import AnnotationLayer, annotation_field from pie_modules.annotations import LabeledSpan, Span from pie_modules.documents import TextBasedDocument