Skip to content

Commit ac50b55

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent a6ef20d commit ac50b55

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

tests/accelerators/test_accelerator_connector.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1042,17 +1042,17 @@ def test_accelerator_auto_choice_and_devices_gpu(is_gpu_available_mock, device_c
10421042
assert trainer.devices == 1
10431043
assert trainer.gpus == 1
10441044

1045-
1045+
10461046
@mock.patch("torch.cuda.is_available", return_value=False)
10471047
@mock.patch("pytorch_lightning.utilities.imports._TPU_AVAILABLE", return_value=False)
1048-
@mock.patch("pytorch_lightning.utilities.imports._IPU_AVAILABLE", return_value=False)
1048+
@mock.patch("pytorch_lightning.utilities.imports._IPU_AVAILABLE", return_value=False)
10491049
def test_devices_auto_choice_cpu(is_ipu_available_mock, is_tpu_available_mock, is_gpu_available_mock):
10501050
trainer = Trainer(accelerator="auto", devices="auto")
10511051
assert trainer.devices == 1
10521052
assert trainer.num_processes == 1
10531053

1054-
1055-
@mock.patch("torch.cuda.is_available", return_value=True)
1054+
1055+
@mock.patch("torch.cuda.is_available", return_value=True)
10561056
@mock.patch("torch.cuda.device_count", return_value=2)
10571057
def test_devices_auto_choice_gpu(is_gpu_available_mock, device_count_mock):
10581058
trainer = Trainer(accelerator="auto", devices="auto")

tests/accelerators/test_ipu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -605,7 +605,7 @@ def test_devices_auto_choice_ipu():
605605
assert trainer.devices == 4
606606
assert trainer.ipus == 4
607607

608-
608+
609609
@RunIf(ipu=True)
610610
def test_accelerator_auto_choice_and_devices_ipu():
611611
trainer = Trainer(accelerator="auto")

tests/accelerators/test_tpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -323,4 +323,4 @@ def test_devices_auto_choice_tpu():
323323
def test_accelerator_auto_choice_and_devices_tpu():
324324
trainer = Trainer(accelerator="auto")
325325
assert trainer.devices == 1
326-
assert trainer.tpu_cores == 1
326+
assert trainer.tpu_cores == 1

0 commit comments

Comments
 (0)