make magnus series tolerance a parameter

rwirth · Nov 7, 2018 · 71d32e8 · 71d32e8
1 parent 67df1d7
commit 71d32e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/imsrg/__init__.py b/imsrg/__init__.py
@@ -90,7 +90,7 @@ def rhs(s, y):
     return rhs
 
 
-def _magnus_rhs(generator, href0, reference, hole_levels=None):
+def _magnus_rhs(generator, href0, reference, hole_levels=None, rtol=1e-6):
     def rhs(s, y, href):
         omega = Operator.unpack(y, reference, symmetry='antihermitian')
         href = href0.bch(omega)
@@ -105,7 +105,7 @@ def rhs(s, y, href):
         termnorm0 = domegads.norm()
         termnorm = 1.
         m = 1
-        while termnorm > 1e-6:
+        while termnorm > rtol:
             domegads_term = omega.comm(domegads_term)
 
             coeff = next(coeffs)