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

fix: complex correlator fix extended to array valued complex

Browse source 
correlators.
This commit is contained in:
Fabian Joswig 2022-12-20 18:13:15 +01:00
parent f39a70c5a8
commit 6343968404
No known key found for this signature in database
2 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
pyerrors/correlators.py
View file

@ -1198,8 +1198,8 @@ class Corr:
@property
def real(self):
def return_real(obs_OR_cobs):
if isinstance(obs_OR_cobs[0], CObs):
return obs_OR_cobs[0].real
if isinstance(obs_OR_cobs.flatten()[0], CObs):
return np.vectorize(lambda x: x.real)(obs_OR_cobs)
else:
return obs_OR_cobs
@ -1208,8 +1208,8 @@ class Corr:
@property
def imag(self):
def return_imag(obs_OR_cobs):
if isinstance(obs_OR_cobs[0], CObs):
return obs_OR_cobs[0].imag
if isinstance(obs_OR_cobs.flatten()[0], CObs):
return np.vectorize(lambda x: x.imag)(obs_OR_cobs)
else:
return obs_OR_cobs * 0 # So it stays the right type