Update pytorch ie 0.17.0 (#48)

* update pytorch-ie to 0.17.0

* use remove DatasetDict and import it from pytorch-ie

* remove conll2003_extract fixture data

* use target_layer property

Author: ArneBinder

setup.py

@@ -3,7 +3,7 @@
 from setuptools import find_packages, setup
 
 REQUIRED_PKGS = [
-    "pytorch-ie>=0.16.0,<1.0.0",
+    "pytorch-ie>=0.17.0,<1.0.0",
     "asciidag",  # for visualization
 ]
 

src/pie_utils/__init__.py

@@ -1 +0,0 @@
-from .dataset_dict import DatasetDict

src/pie_utils/dataset_dict.py

@@ -9,7 +9,6 @@
 from pytorch_ie import Dataset, IterableDataset
 from pytorch_ie.annotations import BinaryRelation
 from pytorch_ie.core import AnnotationList, Document
-from pytorch_ie.core.document import BaseAnnotationList
 from pytorch_ie.utils.span import is_contained_in
 
 from pie_utils.document.processors.common import EnterDatasetMixin, ExitDatasetMixin
@@ -20,14 +19,6 @@
 D = TypeVar("D", bound=Document)
 
 
-def target_layers(layer: BaseAnnotationList) -> dict[str, AnnotationList]:
-    return {
-        target_layer_name: layer._document[target_layer_name]
-        for target_layer_name in layer._targets
-        if target_layer_name in layer._document
-    }
-
-
 class CandidateRelationAdder(EnterDatasetMixin, ExitDatasetMixin):
     """CandidateRelationAdder adds binary relations to a document based on various parameters. It
     goes through combinations of available entity pairs as possible candidates for new relations.
@@ -149,13 +140,8 @@ def __call__(self, document: D) -> D:
             available_partitions = document[self.partition_layer]
         else:
             available_partitions = [None]
-        rel_target_layers = target_layers(layer=rel_layer)
-        if not len(rel_target_layers) == 1:
-            raise ValueError(
-                f"Relation layer must have exactly one target layer but found the following target layers: "
-                f"{list(rel_target_layers)}"
-            )
-        entity_layer = list(rel_target_layers.values())[0]
+
+        entity_layer = rel_layer.target_layer
         if self.use_predictions:
             entity_layer = entity_layer.predictions