From 255c2e7b7bf721296d4d4f22cb106ab3cf197e0a Mon Sep 17 00:00:00 2001
From: Thomas Albers Raviola <thomas@thomaslabs.org>
Date: Tue, 28 May 2024 11:23:02 +0200
Subject: Add unit test with pytest

---
 solvers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'solvers.py')

diff --git a/solvers.py b/solvers.py
index b544f1b..040c6f4 100644
--- a/solvers.py
+++ b/solvers.py
@@ -114,4 +114,4 @@ def gaussian_eliminate(
     y = forward_substitution(ll, pp @ bb)
     # U @ x = y
     x = back_substitution(uu, y)
-    return x
+    return x[:, 0]
-- 
cgit v1.2.3