fix encode_annotations

ArneBinder · Nov 13, 2024 · c2ad563 · c2ad563
1 parent e9acb06
commit c2ad563
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pie_modules/taskmodules/pointer_network_for_end2end_re.py b/src/pie_modules/taskmodules/pointer_network_for_end2end_re.py
@@ -497,7 +497,7 @@ def encode_annotations(
                     reversed_rel = BinaryRelation(
                         head=rel.tail,
                         tail=rel.head,
-                        label=rel.label + self.REVERSED_RELATION_LABEL_SUFFIX,
+                        label=reversed_label,
                     )
                     encoded_reversed_rel = self.relation_encoder_decoder.encode(
                         annotation=reversed_rel, metadata=metadata