From e494f937f6d270b8a8ded5aca40a84233a212fae Mon Sep 17 00:00:00 2001 From: Tom Donoghue Date: Mon, 2 Sep 2024 10:25:06 -0400 Subject: [PATCH] squeeze multi array outputs --- neurodsp/sim/multi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/neurodsp/sim/multi.py b/neurodsp/sim/multi.py index 792da25c..f0020866 100644 --- a/neurodsp/sim/multi.py +++ b/neurodsp/sim/multi.py @@ -162,7 +162,7 @@ def sim_across_values(sim_func, sim_params, n_sims, output='object'): sims.add_signals(sim_multiple(sim_func, cur_sim_params, n_sims, 'object')) if output == 'array': - sims = np.array([el.signals for el in sims]) + sims = np.squeeze(np.array([el.signals for el in sims])) return sims