MNT: remove_nan and remove_neg work with lists

tomopy · Apr 5, 2015 · 6c4832e · 6c4832e
1 parent bc69272
commit 6c4832e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tomopy/io/data.py b/tomopy/io/data.py
@@ -209,6 +209,7 @@ def remove_neg(dat, val=0.):
     out : ndarray
        Corrected data.
     """
+    dat = as_float32(dat)
     dat[dat < 0.0] = val
     return dat
 
@@ -230,6 +231,7 @@ def remove_nan(dat, val=0.):
     out : ndarray
        Corrected data.
     """
+    dat = as_float32(dat)
     dat[np.isnan(dat)] = val
     return dat