fix test_brat_builder.py

ArneBinder · Apr 17, 2024 · 876a7f0 · 876a7f0
1 parent 3a313ab
commit 876a7f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/builder/test_brat_builder.py b/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