patch os environ

2021-02-26 01:55:07 +01:00 · 2021-02-26 01:55:07 +01:00 · 02ea55bbf6
parent c914724035
commit 02ea55bbf6
1 changed files with 1 additions and 0 deletions
--- a/tests/accelerators/test_accelerator_connector.py
+++ b/tests/accelerators/test_accelerator_connector.py
@ -385,6 +385,7 @@ def test_accelerator_choice_ddp_cpu_slurm(device_count_mock, setup_distributed_m
        trainer.fit(model)


+@mock.patch.dict(os.environ, {})
@mock.patch('torch.cuda.device_count', return_value=0)
@pytest.mark.parametrize("ddp_plugin_class", [DDPPlugin, DDPSpawnPlugin])
 def test_accelerator_choice_ddp_cpu_custom_plugin(_, ddp_plugin_class):