Skip to content

Commit 635d485

Browse files
DN6sayakpaul
andauthored
Update tests/testing_utils.py
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
1 parent 3ceac24 commit 635d485

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/testing_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -334,7 +334,7 @@ def decorator(test_case):
334334
def require_torch_gpu(test_case):
335335
"""Decorator marking a test that requires CUDA and PyTorch."""
336336
return pytest.mark.skipif(
337-
not (is_torch_available() and torch_device == "cuda"), reason="test requires PyTorch+CUDA"
337+
torch_device != "cuda", reason="test requires PyTorch+CUDA"
338338
)(test_case)
339339

340340

0 commit comments

Comments
 (0)