nonlin_ode: add a verbosity parameter

1 files changed, 3 insertions, 2 deletions

diff --git a/nonlin_ode.py b/nonlin_ode.py
index 106a449..dd10385 100644
--- a/nonlin_ode.py
+++ b/nonlin_ode.py
@@ -31,7 +31,7 @@ def _nonlin_solve_1d_iter(prev, grid, basis_vals, Fs, args):
 
     return series_expansion.SeriesExpansion(np.linalg.solve(mat, rhs), prev.basis)
 
-def nonlin_solve_1d(initial_guess, Fs, args, maxiter = 100, atol = 1e-14, grid = None):
+def nonlin_solve_1d(initial_guess, Fs, args, maxiter = 100, atol = 1e-14, grid = None, verbose = True):
     """
     Solve a non-linear ODE using a spectral method with Newton iteration.
 
@@ -85,7 +85,8 @@ def nonlin_solve_1d(initial_guess, Fs, args, maxiter = 100, atol = 1e-14, grid =
         delta = np.max(np.abs(solution_new.coeffs - solution_old.coeffs))
         if np.isnan(delta):
             raise RuntimeError('nan')
-        sys.stderr.write('delta: %g, coeffs[0]: %g\n' % (delta, solution_new.coeffs[0]))
+        if verbose:
+            sys.stderr.write('delta: %g, coeffs[0]: %g\n' % (delta, solution_new.coeffs[0]))
         if delta < atol:
             return solution_new
         solution_old = solution_new