From fe852453c0cb80e05a0adc4a95ceb52873461851 Mon Sep 17 00:00:00 2001
From: Thomas Albers Raviola <thomas@thomaslabs.org>
Date: Tue, 4 Jun 2024 11:11:20 +0200
Subject: Modify substitution routines to return vectors

---
 solvers.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'solvers.py')

diff --git a/solvers.py b/solvers.py
index c7d421c..72f1b1d 100644
--- a/solvers.py
+++ b/solvers.py
@@ -64,7 +64,7 @@ def forward_substitution(
         x: solution of the system of equations
     '''
     nn = aa.shape[0]
-    xx = np.zeros((nn, 1), dtype=np.float_)
+    xx = np.zeros(nn, dtype=np.float_)
     for i in range(nn):
         xx[i] = (bb[i] - np.dot(aa[i, 0:i], xx[0:i])) / aa[i, i]
     return xx
@@ -83,7 +83,7 @@ def back_substitution(
         x: solution of the system of equations
     '''
     nn = aa.shape[0]
-    xx = np.zeros((nn, 1), dtype=np.float_)
+    xx = np.zeros(nn, dtype=np.float_)
     for i in range(nn - 1, -1, -1):
         xx[i] = (bb[i] - np.dot(aa[i, i:], xx[i:nn])) / aa[i, i]
     return xx
@@ -114,4 +114,4 @@ def gaussian_eliminate(
     y = forward_substitution(ll, pp @ bb)
     # U @ x = y
     x = back_substitution(uu, y)
-    return x[:, 0]
+    return x
-- 
cgit v1.2.3