clean: quiet librosa load warning

jaswon · Dec 2, 2022 · f0fb5e9 · f0fb5e9
1 parent 914671d
commit f0fb5e9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/osu_dreamer/model/data.py b/osu_dreamer/model/data.py
@@ -1,5 +1,6 @@
 import os
 import random
+import warnings
 
 from pathlib import Path
 from functools import partial
@@ -46,7 +47,9 @@ def reclaim_memory():
         pass
 
 def load_audio(audio_file):
-    wave, sr = librosa.load(audio_file)
+    with warnings.catch_warnings():
+        warnings.simplefilter("ignore")
+        wave, sr = librosa.load(audio_file)
 
     # compute spectrogram
     spec: "A,L" = torchaudio.transforms.MFCC(