squeeze multi array outputs

neurodsp-tools · Sep 2, 2024 · e494f93 · e494f93
1 parent 9808874
commit e494f93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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