flagCosmicRays(): ensure additional extensions output when debugging …

…are simple ndarrays, not MaskedArrays
GeminiDRSoftware · Mar 1, 2023 · bc51afb · bc51afb
1 parent 1c943b2
commit bc51afb
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/geminidr/core/primitives_spect.py b/geminidr/core/primitives_spect.py
@@ -2583,8 +2583,6 @@ def flagCosmicRays(self, adinputs=None, **params):
 
                 # Set up the background and models to be blank initially:
                 background = np.zeros(ext.shape)
-                objfit = np.zeros(ext.shape)
-                skyfit = np.zeros(ext.shape)
 
                 # Fit the object spectrum:
                 if bkgmodel in ('both', 'object'):
@@ -2595,7 +2593,7 @@ def flagCosmicRays(self, adinputs=None, **params):
                                     weights=weights,
                                     **fit_1D_params).evaluate()
                 else:
-                    objfit = np.zeros_like(data)
+                    objfit = np.zeros(ext.shape)
                 if debug:
                     ext.OBJFIT = objfit.copy()
 
@@ -2614,7 +2612,7 @@ def flagCosmicRays(self, adinputs=None, **params):
                                     weights=weights,
                                     **fit_1D_params).evaluate()
                 else:
-                    skyfit = np.zeros_like(data)
+                    skyfit = np.zeros(ext.shape)
                 if debug:
                     ext.SKYFIT = skyfit