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

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/lightning/pytorch/plugins/precision/amp.py

@@ -113,9 +113,7 @@ def clip_gradients(
 
     def autocast_context_manager(self) -> torch.autocast:
         return torch.autocast(
-            self.device,
-            dtype=(torch.bfloat16 if self.precision == "bf16-mixed" else torch.half),
-            cache_enabled=False
+            self.device, dtype=(torch.bfloat16 if self.precision == "bf16-mixed" else torch.half), cache_enabled=False
         )
 
     @override

tests/tests_pytorch/plugins/precision/test_amp.py

@@ -14,15 +14,13 @@
 from unittest.mock import Mock
 
 import pytest
+import torch
+from torch import nn
 from torch.optim import Optimizer
 
 from lightning.pytorch.plugins import MixedPrecision
-from lightning.pytorch.utilities import GradClipAlgorithmType
-
-from torch import nn
-import torch
-
 from lightning.pytorch.plugins.precision import MixedPrecision
+from lightning.pytorch.utilities import GradClipAlgorithmType
 
 
 def test_clip_gradients():
@@ -62,7 +60,7 @@ def test_optimizer_amp_scaling_support_in_step_method():
 def test_amp_with_no_grad(precision: str):
     layer = nn.Linear(2, 1)
     x = torch.randn(1, 2)
-    amp = MixedPrecision(precision=precision, device='cpu')
+    amp = MixedPrecision(precision=precision, device="cpu")
 
     with amp.autocast_context_manager():
         with torch.no_grad():
@@ -72,4 +70,4 @@ def test_amp_with_no_grad(precision: str):
 
         loss.backward()
 
-        assert loss.grad_fn is not None
+        assert loss.grad_fn is not None