From 4fa4f306c7c584bb50091793acc9a942a002c4a1 Mon Sep 17 00:00:00 2001 From: Krzysztof Rudnicki Date: Mon, 9 Dec 2024 22:38:15 +0100 Subject: [PATCH] hotfix: replaced vector norm calculations wiith cmath squrt function --- code/linear_algebra_utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/code/linear_algebra_utils.py b/code/linear_algebra_utils.py index ced277e4..a8739a5d 100644 --- a/code/linear_algebra_utils.py +++ b/code/linear_algebra_utils.py @@ -1,3 +1,4 @@ +import cmath import math import itertools import operator @@ -72,7 +73,9 @@ class SequentialLinearAlgebraUtils(ABC): @staticmethod def vector_norm(v): - return math.sqrt(sum(x*x for x in v)) + x_values = (x*x for x in v) + x_values_sum = sum(x_values) + return cmath.sqrt(x_values_sum).real @staticmethod def vector_scalar_divide(x, scalar):