fixed split_by_punctuation()

-fixed `.result.WhisperResult.split_by_punctuation()` not working if `min_words`/`min_chars`/`min_dur` are unspecified after df4a199 (#261)
jianfch · Dec 8, 2023 · d51edb6 · d51edb6
1 parent df4a199
commit d51edb6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/stable_whisper/_version.py b/stable_whisper/_version.py
@@ -1 +1 @@
-__version__ = "2.13.6"
+__version__ = "2.13.7"
diff --git a/stable_whisper/result.py b/stable_whisper/result.py
@@ -1191,10 +1191,10 @@ def _over_max(x: Segment):
                     (min_dur and x.duration >= min_dur)
             )
 
-        indices = set(s.id for s in self.segments if _over_max(s))
+        indices = set(s.id for s in self.segments if _over_max(s)) if any((min_words, min_chars, min_dur)) else None
 
         def _get_indices(x: Segment):
-            return x.get_punctuation_indices(punctuation) if x.id in indices else []
+            return x.get_punctuation_indices(punctuation) if indices is None or x.id in indices else []
 
         self._split_segments(_get_indices, lock=lock, newline=newline)
         if self._regroup_history: