small changes

flairNLP · Jan 18, 2024 · 3f3361a · 3f3361a
1 parent 897b3c2
commit 3f3361a
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/flair/models/sequence_tagger_model.py b/flair/models/sequence_tagger_model.py
@@ -1183,9 +1183,7 @@ def augment_sentence(
                     ]
                 else:
                     label_span = augmented_sentence[
-                        len_task_prompt
-                        + label.data_point.tokens[0].idx
-                        - 1 : len_task_prompt
+                        len_task_prompt + label.data_point.tokens[0].idx - 1 : len_task_prompt
                         + label.data_point.tokens[-1].idx
                     ]