Update sktalk/corpus/conversation.py

Co-authored-by: Ji Qi <[email protected]>
elpaco-escience · Nov 29, 2023 · 574a729 · 574a729
1 parent 9229e8a
commit 574a729
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/sktalk/corpus/conversation.py b/sktalk/corpus/conversation.py
@@ -92,11 +92,9 @@ def _subconversation_by_index(self,
             raise IndexError("Index out of range")
         if after is None:
             after = before
-        if index - before < 0:
-            before = index
-        if index + after + 1 > len(self._utterances):
-            after = len(self._utterances) - index - 1
-        returned_utterances = self._utterances[index-before:index+after+1]
+        left_bound = max(index-before, 0)
+        right_bound = min(index + after + 1, len(self._utterances))
+        returned_utterances = self._utterances[left_bound:right_bound]
         return Conversation(utterances=returned_utterances, suppress_warnings=True)
 
     def _subconversation_by_time(self,