防止末尾空切片

Aaaou · Feb 14, 2023 · 0735473 · 0735473
1 parent 564d7cc
commit 0735473
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/fish_audio_preprocess/utils/slice_audio.py b/fish_audio_preprocess/utils/slice_audio.py
@@ -87,8 +87,9 @@ def slice_audio(
 
         arr.append(np.zeros(silent_shape, dtype=audio.dtype))
 
-    _gen = np.concatenate(arr)
-    yield from slice_by_max_duration(_gen, max_duration, rate)
+    if duration > 0:
+        _gen = np.concatenate(arr)
+        yield from slice_by_max_duration(_gen, max_duration, rate)
 
 
 def slice_audio_file(

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "fish-audio-preprocess"
-version = "0.1.8"
+version = "0.1.9"
 description = "Preprocess audio data"
 authors = ["Lengyue <[email protected]>"]
 license = "Apache"