Fix test backend initialization (#537)

21558490 · moto · GitHub · 9bdaefc8 · 21558490
Unverified Commit 21558490 authored Apr 13, 2020 by moto Committed by GitHub Apr 13, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test/common_utils.py test/common_utils.py +1 -1

No files found.
--- a/test/common_utils.py
+++ b/test/common_utils.py
@@ -76,7 +76,7 @@ def filter_backends_with_mp3(backends):
            with AudioBackendScope(backend):
                torchaudio.load(test_filepath)
            return True
-        except RuntimeError:
+        except (RuntimeError, ImportError):
            return False

    return [backend for backend in backends if supports_mp3(backend)]