[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
Lightning-AI · Aug 3, 2024 · 75f7783 · 75f7783
1 parent 192ecce
commit 75f7783
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/tests/tests_fabric/utilities/test_optimizer.py b/tests/tests_fabric/utilities/test_optimizer.py
@@ -21,14 +21,20 @@
         torch.optim.Adamax,
     ],
 )
-@pytest.mark.parametrize("src_device", [
-    torch.device("cpu"),
-    pytest.param(torch.device("cuda"), marks=RunIf(min_cuda_gpus=1)),
-])
-@pytest.mark.parametrize("dst_device", [
-    torch.device("cpu"),
-    pytest.param(torch.device("cuda"), marks=RunIf(min_cuda_gpus=1)),
-])
+@pytest.mark.parametrize(
+    "src_device",
+    [
+        torch.device("cpu"),
+        pytest.param(torch.device("cuda"), marks=RunIf(min_cuda_gpus=1)),
+    ],
+)
+@pytest.mark.parametrize(
+    "dst_device",
+    [
+        torch.device("cpu"),
+        pytest.param(torch.device("cuda"), marks=RunIf(min_cuda_gpus=1)),
+    ],
+)
 def test_optimizer_to_device(optimizer_class, src_device, dst_device):
     # Optimizer with no state initialized
     model = torch.nn.Linear(2, 2, device=src_device)