From 09e74dd9be959b72c12710d75201b5dc8213abcc Mon Sep 17 00:00:00 2001
From: Anton Khirnov <anton@khirnov.net>
Date: Wed, 24 Feb 2016 11:22:55 +0100
Subject: nonlin_ode: stop iterating on encountering a NaN

---
 nonlin_ode.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nonlin_ode.py b/nonlin_ode.py
index 1a145b9..b6d73d8 100644
--- a/nonlin_ode.py
+++ b/nonlin_ode.py
@@ -81,6 +81,8 @@ def nonlin_solve_1d(initial_guess, Fs, args, maxiter = 100, atol = 1e-14, grid =
         solution_new = _nonlin_solve_1d_iter(solution_old, grid, basis_vals, Fs, args)
 
         delta = np.max(np.abs(solution_new.coeffs - solution_old.coeffs))
+        if np.isnan(delta):
+            raise RuntimeError('nan')
         print delta
         if delta < atol:
             return solution_new
-- 
cgit v1.2.3