Fix masking for MC mocks

andreicuceu · Feb 8, 2024 · 2d059df · 2d059df
1 parent d885e67
commit 2d059df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vega/analysis.py b/vega/analysis.py
@@ -196,9 +196,9 @@ def create_global_monte_carlo(self, fiducial_model, scale=None, forecast=False):
         for name, data in self._data.items():
             mask = data.dist_model_coordinates.get_mask_to_other(data.data_coordinates)
             if data.data_mask.size == fiducial_model[name].size:
-                masked_fiducial.append(fiducial_model[name][data.data_mask])
-            elif mask.size == fiducial_model[name].size:
                 masked_fiducial.append(fiducial_model[name])
+            elif mask.size == fiducial_model[name].size:
+                masked_fiducial.append(fiducial_model[name][mask])
             else:
                 raise ValueError('Input fiducial has unknown size. '
                                  'It must match the data or the model.')