Fix issue with librosa.load (#114)

* Add ability to handle length too small error from librosa * Bump for 0.17.2
joeweiss · Apr 24, 2024 · da7f966 · da7f966
1 parent aec4385
commit da7f966
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -11,7 +11,7 @@ exclude = [
 
 [project]
 name = "birdnetlib"
-version = "0.17.1"
+version = "0.17.2"
 authors = [
   { name="Joe Weiss", email="[email protected]" },
 ]

diff --git a/src/birdnetlib/utils.py b/src/birdnetlib/utils.py
@@ -30,14 +30,18 @@ def read_audio_segments(
     start_sample = 0
 
     while True:
-        audio_chunk, _ = librosa.load(
-            file_path,
-            sr=sr,
-            mono=True,
-            offset=start_sample / sr,
-            duration=chunk_duration,
-            res_type="kaiser_fast",
-        )
+        try:
+            audio_chunk, _ = librosa.load(
+                file_path,
+                sr=sr,
+                mono=True,
+                offset=start_sample / sr,
+                duration=chunk_duration,
+                res_type="kaiser_fast",
+            )
+        except ValueError:
+            # Specifically to catch "ValueError: Input signal length=0 is too small to resample"
+            break
 
         # Check if the chunk is empty, indicating the end of the file
         if not audio_chunk.any():