fix(RFIMask): inverse of RFI mask was applied

radiocosmology · Apr 30, 2020 · 80f55bf · 80f55bf
1 parent 3e873b9
commit 80f55bf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/draco/analysis/flagging.py b/draco/analysis/flagging.py
@@ -834,7 +834,7 @@ def process(self, tstream, rfimask):
         ef = sf + tstream.weight.local_shape[0]
 
         # Mask the data
-        tstream.weight[:] *= rfimask.mask[sf:ef][bcast_slice]
+        tstream.weight[:] *= (~rfimask.mask[sf:ef][bcast_slice]).astype(np.float32)
 
         return tstream
 

diff --git a/draco/core/containers.py b/draco/core/containers.py
@@ -894,6 +894,9 @@ def pol(self):
 
 class RFIMask(TODContainer):
     """A container for holding RFI mask.
+
+    The mask is `True` for contaminated samples that should be excluded, and
+    `False` for clean samples.
     """
 
     _axes = ("freq",)