From 27cb9e4524e4d6f3a566ed2514b0574b3268422e Mon Sep 17 00:00:00 2001 From: Andrei Cuceu Date: Sun, 26 Nov 2023 14:15:50 -0500 Subject: [PATCH] Fix mask for fiducial model --- vega/data.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vega/data.py b/vega/data.py index 244fe21..15b8604 100644 --- 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)