Make edits suggested by Carlos in Issue Keck-DataReductionPipelines#126

joshwalawender · May 31, 2019 · f896a68 · f896a68
1 parent 7bf8247
commit f896a68
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/MOSFIRE/Extract.py b/MOSFIRE/Extract.py
@@ -409,7 +409,7 @@ def iterate_spatial_profile(P, DmS, V, f,\
                                    weights=weights[~weights.mask])
             fit = np.array([fitted_poly(x) for x in xcoord])
             resid = (DmS[i]-f*srow)**2 / V[i]
-            newmask = (resid > sigma)
+            newmask = (resid > sigma**2)
 
             weights.mask = weights.mask | newmask
             srow.mask = srow.mask | newmask
@@ -434,7 +434,8 @@ def iterate_spatial_profile(P, DmS, V, f,\
             info('Row {:3d}: Reset {:d} negative pixels in fit to 0'.format(\
                   i, np.sum((fit<0))))
         fit[(fit < 0)] = 0
-        Pnew[i] = fit
+        fit_masked = np.ma.MaskedArray(data=fit, mask=weights.mask)
+        Pnew[i] = np.ma.filled(fit_masked, 0.0)
 
     return Pnew