Mocking Librosa in tests

.github/scripts/unittest-linux/install.sh

@@ -102,7 +102,13 @@ fi
     pip install kaldi-io SoundFile librosa coverage pytest pytest-cov scipy expecttest unidecode inflect Pillow sentencepiece pytorch-lightning 'protobuf<4.21.0' demucs tinytag pyroomacoustics flashlight-text git+https://github.com/kpu/kenlm
 
     # TODO: might be better to fix the single call to `pip install` above
-    pip install "pillow<10.0" "scipy<1.10" "numpy<2.0"
+    pip install pillow scipy numpy
+    echo
+    echo  ====================
+    echo HERE ARE ALL THE PACKAGES INSTALLED:
+    pip list
+    echo ====================
+
 )
 # Install fairseq
 git clone https://github.com/pytorch/fairseq

test/librosa_mock.py

@@ -0,0 +1,37 @@
+import re
+import os
+from pathlib import Path
+import torch
+
+def mock_function(f):
+    prefix = "torchaudio_unittest/assets/librosa_expected_results/"
+    def wrapper(request, *args, **kwargs):
+        if request is not None:
+            if os.path.exists(f"{prefix}{request}.pt"):
+                return torch.load(f"{prefix}{request}.pt", weights_only=False)
+        import librosa
+        result = eval(f)(*args, **kwargs)
+        if request is not None:
+            path = Path(f"{prefix}{request}.pt")
+            path.parent.mkdir(parents=True, exist_ok=True)
+            torch.save(result, path)
+        return result
+    return wrapper
+
+griffinlim = mock_function("librosa.griffinlim")
+
+mel = mock_function("librosa.filters.mel")
+
+power_to_db = mock_function("librosa.core.power_to_db")
+
+amplitude_to_db = mock_function("librosa.core.amplitude_to_db")
+
+phase_vocoder = mock_function("librosa.phase_vocoder")
+
+spectrogram = mock_function("librosa.core.spectrum._spectrogram")
+
+mel_spectrogram = mock_function("librosa.feature.melspectrogram")
+
+mfcc = mock_function("librosa.feature.mfcc")
+
+spectral_centroid = mock_function("librosa.feature.spectral_centroid")