From 0635d822423a1fd18f06f691dd58b9a783d2b47d Mon Sep 17 00:00:00 2001
From: Hu Zhao <zhao@mbd.rwth-aachen.de>
Date: Mon, 31 Oct 2022 21:22:23 +0100
Subject: [PATCH] refactor: change np.atleast_2d to np.reshape
---
src/psimpy/simulator/run_simulator.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/psimpy/simulator/run_simulator.py b/src/psimpy/simulator/run_simulator.py
index 036d428..03d1cb4 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(
--
GitLab