Skip to content

Commit a9f084d

Browse files
committed
Fix mc shapes
1 parent 93e2092 commit a9f084d

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

vega/output.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -422,15 +422,15 @@ def write_monte_carlo(self, cpu_id=None):
422422
covariances = np.array(self.analysis.mc_covariances)
423423

424424
names = np.array(list(bestfits.keys()))
425-
bestfit_table = np.array([bestfits[name][:, 0] for name in names]).T
426-
errors_table = np.array([bestfits[name][:, 1] for name in names]).T
427-
covariances = covariances.reshape(bestfit_table.shape[0]*len(names), len(names))
425+
bestfit_table = np.array([bestfits[name][:, 0] for name in names])
426+
errors_table = np.array([bestfits[name][:, 1] for name in names])
427+
covariances = covariances.reshape(bestfit_table.shape[1]*len(names), len(names))
428428

429429
# Get the data types for the columns
430430
max_length = np.max([len(name) for name in names])
431431
name_format = str(max_length) + 'A'
432-
fit_format = f'{bestfit_table.shape[0]}D'
433-
cov_format = f'{covariances.shape[0]}D'
432+
fit_format = f'{bestfit_table.shape[1]}D'
433+
cov_format = f'{covariances.shape[1]}D'
434434

435435
# Create the columns with the bestfit data
436436
col1 = fits.Column(name='names', format=name_format, array=names)

0 commit comments

Comments
 (0)