From c2ad563e5f0750a4ad6bc6e17ac5f1be2a9eaad4 Mon Sep 17 00:00:00 2001
From: Arne Binder <arne.binder@dfki.de>
Date: Wed, 13 Nov 2024 15:41:28 +0100
Subject: [PATCH] fix encode_annotations

---
 src/pie_modules/taskmodules/pointer_network_for_end2end_re.py | 2 +-
 1 file changed, 1 insertion(+), 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
index 6decf4612..452003f28 100644
--- 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