Skip to content

GitLab

"sgl-kernel/git@developer.sourcefind.cn:change/sglang.git" did not exist on "4caca1ba0430a4ce4e96162c7d25cfebacb1153e"
Unverified Commit 11b00d5c authored by moto's avatar moto Committed by GitHub
Browse files

Fix skipIfNoExtension (#849)

parent 08a71271
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
test/common_utils/case_utils.py
View file @ 11b00d5c
...@@ -70,9 +70,8 @@ skipIfNoCuda = unittest.skipIf(not torch.cuda.is_available(), reason='CUDA not a ...@@ -70,9 +70,8 @@ skipIfNoCuda = unittest.skipIf(not torch.cuda.is_available(), reason='CUDA not a
def skipIfNoExtension(test_item): def skipIfNoExtension(test_item):
if ( if is_module_available('torchaudio._torchaudio'):
not is_module_available('torchaudio._torchaudio') return test_item
and 'TORCHAUDIO_TEST_FAIL_IF_NO_EXTENSION' in os.environ if 'TORCHAUDIO_TEST_FAIL_IF_NO_EXTENSION' in os.environ:
):
raise RuntimeError('torchaudio C++ extension is not available.') raise RuntimeError('torchaudio C++ extension is not available.')
return unittest.skip('torchaudio C++ extension is not available')(test_item) return unittest.skip('torchaudio C++ extension is not available')(test_item)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment