refactor: readability of error propagation for covobs improved

This commit is contained in:
Fabian Joswig 2021-12-02 12:22:07 +00:00
parent 85a438fd46
commit ed47d50286

View file

@ -1144,10 +1144,7 @@ def derived_observable(func, data, **kwargs):
for j_obs, obs in np.ndenumerate(data):
for name in obs.names:
if name in obs.cov_names:
if name in new_grad:
new_grad[name] += deriv[i_val + j_obs] * obs.covobs[name].grad
else:
new_grad[name] = deriv[i_val + j_obs] * obs.covobs[name].grad
new_grad[name] = new_grad.get(name, 0) + deriv[i_val + j_obs] * obs.covobs[name].grad
else:
new_deltas[name] = new_deltas.get(name, 0) + deriv[i_val + j_obs] * _expand_deltas_for_merge(obs.deltas[name], obs.idl[name], obs.shape[name], new_idl_d[name])