lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-11-30 13:06:53 +01:00
Code Issues Projects Releases Packages Wiki Activity

pyerrors/correlators.py: Allowing for None values in pe.Corr.prune

Browse source 
Closes: fjosw#271
This commit is contained in:
Ruaidhrí 2025-10-17 19:52:08 +01:00
commit 3c3fd70ce5
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
pyerrors/correlators.py
View file

@ -1405,13 +1405,15 @@ class Corr:
tmpmat = np.empty((Ntrunc, Ntrunc), dtype=object)
rmat = []
for t in range(basematrix.T):
for i in range(Ntrunc):
for j in range(Ntrunc):
tmpmat[i][j] = evecs[i].T @ self[t] @ evecs[j]
rmat.append(np.copy(tmpmat))
if self.content[t] is None:
rmat.append(None)
else:
for i in range(Ntrunc):
for j in range(Ntrunc):
tmpmat[i][j] = evecs[i].T @ self[t] @ evecs[j]
rmat.append(np.copy(tmpmat))
newcontent = [None if (self.content[t] is None) else rmat[t] for t in range(self.T)]
return Corr(newcontent)
return Corr(rmat)
def _sort_vectors(vec_set_in, ts):