fix(sidereal): revert a change made in 5e74f33

This change had the unintended behaviour of masking large RA regions if
even a single baseline was corrupted.

draco/analysis/sidereal.py

@@ -641,12 +641,6 @@ def process(self, sstream: containers.SiderealStream) -> containers.SiderealStre
             if not np.any(weight[fi]):
                 continue
 
-            # We need to track the full time mask applied to each
-            # baseline. Because the `effective_ra` dataset is baseline
-            # dependent, the gradient could end up producing a slightly
-            # baseline dependent mask
-            fmask = np.zeros(weight.shape[-1], dtype=bool)
-
             for vi in range(vis.shape[1]):
                 # Skip if entire baseline is masked
                 if not np.any(weight[fi, vi]):
@@ -667,11 +661,7 @@ def process(self, sstream: containers.SiderealStream) -> containers.SiderealStre
                 sel = weight[fi, vi] > 0.0
                 vis[fi, vi] -= grad * sel * (era[fi, vi] - sstream.ra)
                 # Keep track of the time mask being applied
-                fmask |= ref_mask
-
-            # Zero any weights that could not be corrected for at least
-            # one baseline
-            weight[fi] *= (~fmask).astype(weight.dtype)[np.newaxis]
+                weight[fi, vi] *= (~ref_mask).astype(weight.dtype)
 
         # Delete the effective ra dataset since it is not needed anymore
         del sstream["effective_ra"]