ran black

cai4cai · Nov 30, 2022 · 4e5ae6d · 4e5ae6d
1 parent 80e5f8b
commit 4e5ae6d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/torchsparsegradutils/jax/jax_sparse_solve.py b/torchsparsegradutils/jax/jax_sparse_solve.py
@@ -16,8 +16,8 @@ def sparse_solve_j4t(A, B, solve=None, transpose_solve=None):
         if solve == None:
             solve = jax.scipy.sparse.linalg.bicgstab
         if transpose_solve == None:
-            transpose_solve = lambda A, B : jax.scipy.sparse.linalg.bicgstab(A.transpose(), B)
-            
+            transpose_solve = lambda A, B: jax.scipy.sparse.linalg.bicgstab(A.transpose(), B)
+
     return SparseSolveJ4T.apply(A, B, solve, transpose_solve)
 
 

diff --git a/torchsparsegradutils/sparse_solve.py b/torchsparsegradutils/sparse_solve.py
@@ -104,7 +104,7 @@ def sparse_generic_solve(A, B, solve=None, transpose_solve=None):
         if transpose_solve == None:
             # MINRES assumes A to be symmetric -> no need to transpose A
             transpose_solve = minres
-            
+
     return SparseGenericSolve.apply(A, B, solve, transpose_solve)