OptimalExtract: handle variance input with nans

* handle similar to zeros: exclude those elements by modifying the mask
astropy · Sep 2, 2022 · 87eaad0 · 87eaad0
1 parent 7e7e4a6
commit 87eaad0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/specreduce/extract.py b/specreduce/extract.py
@@ -353,6 +353,10 @@ def __call__(self, image=None, trace_object=None,
         if np.any(variance < 0):
             raise ValueError("variance must be fully positive")
 
+        if np.any(np.isnan(variance)):
+            # exclude these elements by editing the input mask
+            img.mask[np.isnan(variance)] = True
+
         if np.any(variance == 0):
             # exclude these elements by editing the input mask
             img.mask[variance == 0] = True