Add wave.Error in clause when loading audio

Sometimes this gets thrown
2019-05-01 01:03:42 -05:00 · 2019-05-01 01:03:42 -05:00 · d5441f4898
commit d5441f4898
parent d2d9c34931
1 changed files with 2 additions and 1 deletions
--- a/precise/util.py
+++ b/precise/util.py
@ -46,9 +46,10 @@ def load_audio(file: Any) -> np.ndarray:
        samples: Sample rate and audio samples from 0..1
    """
    import wavio
+    import wave
    try:
        wav = wavio.read(file)
-    except EOFError:
+    except (EOFError, wave.Error):
        wav = wavio.Wav(np.array([[]], dtype=np.int16), 16000, 2)
    if wav.data.dtype != np.int16:
        raise InvalidAudio('Unsupported data type: ' + str(wav.data.dtype))