🐛 Fix minor bug in create_input for regen

Signed-off-by: Marcos Martinez <[email protected]>
IBM · Feb 21, 2024 · b9a71c6 · b9a71c6
1 parent 7d4abf4
commit b9a71c6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/zshot/linker/linker_regen/utils.py b/zshot/linker/linker_regen/utils.py
@@ -21,6 +21,9 @@ def create_input(sentence, max_length, start_delimiter, end_delimiter):
         left_index = max(0, start_delimiter_index - half_context)
         right_index = min(len(sent_list),
                           end_delimiter_index + half_context + (half_context - (start_delimiter_index - left_index)))
+        if right_index == end_delimiter_index:
+            right_index += 1
+
         left_index = left_index - max(0, (half_context - (right_index - end_delimiter_index)))
         return " ".join(sent_list[left_index:right_index])