Minor fix decode_wav (#2340)

wenet-e2e · Feb 5, 2024 · 331c414 · 331c414
1 parent a6a7f30
commit 331c414
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/wenet/dataset/processor.py b/wenet/dataset/processor.py
@@ -136,9 +136,9 @@ def decode_wav(sample):
         with io.BytesIO(wav_file) as file_obj:
             waveform, sample_rate = torchaudio.load(file_obj)
             # del wav_file
-        del sample['wav']
-        sample['wav'] = waveform  # overwrite wav
-        sample['sample_rate'] = sample_rate
+    del sample['wav']
+    sample['wav'] = waveform  # overwrite wav
+    sample['sample_rate'] = sample_rate
     return sample