[Numpy-discussion] Stop program when getting a warning (invalid value encountered in sqrt)

[Scott Sinclair]

2010/11/9 Jóan Petur Petersen <joanpeturpetersen at gmail.com>:
> I have write quite a large program, but I occationally get the following
> warning message:
>
> Warning: invalid value encountered in sqrt
>
> Can I make the program stop instead of just giving a warning, so that I get
> a traceback to where it comes from?

I think np.seterr(invalid='raise') should do the trick.

Cheers,
Scott