From 293e73bc01b840c1f5a3c5e3f10bcc9e0c653357 Mon Sep 17 00:00:00 2001 From: Arnab Chakraborty Date: Sun, 8 Dec 2024 19:43:52 -0500 Subject: [PATCH] apply_time_freq_mask_bug a small bug in applying a time-freq mask to data --- draco/analysis/flagging.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/draco/analysis/flagging.py b/draco/analysis/flagging.py index 3e3512de..493aa8f5 100644 --- a/draco/analysis/flagging.py +++ b/draco/analysis/flagging.py @@ -2259,8 +2259,7 @@ def process(self, tstream, rfimask): sf = tstream.weight.local_offset[t_ax] ef = sf + tstream.weight.local_shape[t_ax] - m_ax = m_axes.index("freq") - bcast_slice[m_ax] = slice(sf, ef) + bcast_slice[t_ax] = slice(sf, ef) bcast_slice = tuple(bcast_slice) # Create output container by copying based on share parameter