Update refspec (#5321)

Signed-off-by: smajumdar <[email protected]> Signed-off-by: smajumdar <[email protected]>
NVIDIA · Nov 4, 2022 · 2531d04 · 2531d04
1 parent 0fafc45
commit 2531d04
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