Restore test after #11448

Lightning-AI · Feb 18, 2022 · fc5211c · fc5211c
1 parent cf64f34
commit fc5211c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/trainer/flags/test_env_vars.py b/tests/trainer/flags/test_env_vars.py
@@ -45,12 +45,12 @@ def test_passing_env_variables_defaults():
     assert trainer.max_steps == 42
 
 
-@mock.patch.dict(os.environ, {"CUDA_VISIBLE_DEVICES": "0,1", "PL_TRAINER_GPUS": "2"})
+@mock.patch.dict(os.environ, {"CUDA_VISIBLE_DEVICES": "0,1", "PL_TRAINER_DEVICES": "2"})
 @mock.patch("torch.cuda.device_count", return_value=2)
 @mock.patch("torch.cuda.is_available", return_value=True)
 def test_passing_env_variables_devices(cuda_available_mock, device_count_mock):
     """Testing overwriting trainer arguments."""
     trainer = Trainer()
     assert trainer.devices == 2
     trainer = Trainer(accelerator="gpu", devices=1)
-    assert trainer.devices == 2
+    assert trainer.devices == 1