diff options
Diffstat (limited to 'nonlin_ode.py')
| -rw-r--r-- | nonlin_ode.py | 5 |
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 |