From 42f1e6e7fb6a0904cd2dfccbe654078e34776f34 Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Wed, 30 Nov 2022 16:36:51 +0000 Subject: [PATCH] refactor: matrix Obs to double cast simplified. --- pyerrors/correlators.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/pyerrors/correlators.py b/pyerrors/correlators.py index edf1117f..322f6043 100644 --- a/pyerrors/correlators.py +++ b/pyerrors/correlators.py @@ -323,14 +323,10 @@ class Corr: if sort == "Eigenvalue" and ts is not None: warnings.warn("ts has no effect when sorting by eigenvalue is chosen.", RuntimeWarning) all_vecs = [None] * (t0 + 1) + G0 = np.vectorize(lambda x: x.value)(symmetric_corr[t0]) for t in range(t0 + 1, self.T): try: - 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[t][i, j].value - + Gt = np.vectorize(lambda x: x.value)(symmetric_corr[t]) all_vecs.append(_GEVP_solver(Gt, G0)) except Exception: all_vecs.append(None)