From e22488f12c98301568da3ca9baa28ef526006d50 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 25 Dec 2024 18:22:35 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- thunder/torch/__init__.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/thunder/torch/__init__.py b/thunder/torch/__init__.py index a5fd520fa..e91ec078e 100644 --- a/thunder/torch/__init__.py +++ b/thunder/torch/__init__.py @@ -3143,18 +3143,15 @@ def amin(a, /, dim=None, keepdim: bool = False): # NOTE: Using name `torch_max` to avoid conflict with Python's `max` @overload -def torch_max(a: TensorLike, /) -> TensorLike: - ... +def torch_max(a: TensorLike, /) -> TensorLike: ... @overload -def torch_max(a: TensorLike, /, dim: NumberLike, keepdim: bool = False) -> tuple[TensorLike, TensorLike]: - ... +def torch_max(a: TensorLike, /, dim: NumberLike, keepdim: bool = False) -> tuple[TensorLike, TensorLike]: ... @overload -def torch_max(a: TensorLike, b: TensorLike, /) -> TensorLike: - ... +def torch_max(a: TensorLike, b: TensorLike, /) -> TensorLike: ... @torchsymbol(torch.max, is_method=True, method_name="max", id="torch.max")