From 60639f44cf5196c4a23fe0ba696604190706174d Mon Sep 17 00:00:00 2001
From: George Darkwah <61522557+gdarkwah@users.noreply.github.com>
Date: Fri, 5 Jan 2024 15:08:06 -0800
Subject: [PATCH] Update numerical_approximator.py

---
 superflexpy/utils/numerical_approximator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 = []