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

pre-commit-ci[bot] · pre-commit-ci[bot] · commit 2cbe3c504fc7 · 2024-10-22T05:46:14.000Z
for more information, see https://pre-commit.ci
diff --git a/src/lightning/fabric/plugins/precision/amp.py b/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}.")
diff --git a/src/lightning/pytorch/plugins/precision/amp.py b/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}.")

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