Skip to content

Commit d848d4d

Browse files
Lint fix
1 parent 6a07ffe commit d848d4d

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

tests/test_functional.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1418,7 +1418,11 @@ def test_gemv_eye_4bit(self, device, storage_type, dtype):
14181418
if device == "hpu" and not is_supported_on_hpu(storage_type, dtype):
14191419
pytest.skip("This configuration is not supported on HPU.")
14201420

1421-
if device == "cpu" and platform.system() == "Windows" and version.parse(torch.__version__).release == (2, 8, 0):
1421+
if (
1422+
device == "cpu"
1423+
and platform.system() == "Windows"
1424+
and version.parse(torch.__version__).release == (2, 8, 0)
1425+
):
14221426
pytest.skip("Regression: CPU crash on Windows with torch 2.8.0")
14231427

14241428
dims = 4

tests/test_optim.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,6 @@ def rm_path(path):
172172
@pytest.mark.parametrize("device", get_available_devices(no_cpu=True), ids=id_formatter("device"))
173173
@pytest.mark.skipif(not get_available_devices(no_cpu=True), reason="No device")
174174
def test_optimizer32bit(dim1, dim2, gtype, optim_name, device):
175-
176175
if device not in ["cuda", "xpu"]:
177176
pytest.skip("Optimizers are only supported on CUDA and XPU")
178177

@@ -317,7 +316,6 @@ def test_global_config(dim1, dim2, gtype, device):
317316
@pytest.mark.parametrize("device", get_available_devices(no_cpu=True))
318317
@pytest.mark.skipif(not get_available_devices(no_cpu=True), reason="No device")
319318
def test_optimizer8bit(dim1, dim2, gtype, optim_name, device):
320-
321319
if device not in ["cuda", "xpu"]:
322320
pytest.skip("8-bit optimizers are only supported on CUDA and XPU")
323321

0 commit comments

Comments
 (0)