Merge branch 'SapienzaNLP:main' into main

SapienzaNLP · Sep 24, 2024 · 351e91a · 351e91a
2 parents 4eecc0d + 48e9b54
commit 351e91a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/relik/reader/data/relik_reader_re_data.py b/relik/reader/data/relik_reader_re_data.py
@@ -730,7 +730,7 @@ def flip_cands(flip_candidates, candidates):
 
                     candidates_encoding_result = (
                         candidates_encoding_result[:i]
-                        + candidates_encoding_result[len(sample.span_candidates) : i]
+                        + candidates_encoding_result[len(sample.span_candidates) : len(sample.span_candidates) + i]
                     )
                     if len(self.special_symbols_types) > 0:
                         candidates_entities_symbols = candidates_entities_symbols[:i]