From d7649834d5f8953ca30441c4c5b7c346330915ad Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Wed, 30 Nov 2022 16:39:12 +0000 Subject: [PATCH] refactor: GEVP matrix cast also simplified for unsorted case. --- pyerrors/correlators.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/pyerrors/correlators.py b/pyerrors/correlators.py index 322f6043..dd0d51d6 100644 --- a/pyerrors/correlators.py +++ b/pyerrors/correlators.py @@ -311,12 +311,8 @@ class Corr: raise Exception("ts is required if sort=None.") if (self.content[t0] is None) or (self.content[ts] is None): raise Exception("Corr not defined at t0/ts.") - G0, Gt = np.empty([self.N, self.N], dtype="double"), np.empty([self.N, self.N], dtype="double") - for i in range(self.N): - for j in range(self.N): - G0[i, j] = symmetric_corr[t0][i, j].value - Gt[i, j] = symmetric_corr[ts][i, j].value - + G0 = np.vectorize(lambda x: x.value)(symmetric_corr[t0]) + Gt = np.vectorize(lambda x: x.value)(symmetric_corr[ts]) reordered_vecs = _GEVP_solver(Gt, G0) elif sort in ["Eigenvalue", "Eigenvector"]: