Update refspec (NVIDIA#5321)

Signed-off-by: smajumdar <[email protected]> Signed-off-by: smajumdar <[email protected]> Signed-off-by: Hainan Xu <[email protected]>
hainan-xv · Nov 29, 2022 · 76da91c · 76da91c
1 parent 7966e2d
commit 76da91c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/speech_recognition/convert_hf_dataset_to_nemo.py b/scripts/speech_recognition/convert_hf_dataset_to_nemo.py
@@ -208,7 +208,7 @@ def map_dataset_to_nemo(batch):
 
         soundfile.write(audio_filepath, batch['audio']['array'], samplerate=cfg.sampling_rate, format='wav')
 
-        batch['duration'] = librosa.get_duration(batch['audio']['array'], sr=batch['audio']['sampling_rate'])
+        batch['duration'] = librosa.get_duration(y=batch['audio']['array'], sr=batch['audio']['sampling_rate'])
         return batch
 
     return map_dataset_to_nemo