From 68870236395435606a767c4c8c1a6b45f5c1ff64 Mon Sep 17 00:00:00 2001
From: Thomas Albers Raviola <thomas@thomaslabs.org>
Date: Tue, 4 Jun 2024 11:08:51 +0200
Subject: Add tests for LU routine

---
 solvers.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'solvers.py')

diff --git a/solvers.py b/solvers.py
index 040c6f4..c7d421c 100644
--- a/solvers.py
+++ b/solvers.py
@@ -6,7 +6,7 @@ from numpy.typing import NDArray
 def lu_factorization(
         mm: NDArray[np.float_],
         tolerance: float = 1e-6
-) -> NDArray[np.float_]:
+) -> tuple[NDArray[np.float_], NDArray[np.float_], NDArray[np.float_]]:
     '''Computes the LUP factorization of a matrix
 
     Args:
@@ -28,7 +28,7 @@ def lu_factorization(
     j = 0
 
     while i < nn and j < nn:
-        pivot_row = i + np.argmax(np.abs(uu[i:, j]))
+        pivot_row = i + int(np.argmax(np.abs(uu[i:, j])))
 
         if np.abs(uu[pivot_row, j]) < tolerance:
             j = j + 1
-- 
cgit v1.2.3