diff --git a/superflexpy/utils/numerical_approximator.py b/superflexpy/utils/numerical_approximator.py index effa0af..54b70fa 100644 --- a/superflexpy/utils/numerical_approximator.py +++ b/superflexpy/utils/numerical_approximator.py @@ -171,7 +171,7 @@ def solve(self, fun, S0, **kwargs): ) ) - return np.array(output).reshape((-1, len(fun))) + return np.array(output).reshape((len(fun), -1)).T def get_fluxes(self, fluxes, S, S0, **kwargs): output = []