Merge pull request #123 from ArneBinder/use_annotations_from_pytorch_ie

brat: use annotation types from pytorch-ie

pyproject.toml

@@ -24,7 +24,7 @@ classifiers = [
 
 [tool.poetry.dependencies]
 python = "^3.9"
-pytorch-ie = ">=0.29.4,<0.31.0"
+pytorch-ie = ">=0.30.3,<0.31.0"
 # <2.16.0 because of https://github.com/ArneBinder/pie-datasets/issues/93
 datasets = ">=2.14.0,<2.16.0"
 # this was manually added because we get a conflict with pyarrow otherwise

src/pie_datasets/builders/brat.py

@@ -4,9 +4,8 @@
 from typing import Any, Dict, List, Optional, Tuple, Union
 
 import datasets
-from pie_modules.annotations import BinaryRelation, LabeledMultiSpan, LabeledSpan
-from pytorch_ie import Document
-from pytorch_ie.core import Annotation, AnnotationList, annotation_field
+from pytorch_ie.annotations import BinaryRelation, LabeledMultiSpan, LabeledSpan
+from pytorch_ie.core import Annotation, AnnotationList, Document, annotation_field
 from pytorch_ie.documents import TextBasedDocument
 
 from pie_datasets import GeneratorBasedBuilder

tests/unit/builder/test_brat_builder.py

@@ -1,8 +1,8 @@
 from typing import Any
 
 import pytest
-from pie_modules.annotations import BinaryRelation, LabeledMultiSpan, LabeledSpan
-from pytorch_ie import Annotation
+from pytorch_ie.annotations import BinaryRelation, LabeledMultiSpan, LabeledSpan
+from pytorch_ie.core import Annotation
 from pytorch_ie.documents import TextBasedDocument
 
 from pie_datasets.builders.brat import BratAttribute, BratBuilder