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

for more information, see https://pre-commit.ci
tad-mctc · Feb 19, 2024 · e414cb2 · e414cb2
1 parent 5a4966c
commit e414cb2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/test/test_grad/test_dedr.py b/test/test_grad/test_dedr.py
@@ -47,9 +47,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
 ]:
@@ -92,9 +90,7 @@ def test_gradgradcheck(dtype: torch.dtype, name: str) -> None:
     assert dgradgradcheck(func, diffvars, atol=tol, fast_mode=FAST_MODE)
 
 
-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
 ]:

diff --git a/test/test_grad/test_dqdr.py b/test/test_grad/test_dqdr.py
@@ -47,9 +47,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
 ]:
@@ -92,9 +90,7 @@ def test_gradgradcheck(dtype: torch.dtype, name: str) -> None:
     assert dgradgradcheck(func, diffvars, atol=tol, fast_mode=FAST_MODE)
 
 
-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
 ]: