From 42d2d527e76d32f92cbedb395df922d1bd032dfe Mon Sep 17 00:00:00 2001 From: beverlylytle Date: Tue, 26 Nov 2024 18:38:57 +0200 Subject: [PATCH] update type suggestion --- thunder/torch/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/thunder/torch/__init__.py b/thunder/torch/__init__.py index c1da24f61..5ec568e02 100644 --- 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)