fix(dayenu): check for flagged times when applying dayenu filter

radiocosmology · Dec 24, 2024 · eedceff · eedceff
1 parent 945187f
commit eedceff
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/draco/analysis/dayenu.py b/draco/analysis/dayenu.py
@@ -616,9 +616,17 @@ def process(self, hv, source):
         for tt in range(ntime):
             t0 = time.time()
 
+            # Skip fully masked times
+            flag = weight[..., tt] > 0.0
+            flag = np.all(flag, axis=0, keepdims=True)
+            weight[..., tt] *= flag.astype(weight.dtype)
+
             for xx in range(new):
                 self.log.debug(f"Filter time {tt} of {ntime}, baseline {xx} of {new}.")
 
+                if not np.any(flag[0, :, xx]):
+                    continue
+
                 for pp in range(npol):
                     # Grab datasets for this pol and ew baseline
                     tvis = np.ascontiguousarray(vis[pp, :, xx, :, tt])