style

lehner · Jan 22, 2025 · e58b75d · e58b75d
1 parent 482b781
commit e58b75d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/lib/gpt/algorithms/eigen/arnoldi.py b/lib/gpt/algorithms/eigen/arnoldi.py
@@ -123,7 +123,10 @@ def implicit_restart(self, H, evals, p):
         if self.verbose:
             g.message(f"Arnoldi: QR in {t1-t0} s")
 
-        r = g.eval(self.basis[k] * H[k, k - 1] + self.basis[-1] * self.H[-1][-1] * Q[n - 1, k - 1])
+        r = g.eval(
+            self.basis[k] * H[k, k - 1]
+            + self.basis[-1] * self.H[-1][-1] * Q[n - 1, k - 1]
+        )
         rn = g.norm2(r) ** 0.5
 
         t0 = g.time()
@@ -147,7 +150,7 @@ class arnoldi:
         Nstop=None,
         resid=None,
         implicit_restart=False,
-        sort_eigenvalues=None
+        sort_eigenvalues=None,
     )
     def __init__(self, params):
         self.params = params
@@ -161,7 +164,7 @@ def __call__(self, mat, src):
         # Nstop
         Nstop = self.params["Nstop"]
         sort_eigenvalues = self.params["sort_eigenvalues"]
-        
+
         # arnoldi base
         a = arnoldi_iteration(mat, src)
 

diff --git a/lib/gpt/algorithms/eigen/irl.py b/lib/gpt/algorithms/eigen/irl.py
@@ -39,7 +39,7 @@ class irl:
         betastp=None,
         maxiter=None,
         Nminres=None,
-        sort_eigenvalues=None
+        sort_eigenvalues=None,
     )
     def __init__(self, params):
         self.params = params
@@ -242,7 +242,8 @@ def step(self, mat, lmd, lme, evec, w, Nm, k):
 
             if verbose:
                 g.message(
-                    "%-65s %-45s" % ("alpha[ %d ] = %s" % (k, alph), "beta[ %d ] = %s" % (k, beta))
+                    "%-65s %-45s"
+                    % ("alpha[ %d ] = %s" % (k, alph), "beta[ %d ] = %s" % (k, beta))
                 )
 
         else:
@@ -275,7 +276,9 @@ def step(self, mat, lmd, lme, evec, w, Nm, k):
 
             t2 = g.time()
             if k > 0:
-                g.orthogonalize(w, evec[0:k], nblock=self.params["orthogonalize_nblock"])
+                g.orthogonalize(
+                    w, evec[0:k], nblock=self.params["orthogonalize_nblock"]
+                )
             t3 = g.time()
 
             ckpt.save([w, alph, beta])