Fix skipIfNoExtension (#849)

11b00d5c · moto · GitHub · 08a71271 · 11b00d5c
Unverified Commit 11b00d5c authored Aug 04, 2020 by moto Committed by GitHub Aug 04, 2020
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 4 deletions

test/common_utils/case_utils.py test/common_utils/case_utils.py +3 -4

No files found.
--- a/test/common_utils/case_utils.py
+++ b/test/common_utils/case_utils.py
@@ -70,9 +70,8 @@ skipIfNoCuda = unittest.skipIf(not torch.cuda.is_available(), reason='CUDA not a


 def skipIfNoExtension(test_item):
-    if (
-            not is_module_available('torchaudio._torchaudio')
-            and 'TORCHAUDIO_TEST_FAIL_IF_NO_EXTENSION' in os.environ
-    ):
+    if is_module_available('torchaudio._torchaudio'):
+        return test_item
+    if 'TORCHAUDIO_TEST_FAIL_IF_NO_EXTENSION' in os.environ:
        raise RuntimeError('torchaudio C++ extension is not available.')
    return unittest.skip('torchaudio C++ extension is not available')(test_item)