diff --git a/pyerrors/input/hadrons.py b/pyerrors/input/hadrons.py index e4671933..dc8429e1 100644 --- a/pyerrors/input/hadrons.py +++ b/pyerrors/input/hadrons.py @@ -172,9 +172,15 @@ def read_DistillationContraction_hd5(path, ens_id, diagrams=["direct"], idl=None # "DistillationContraction/Metadata/DmfSuffix" contains info about different quarks, irrelevant in the SU(3) case. for diagram in diagrams: + + if diagram == "triangle" and "Identity" not in str(identifier): + part = "im" + else: + part = "re" + real_data = np.zeros(Nt) for x0 in range(Nt): - raw_data = h5file["DistillationContraction/Correlators/" + diagram + "/" + str(x0)][:]["re"].astype(np.double) + raw_data = h5file["DistillationContraction/Correlators/" + diagram + "/" + str(x0)][:][part].astype(np.double) real_data += np.roll(raw_data, -x0) real_data /= Nt