From d2afa4cc8be0e13474d655f51c42d6f67929071e Mon Sep 17 00:00:00 2001 From: Pablito Date: Wed, 20 Nov 2024 10:43:25 +0100 Subject: [PATCH] Changed the structure to correspond with PEP8 convention. --- linear_algebra/lanczos_algorithm.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/linear_algebra/lanczos_algorithm.py b/linear_algebra/lanczos_algorithm.py index 601e8841f..4d792d3b0 100644 --- a/linear_algebra/lanczos_algorithm.py +++ b/linear_algebra/lanczos_algorithm.py @@ -1,4 +1,6 @@ import numpy as np + + def lanczos(a: np.ndarray) -> tuple[list[float], list[float]]: """ Implements the Lanczos algorithm for a symmetric matrix. @@ -20,8 +22,8 @@ def lanczos(a: np.ndarray) -> tuple[list[float], list[float]]: rng = np.random.default_rng() v[:, 0] = rng.standard_normal(n) v[:, 0] /= np.linalg.norm(v[:, 0]) - alpha : list[float] = [] - beta : list[float] = [] + alpha: list[float] = [] + beta: list[float] = [] for j in range(n): w = np.dot(a, v[:, j]) alpha.append(np.dot(w, v[:, j])) @@ -32,4 +34,4 @@ def lanczos(a: np.ndarray) -> tuple[list[float], list[float]]: w -= beta[j - 1] * v[:, j - 1] beta.append(np.linalg.norm(w)) v[:, j + 1] = w / beta[j] - return alpha, beta \ No newline at end of file + return alpha, beta