Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

patch os environ

This commit is contained in:
Adrian Wälchli 2021-02-26 01:55:07 +01:00
parent c914724035
commit 02ea55bbf6
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/accelerators/test_accelerator_connector.py
View File

@ -385,6 +385,7 @@ def test_accelerator_choice_ddp_cpu_slurm(device_count_mock, setup_distributed_m
trainer.fit(model) trainer.fit(model)
@mock.patch.dict(os.environ, {})
@mock.patch('torch.cuda.device_count', return_value=0) @mock.patch('torch.cuda.device_count', return_value=0)
@pytest.mark.parametrize("ddp_plugin_class", [DDPPlugin, DDPSpawnPlugin]) @pytest.mark.parametrize("ddp_plugin_class", [DDPPlugin, DDPSpawnPlugin])
def test_accelerator_choice_ddp_cpu_custom_plugin(_, ddp_plugin_class): def test_accelerator_choice_ddp_cpu_custom_plugin(_, ddp_plugin_class):