Skip to content

Commit 0a8ad3d

Browse files
committed
fix the tests for devicess on cpu macos
1 parent 7d1d814 commit 0a8ad3d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/tests_fabric/test_cli.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def test_run_env_vars_unsupported_strategy(strategy, fake_script):
8585
assert f"Invalid value for '--strategy': '{strategy}'" in ioerr.getvalue()
8686

8787

88-
@pytest.mark.parametrize("devices", ["1", "2", "0,", "1,0", "-1", "auto"])
88+
@pytest.mark.parametrize("devices", ["1", "2", "0,", "1,0", "-1"])
8989
@mock.patch.dict(os.environ, os.environ.copy(), clear=True)
9090
@mock.patch("lightning.fabric.accelerators.cuda.num_cuda_devices", return_value=2)
9191
def test_run_env_vars_devices_cuda(_, devices, monkeypatch, fake_script):
@@ -97,7 +97,7 @@ def test_run_env_vars_devices_cuda(_, devices, monkeypatch, fake_script):
9797

9898

9999
@RunIf(mps=True)
100-
@pytest.mark.parametrize("accelerator", ["mps", "gpu", "auto"])
100+
@pytest.mark.parametrize("accelerator", ["mps", "gpu"])
101101
@mock.patch.dict(os.environ, os.environ.copy(), clear=True)
102102
def test_run_env_vars_devices_mps(accelerator, monkeypatch, fake_script):
103103
monkeypatch.setitem(sys.modules, "torch.distributed.run", Mock())

0 commit comments

Comments
 (0)