From 42654ce6c0264fc4c31b652446d26e3e880a0f07 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 29 Jan 2024 20:42:51 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- test/test_grad/test_param.py | 8 ++------ test/test_grad/test_pos.py | 8 ++------ 2 files changed, 4 insertions(+), 12 deletions(-) diff --git a/test/test_grad/test_param.py b/test/test_grad/test_param.py index b3c9c57..a8815ed 100644 --- a/test/test_grad/test_param.py +++ b/test/test_grad/test_param.py @@ -36,9 +36,7 @@ device = None -def gradchecker( - dtype: torch.dtype, name: str -) -> tuple[ +def gradchecker(dtype: torch.dtype, name: str) -> tuple[ Callable[[Tensor, Tensor, Tensor, Tensor], Tensor], # autograd function tuple[Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor], ]: @@ -103,9 +101,7 @@ def test_gradgradcheck_slow(dtype: torch.dtype, name: str) -> None: assert dgradgradcheck(func, diffvars, atol=1e-5, rtol=1e-5, fast_mode=False) -def gradchecker_batch( - dtype: torch.dtype, name1: str, name2: str -) -> tuple[ +def gradchecker_batch(dtype: torch.dtype, name1: str, name2: str) -> tuple[ Callable[[Tensor, Tensor, Tensor, Tensor], Tensor], # autograd function tuple[Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor], ]: diff --git a/test/test_grad/test_pos.py b/test/test_grad/test_pos.py index 96f9938..a5bf2d8 100644 --- a/test/test_grad/test_pos.py +++ b/test/test_grad/test_pos.py @@ -36,9 +36,7 @@ tol = 1e-8 -def gradchecker( - dtype: torch.dtype, name: str -) -> tuple[ +def gradchecker(dtype: torch.dtype, name: str) -> tuple[ Callable[[Tensor], Tensor], # autograd function Tensor, # differentiable variables ]: @@ -91,9 +89,7 @@ def test_gradgradcheck(dtype: torch.dtype, name: str) -> None: assert dgradgradcheck(func, diffvars, atol=tol) -def gradchecker_batch( - dtype: torch.dtype, name1: str, name2: str -) -> tuple[ +def gradchecker_batch(dtype: torch.dtype, name1: str, name2: str) -> tuple[ Callable[[Tensor], Tensor], # autograd function Tensor, # differentiable variables ]: