Fix mask for fiducial model

andreicuceu · Nov 26, 2023 · 27cb9e4 · 27cb9e4
1 parent fbf6532
commit 27cb9e4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vega/data.py b/vega/data.py
@@ -601,7 +601,8 @@ def create_monte_carlo(self, fiducial_model, scale=None, seed=None, forecast=Fal
             self.mc_mock = np.full(self.full_data_size, np.nan)
             if self.cholesky_masked_cov:
                 ran_vec = np.random.randn(self.data_mask.sum())
-                self.mc_mock[self.data_mask] = masked_fiducial + self._cholesky.dot(ran_vec)
+                self.mc_mock[self.data_mask] = \
+                    masked_fiducial[self.data_mask] + self._cholesky.dot(ran_vec)
             else:
                 ran_vec = np.random.randn(self.full_data_size)
                 self.mc_mock = masked_fiducial + self._cholesky.dot(ran_vec)