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

for more information, see https://pre-commit.ci
Lightning-AI · May 30, 2023 · 0166e1f · 0166e1f
1 parent 5730dc1
commit 0166e1f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/lightning/fabric/plugins/precision/amp.py b/src/lightning/fabric/plugins/precision/amp.py
@@ -42,7 +42,9 @@ def __init__(
         scaler: Optional[torch.cuda.amp.GradScaler] = None,
     ) -> None:
         if precision not in ("16-mixed", "bf16-mixed"):
-            raise ValueError(f"Passed `{type(self).__name__}(precision={precision!r})`. Precision must be '16-mixed' or 'bf16-mixed'")
+            raise ValueError(
+                f"Passed `{type(self).__name__}(precision={precision!r})`. Precision must be '16-mixed' or 'bf16-mixed'"
+            )
 
         self.precision = cast(Literal["16-mixed", "bf16-mixed"], str(precision))
         if scaler is None and self.precision == "16-mixed":

diff --git a/src/lightning/pytorch/plugins/precision/amp.py b/src/lightning/pytorch/plugins/precision/amp.py
@@ -41,7 +41,9 @@ def __init__(
         scaler: Optional[torch.cuda.amp.GradScaler] = None,
     ) -> None:
         if precision not in ("16-mixed", "bf16-mixed"):
-            raise ValueError(f"`Passed `{type(self).__name__}(precision={precision!r})`. Precision must be '16-mixed' or 'bf16-mixed'")
+            raise ValueError(
+                f"`Passed `{type(self).__name__}(precision={precision!r})`. Precision must be '16-mixed' or 'bf16-mixed'"
+            )
 
         self.precision = cast(Literal["16-mixed", "bf16-mixed"], str(precision))
         if scaler is None and self.precision == "16-mixed":