lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-06-30 08:49:28 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix: array mode now works for elements with different covobs

Browse source
This commit is contained in:
Fabian Joswig 2021-12-07 07:29:05 +00:00
parent 674a1ea6f6
commit b0610544a8
2 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
pyerrors/obs.py
View file

@ -1148,6 +1148,13 @@ def derived_observable(func, data, array_mode=False, **kwargs):
# TODO: array mode does not work when matrices are defined on differenet ensembles
if array_mode is True:
class Zero_grad():
def __init__(self):
self.grad = 0
zero_grad = Zero_grad()
d_extracted = {}
g_extracted = {}
for name in new_sample_names:
@ -1158,7 +1165,7 @@ def derived_observable(func, data, array_mode=False, **kwargs):
for name in new_cov_names:
g_extracted[name] = []
for i_dat, dat in enumerate(data):
g_extracted[name].append(np.array([o.covobs[name].grad for o in dat.reshape(np.prod(dat.shape))]).reshape(dat.shape + (1, )))
g_extracted[name].append(np.array([o.covobs.get(name, zero_grad).grad for o in dat.reshape(np.prod(dat.shape))]).reshape(dat.shape + (1, )))
for i_val, new_val in np.ndenumerate(new_values):
new_deltas = {}