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