diff --git a/wsds/ws_audio.py b/wsds/ws_audio.py index ce1caa8..3720040 100644 --- a/wsds/ws_audio.py +++ b/wsds/ws_audio.py @@ -90,7 +90,7 @@ def _audio_to_mp3(samples): try: from torchcodec.encoders import AudioEncoder AudioEncoder(samples, sample_rate=int(samples.sample_rate)).to_file_like(out, "mp3") - except ImportError: + except (ImportError, RuntimeError): import torchaudio torchaudio.save(out, samples, int(samples.sample_rate), format="mp3") @@ -127,7 +127,7 @@ def get_reader(self, sample_rate=None): if self.reader is None or sample_rate_switch: try: from torchcodec.decoders import AudioDecoder - except ImportError: + except (ImportError, RuntimeError): AudioDecoder = CompatAudioDecoder reader = AudioDecoder(to_filelike(self.src), sample_rate=sample_rate)