lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-05-15 12:03:42 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix: get_item and reweighted now work with padded correlators.

Browse source
This commit is contained in:
Fabian Joswig 2021-12-20 11:35:04 +01:00
parent d2911afa31
commit bc220fceaf
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
pyerrors/correlators.py
View file

@ -65,14 +65,16 @@ class Corr:
def __getitem__(self, idx): def __getitem__(self, idx):
"""Return the content of timeslice idx""" """Return the content of timeslice idx"""
if len(self.content[idx]) == 1: if self.content[idx] is None:
return None
elif len(self.content[idx]) == 1:
return self.content[idx][0] return self.content[idx][0]
else: else:
return self.content[idx] return self.content[idx]
@property @property
def reweighted(self): def reweighted(self):
bool_array = np.array([list(map(lambda x: x.reweighted, o)) for o in self.content]) bool_array = np.array([list(map(lambda x: x.reweighted, o)) for o in list(filter(None.__ne__, self.content))])
if np.all(bool_array == 1): if np.all(bool_array == 1):
return True return True
elif np.all(bool_array == 0): elif np.all(bool_array == 0):