Remove pl_multi_process_test

Lightning-AI · Mar 14, 2022 · 3b357aa · 3b357aa
1 parent 55b29ac
commit 3b357aa
Showing 1 changed file with 15 additions and 16 deletions.
diff --git a/tests/accelerators/test_tpu.py b/tests/accelerators/test_tpu.py
@@ -313,19 +313,18 @@ def test_warning_if_tpus_not_used():
         Trainer()
 
 
-# @RunIf(tpu=True)
-# @pl_multi_process_test
-# @pytest.mark.parametrize(
-#     ["devices", "expected_device_ids"],
-#     [
-#         (1, [0]),
-#         (8, list(range(8))),
-#         ("8", list(range(8))),
-#         ([2], [2]),
-#         ("2,", [2]),
-#     ],
-# )
-# def test_trainer_config_device_ids(devices, expected_device_ids):
-#     trainer = Trainer(accelerator="tpu", devices=devices)
-#     assert trainer.device_ids == expected_device_ids
-#     assert trainer.num_devices == len(expected_device_ids)
+@RunIf(tpu=True)
+@pytest.mark.parametrize(
+    ["devices", "expected_device_ids"],
+    [
+        (1, [0]),
+        (8, list(range(8))),
+        ("8", list(range(8))),
+        ([2], [2]),
+        ("2,", [2]),
+    ],
+)
+def test_trainer_config_device_ids(devices, expected_device_ids):
+    trainer = Trainer(accelerator="tpu", devices=devices)
+    assert trainer.device_ids == expected_device_ids
+    assert trainer.num_devices == len(expected_device_ids)