diff --git a/src/psimpy/simulator/run_simulator.py b/src/psimpy/simulator/run_simulator.py index 036d4281f72636c09101392e940d578b370d8b8e..03d1cb43dacbe38b62a6b84a6a13f647ccf3b72d 100644 --- a/src/psimpy/simulator/run_simulator.py +++ b/src/psimpy/simulator/run_simulator.py @@ -167,8 +167,9 @@ class RunSimulator: Contains num_new_run dictionaries, each of which corresponds to kwargs of one simulation. prefixes : list of str - """ - var_samples = np.atleast_2d(var_samples) + """ + if var_samples.ndim == 0 or var_samples.ndim == 1: + var_samples = np.reshape(var_samples, (1, -1)) if var_samples.shape[1] != len(self.var_inp_parameter): raise ValueError(