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

fix: CObs can now be added and multiplied to as well as subtracted from

Browse source 
Obs in all combinations
This commit is contained in:
Fabian Joswig 2022-02-08 17:07:40 +00:00
parent f923ad06f7
commit f51503555b
2 changed files with 29 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

12
pyerrors/obs.py
View file

@ -693,7 +693,7 @@ class Obs:
else:
if isinstance(y, np.ndarray):
return np.array([self + o for o in y])
elif y.__class__.__name__ == 'Corr':
elif y.__class__.__name__ in ['Corr', 'CObs']:
return NotImplemented
else:
return derived_observable(lambda x, **kwargs: x[0] + y, [self], man_grad=[1])
@ -709,7 +709,7 @@ class Obs:
return np.array([self * o for o in y])
elif isinstance(y, complex):
return CObs(self * y.real, self * y.imag)
elif y.__class__.__name__ == 'Corr':
elif y.__class__.__name__ in ['Corr', 'CObs']:
return NotImplemented
else:
return derived_observable(lambda x, **kwargs: x[0] * y, [self], man_grad=[y])
@ -723,10 +723,8 @@ class Obs:
else:
if isinstance(y, np.ndarray):
return np.array([self - o for o in y])
elif y.__class__.__name__ == 'Corr':
elif y.__class__.__name__ in ['Corr', 'CObs']:
return NotImplemented
else:
return derived_observable(lambda x, **kwargs: x[0] - y, [self], man_grad=[1])
@ -742,7 +740,7 @@ class Obs:
else:
if isinstance(y, np.ndarray):
return np.array([self / o for o in y])
elif y.__class__.__name__ == 'Corr':
elif y.__class__.__name__ in ['Corr', 'CObs']:
return NotImplemented
else:
return derived_observable(lambda x, **kwargs: x[0] / y, [self], man_grad=[1 / y])
@ -753,7 +751,7 @@ class Obs:
else:
if isinstance(y, np.ndarray):
return np.array([o / self for o in y])
elif y.__class__.__name__ == 'Corr':
elif y.__class__.__name__ in ['Corr', 'CObs']:
return NotImplemented
else:
return derived_observable(lambda x, **kwargs: y / x[0], [self], man_grad=[-y / self.value ** 2])