Fix mc mock shape

andreicuceu · Nov 17, 2023 · d9571fb · d9571fb
1 parent ba3b038
commit d9571fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vega/output.py b/vega/output.py
@@ -455,8 +455,8 @@ def write_monte_carlo(self, cpu_id=None):
         mocks = self.analysis.mc_mocks
         columns = []
         for name in mocks.keys():
-            table = np.array(mocks[name]).T
-            columns.append(fits.Column(name=name, format=f'{table.shape[0]}D', array=table))
+            table = np.array(mocks[name])
+            columns.append(fits.Column(name=name, format=f'{table.shape[1]}D', array=table))
 
         mocks_hdu = fits.BinTableHDU.from_columns(columns)
         mocks_hdu.name = 'Mocks'