Fix librosa deprecated usage (#5579)

Signed-off-by: Joaquin Anton Guirao <janton@nvidia.com>
NVIDIA · Jul 26, 2024 · 57d5ead · 57d5ead
1 parent 788ec4c
commit 57d5ead
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dali/test/python/test_torch_pipeline_rnnt.py b/dali/test/python/test_torch_pipeline_rnnt.py
@@ -250,7 +250,7 @@ def torch_mel_fbank(spectrogram, sample_rate, device="cpu", nfilt=64, lowfreq=0,
         spectrogram = spectrogram.cuda()
     n_fft = 2 * (spectrogram.shape[0] - 1)
     filterbanks = torch.tensor(
-        librosa.filters.mel(sample_rate, n_fft, n_mels=nfilt, fmin=lowfreq, fmax=highfreq),
+        librosa.filters.mel(sr=sample_rate, n_fft=n_fft, n_mels=nfilt, fmin=lowfreq, fmax=highfreq),
         dtype=torch.float,
     )
     if device == "gpu":