From c082920c93514fea8fe168a8e9650bfafa3261cf Mon Sep 17 00:00:00 2001 From: Anton Khirnov Date: Wed, 24 May 2017 16:02:59 +0200 Subject: nonlin_ode: add a verbosity parameter --- nonlin_ode.py | 5 +++-- 1 file 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 -- cgit v1.2.3