update type suggestion

Lightning-AI · Nov 26, 2024 · 42d2d52 · 42d2d52
1 parent f8554db
commit 42d2d52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thunder/torch/__init__.py b/thunder/torch/__init__.py
@@ -1802,7 +1802,7 @@ def gelu(a: TensorProxy, /, *, approximate: str = "none") -> TensorLike:
 
 
 @torchsymbol(torch.nn.functional.leaky_relu, is_method=False)
-def leaky_relu(a: TensorProxy, /, negative_slope=0.01, inplace: bool = False) -> TensorLike:
+def leaky_relu(a: TensorProxy, /, negative_slope: float = 0.01, inplace: bool = False) -> TensorLike:
     out = where(a > 0, a, a * negative_slope)
     if inplace:
         return prims.copy_(out, a)