Fix typo in new ApplyGenericMask task.

radiocosmology · Dec 11, 2024 · 34e92fd · 34e92fd
1 parent 4a167e6
commit 34e92fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/draco/analysis/flagging.py b/draco/analysis/flagging.py
@@ -2271,7 +2271,7 @@ def process(self, data: containers.ContainerBase, mask: containers.ContainerBase
         """
         # Pull out the axes of each dataset
         daxes = list(data.weight.attrs["axis"])
-        maxes = list(data.mask.attrs["axis"])
+        maxes = list(mask.mask.attrs["axis"])
 
         # Make sure all the mask axes exist in the data
         if any(ax not in daxes for ax in maxes):