diff options
Diffstat (limited to 'solvers.py')
-rw-r--r-- | solvers.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -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 |