Could use np.isfinite rather than np.isnan

[dvalters]

i.e. avoid statements like:

if not np.isnan(something):
   #...

if np.isfinite(something):
   #....

for readability