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

for more information, see https://pre-commit.ci
dftd4 · Jan 29, 2024 · 42654ce · 42654ce
1 parent f489ab8
commit 42654ce
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git 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
@@ -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
 ]: