Skip to content

Commit 596a97e

Browse files
Apply style fixes
1 parent 9306df9 commit 596a97e

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tests/testing_utils.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -333,9 +333,7 @@ def decorator(test_case):
333333

334334
def require_torch_gpu(test_case):
335335
"""Decorator marking a test that requires CUDA and PyTorch."""
336-
return pytest.mark.skipif(
337-
torch_device != "cuda", reason="test requires PyTorch+CUDA"
338-
)(test_case)
336+
return pytest.mark.skipif(torch_device != "cuda", reason="test requires PyTorch+CUDA")(test_case)
339337

340338

341339
def require_torch_cuda_compatibility(expected_compute_capability):
@@ -354,9 +352,7 @@ def decorator(test_case):
354352
# These decorators are for accelerator-specific behaviours that are not GPU-specific
355353
def require_torch_accelerator(test_case):
356354
"""Decorator marking a test that requires an accelerator backend and PyTorch."""
357-
return pytest.mark.skipif(
358-
torch_device == "cpu", reason="test requires accelerator+PyTorch"
359-
)(test_case)
355+
return pytest.mark.skipif(torch_device == "cpu", reason="test requires accelerator+PyTorch")(test_case)
360356

361357

362358
def require_torch_multi_gpu(test_case):

0 commit comments

Comments
 (0)