fix example_to_document() / document_to_example(): use "attributions"…

… to access the attributes because the base dataset loader still produces this
ArneBinder · Nov 7, 2023 · 4a9c871 · 4a9c871
1 parent 55d3791
commit 4a9c871
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dataset_builders/pie/brat/brat.py b/dataset_builders/pie/brat/brat.py
@@ -130,7 +130,7 @@ def example_to_document(
 
     attribute_annotations: Dict[str, Dict[str, Attribute]] = defaultdict(dict)
     attribute_ids = []
-    for attribute_dict in dl2ld(example["attributes"]):
+    for attribute_dict in dl2ld(example["attributions"]):
         target_id = attribute_dict["target"]
         if target_id in spans:
             target_layer_name = "spans"
@@ -264,7 +264,7 @@ def document_to_example(
                 )
             attribute_dicts[attribute_annotation] = attribute_dict
 
-    example["attributes"] = ld2dl(
+    example["attributions"] = ld2dl(
         list(attribute_dicts.values()), keys=["id", "type", "target", "value"]
     )
     example["normalizations"] = ld2dl(