[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/fabric/connector.py

@@ -464,9 +464,9 @@ def _check_and_init_precision(self) -> Precision:
             return DeepSpeedPrecision(self._precision_input)  # type: ignore
         if isinstance(self.strategy, FSDPStrategy):
             return FSDPPrecision(
-                precision=self._precision_input, # type: ignore[arg-type]
+                precision=self._precision_input,  # type: ignore[arg-type]
                 device=self._accelerator_flag.get_device() if isinstance(self._accelerator_flag, Accelerator) else None,
-            )  
+            )
         mp_precision_supported = ("32-true", "bf16-mixed", "bf16-true", "16-true")
         if isinstance(self.strategy, ModelParallelStrategy) and self._precision_input not in mp_precision_supported:
             raise ValueError(

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

@@ -53,7 +53,7 @@ def __init__(
             scaler = (
                 torch.amp.GradScaler(device=device)
                 if _TORCH_GREATER_EQUAL_2_4
-                else getattr(torch, device.split(':')[0] if device.split(':')[0] != "cpu" else "cuda").amp.GradScaler()
+                else getattr(torch, device.split(":")[0] if device.split(":")[0] != "cpu" else "cuda").amp.GradScaler()
             )
         if scaler is not None and self.precision == "bf16-mixed":
             raise ValueError(f"`precision='bf16-mixed'` does not use a scaler, found {scaler}.")

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

@@ -53,7 +53,7 @@ def __init__(
             scaler = (
                 torch.amp.GradScaler(device=device)
                 if _TORCH_GREATER_EQUAL_2_4
-                else getattr(torch, device.split(':')[0] if device.split(':')[0] != "cpu" else "cuda").amp.GradScaler()
+                else getattr(torch, device.split(":")[0] if device.split(":")[0] != "cpu" else "cuda").amp.GradScaler()
             )
         if scaler is not None and self.precision == "bf16-mixed":
             raise MisconfigurationException(f"`precision='bf16-mixed'` does not use a scaler, found {scaler}.")

src/lightning/pytorch/trainer/connectors/accelerator_connector.py

@@ -508,7 +508,7 @@ def _check_and_init_precision(self) -> Precision:
             return DeepSpeedPrecision(self._precision_flag)  # type: ignore[arg-type]
         if isinstance(self.strategy, FSDPStrategy):
             return FSDPPrecision(
-                precision=self._precision_flag, # type: ignore[arg-type]
+                precision=self._precision_flag,  # type: ignore[arg-type]
                 device=self._accelerator_flag.get_device() if isinstance(self._accelerator_flag, Accelerator) else None,
             )
         if self._precision_flag in ("16-true", "bf16-true"):