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

for more information, see https://pre-commit.ci
Lightning-AI · Apr 1, 2024 · 7d0a9bb · 7d0a9bb
1 parent a460236
commit 7d0a9bb
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/thunder/torch/__init__.py b/thunder/torch/__init__.py
@@ -1211,9 +1211,11 @@ def relu6(a: TensorProxy, /, inplace: bool = False) -> TensorLike:
 @torchsymbol(torch.nn.functional.hardswish, id="torch.hardswish", is_method=False)
 def hardswish(a: TensorProxy, /, inplace: bool = False) -> TensorLike:
     utils.check(not inplace, lambda: f"hardswish only supports inplace=False", exception_type=NotImplementedError)
-    utils.check(dtypes.is_float_dtype(a.dtype),
-                lambda: f"hardswish only supports floating point dtypes, got {a.dtype}",
-                exception_type=ValueError)
+    utils.check(
+        dtypes.is_float_dtype(a.dtype),
+        lambda: f"hardswish only supports floating point dtypes, got {a.dtype}",
+        exception_type=ValueError,
+    )
     return a * relu6(a + 3) / 6