Skip to content

Commit 57d5ead

Browse files
authored
Fix librosa deprecated usage (#5579)
Signed-off-by: Joaquin Anton Guirao <[email protected]>
1 parent 788ec4c commit 57d5ead

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dali/test/python/test_torch_pipeline_rnnt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,7 @@ def torch_mel_fbank(spectrogram, sample_rate, device="cpu", nfilt=64, lowfreq=0,
250250
spectrogram = spectrogram.cuda()
251251
n_fft = 2 * (spectrogram.shape[0] - 1)
252252
filterbanks = torch.tensor(
253-
librosa.filters.mel(sample_rate, n_fft, n_mels=nfilt, fmin=lowfreq, fmax=highfreq),
253+
librosa.filters.mel(sr=sample_rate, n_fft=n_fft, n_mels=nfilt, fmin=lowfreq, fmax=highfreq),
254254
dtype=torch.float,
255255
)
256256
if device == "gpu":

0 commit comments

Comments
 (0)