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

pre-commit-ci[bot] · pre-commit-ci[bot] · commit 2ce6ee1a837c · 2024-01-26T11:43:15.000Z
for more information, see https://pre-commit.ci
diff --git a/tests/tests_fabric/plugins/precision/test_amp_integration.py b/tests/tests_fabric/plugins/precision/test_amp_integration.py
@@ -42,7 +42,8 @@ def forward(self, x):
 
 @pytest.mark.skipif(
     # https://github.com/pytorch/pytorch/issues/116056
-    sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2, reason="Windows + DDP issue in PyTorch 2.2"
+    sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,
+    reason="Windows + DDP issue in PyTorch 2.2",
 )
 @pytest.mark.parametrize(
     ("accelerator", "precision", "expected_dtype"),
diff --git a/tests/tests_fabric/strategies/launchers/test_multiprocessing_integration.py b/tests/tests_fabric/strategies/launchers/test_multiprocessing_integration.py
@@ -33,7 +33,8 @@ def __init__(self):
 
 @pytest.mark.skipif(
     # https://github.com/pytorch/pytorch/issues/116056
-    sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2, reason="Windows + DDP issue in PyTorch 2.2"
+    sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,
+    reason="Windows + DDP issue in PyTorch 2.2",
 )
 @pytest.mark.parametrize("strategy", ["ddp_spawn", pytest.param("ddp_fork", marks=RunIf(skip_windows=True))])
 def test_memory_sharing_disabled(strategy):
diff --git a/tests/tests_fabric/strategies/test_ddp_integration.py b/tests/tests_fabric/strategies/test_ddp_integration.py
@@ -20,18 +20,18 @@
 import pytest
 import torch
 from lightning.fabric import Fabric
-from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_0
+from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_0, _TORCH_GREATER_EQUAL_2_2
 from torch.nn.parallel.distributed import DistributedDataParallel
 
 from tests_fabric.helpers.runif import RunIf
 from tests_fabric.strategies.test_single_device import _run_test_clip_gradients
-from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_2
 from tests_fabric.test_fabric import BoringModel
 
 
 @pytest.mark.skipif(
     # https://github.com/pytorch/pytorch/issues/116056
-    sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2, reason="Windows + DDP issue in PyTorch 2.2"
+    sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,
+    reason="Windows + DDP issue in PyTorch 2.2",
 )
 @pytest.mark.parametrize(
     "accelerator",
diff --git a/tests/tests_fabric/utilities/test_distributed.py b/tests/tests_fabric/utilities/test_distributed.py
@@ -122,7 +122,8 @@ def test_collective_operations(devices, process):
 
 @pytest.mark.skipif(
     # https://github.com/pytorch/pytorch/issues/116056
-    sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,reason="Windows + DDP issue in PyTorch 2.2"
+    sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,
+    reason="Windows + DDP issue in PyTorch 2.2",
 )
 @pytest.mark.flaky(reruns=3)  # flaky with "process 0 terminated with signal SIGABRT" (GLOO)
 def test_is_shared_filesystem(tmp_path, monkeypatch):
diff --git a/tests/tests_fabric/utilities/test_spike.py b/tests/tests_fabric/utilities/test_spike.py
@@ -31,7 +31,8 @@ def spike_detection_test(fabric, global_rank_spike, spike_value, should_raise):
 
 @pytest.mark.skipif(
     # https://github.com/pytorch/pytorch/issues/116056
-    sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,reason="Windows + DDP issue in PyTorch 2.2"
+    sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,
+    reason="Windows + DDP issue in PyTorch 2.2",
 )
 @pytest.mark.flaky(max_runs=3)
 @pytest.mark.parametrize(

Original file line number	Diff line number	Diff line change
`@@ -42,7 +42,8 @@ def forward(self, x):`
`42`	`42`
`43`	`43`	`@pytest.mark.skipif(`
`44`	`44`	`# https://github.com/pytorch/pytorch/issues/116056`
`45`		`- sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2, reason="Windows + DDP issue in PyTorch 2.2"`
	`45`	`+ sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,`
	`46`	`+ reason="Windows + DDP issue in PyTorch 2.2",`
`46`	`47`	`)`
`47`	`48`	`@pytest.mark.parametrize(`
`48`	`49`	`("accelerator", "precision", "expected_dtype"),`
Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,8 @@ def __init__(self):`
`33`	`33`
`34`	`34`	`@pytest.mark.skipif(`
`35`	`35`	`# https://github.com/pytorch/pytorch/issues/116056`
`36`		`- sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2, reason="Windows + DDP issue in PyTorch 2.2"`
	`36`	`+ sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,`
	`37`	`+ reason="Windows + DDP issue in PyTorch 2.2",`
`37`	`38`	`)`
`38`	`39`	`@pytest.mark.parametrize("strategy", ["ddp_spawn", pytest.param("ddp_fork", marks=RunIf(skip_windows=True))])`
`39`	`40`	`def test_memory_sharing_disabled(strategy):`
Original file line number	Diff line number	Diff line change
`@@ -122,7 +122,8 @@ def test_collective_operations(devices, process):`
`122`	`122`
`123`	`123`	`@pytest.mark.skipif(`
`124`	`124`	`# https://github.com/pytorch/pytorch/issues/116056`
`125`		`- sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,reason="Windows + DDP issue in PyTorch 2.2"`
	`125`	`+ sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,`
	`126`	`+ reason="Windows + DDP issue in PyTorch 2.2",`
`126`	`127`	`)`
`127`	`128`	`@pytest.mark.flaky(reruns=3) # flaky with "process 0 terminated with signal SIGABRT" (GLOO)`
`128`	`129`	`def test_is_shared_filesystem(tmp_path, monkeypatch):`
Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,8 @@ def spike_detection_test(fabric, global_rank_spike, spike_value, should_raise):`
`31`	`31`
`32`	`32`	`@pytest.mark.skipif(`
`33`	`33`	`# https://github.com/pytorch/pytorch/issues/116056`
`34`		`- sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,reason="Windows + DDP issue in PyTorch 2.2"`
	`34`	`+ sys.platform == "win32" and _TORCH_GREATER_EQUAL_2_2,`
	`35`	`+ reason="Windows + DDP issue in PyTorch 2.2",`
`35`	`36`	`)`
`36`	`37`	`@pytest.mark.flaky(max_runs=3)`
`37`	`38`	`@pytest.mark.parametrize(`