black

netket · Dec 20, 2021 · 25ba67c · 25ba67c
1 parent fc30101
commit 25ba67c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/netket/optimizer/qgt/qgt_onthefly.py b/netket/optimizer/qgt/qgt_onthefly.py
@@ -49,7 +49,7 @@ def check(x, target):
                 )
             )
 
-    jax.tree_multimap(check,x,target)
+    jax.tree_multimap(check, x, target)
 
 
 def QGTOnTheFly(vstate=None, **kwargs) -> "QGTOnTheFlyT":
@@ -182,7 +182,7 @@ def onthefly_mat_treevec(
         ravel_result = False
 
     check_valid_vector_type(vec, S._params)
-    
+
     vec = nkjax.tree_cast(vec, S._params)
 
     res = S._mat_vec(vec, S.diag_shift)
@@ -197,7 +197,7 @@ def onthefly_mat_treevec(
 def _solve(
     self: QGTOnTheFlyT, solve_fun, y: PyTree, *, x0: Optional[PyTree], **kwargs
 ) -> PyTree:
-    
+
     check_valid_vector_type(y, self._params)
 
     y = nkjax.tree_cast(y, self._params)