Fix shape check

andreicuceu · May 20, 2024 · 1f06508 · 1f06508
1 parent 5e9a176
commit 1f06508
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vega/data.py b/vega/data.py
@@ -218,7 +218,7 @@ def _read_pk_ell(self, data_path, cuts_config, dmat_path, cov_path, cov_rescale=
 
         # Initialize the data coordinates and mask
         self._blind = False
-        self.data_coordinates = PkCoordinates(k_edges, num_ells=pk_ell.shape[1])
+        self.data_coordinates = PkCoordinates(k_edges, num_ells=data.shape[1])
         self.data_mask = self.data_coordinates.get_mask_scale_cuts(cuts_config).flatten()
         self.model_mask = self.data_mask