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

pre-commit-ci[bot] · pre-commit-ci[bot] · commit 064caf79d236 · 2025-06-20T13:50:14.000Z
for more information, see https://pre-commit.ci
diff --git a/src/lightning/pytorch/plugins/precision/amp.py b/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

Original file line number	Diff line number	Diff line change
`@@ -113,9 +113,7 @@ def clip_gradients(`
`113`	`113`
`114`	`114`	`def autocast_context_manager(self) -> torch.autocast:`
`115`	`115`	`return torch.autocast(`
`116`		`- self.device,`
`117`		`- dtype=(torch.bfloat16 if self.precision == "bf16-mixed" else torch.half),`
`118`		`- cache_enabled=False`
	`116`	`+ self.device, dtype=(torch.bfloat16 if self.precision == "bf16-mixed" else torch.half), cache_enabled=False`
`119`	`117`	`)`
`120`	`118`
`121`	`119`	`@override`