diff --git a/corrlib/input/openQCD.py b/corrlib/input/openQCD.py index da92645..71ebec6 100644 --- a/corrlib/input/openQCD.py +++ b/corrlib/input/openQCD.py @@ -190,6 +190,11 @@ def extract_t0(path: str, project: str, dir_in_project: str, param: dict[str, An Specific names for the replica of the ensemble the measurement file belongs to. files: list[str] Specific files to read. + + Returns + ------- + t0_dict: dict + Dictionary of t0 values in the pycorrlib style, with the parameters at hand. """ dataset = os.path.join(path, "projects", project) @@ -259,6 +264,11 @@ def extract_t1(path: str, project: str, dir_in_project: str, param: dict[str, An Specific names for the replica of the ensemble the measurement file belongs to. files: list[str] Specific files to read. + + Returns + ------- + t1_dict: dict + Dictionary of t1 values in the pycorrlib style, with the parameters at hand. """ directory = os.path.join(path, "projects", project, dir_in_project) @@ -289,7 +299,7 @@ def extract_t1(path: str, project: str, dir_in_project: str, param: dict[str, An for k in ["integrator", "eps", "ntot", "dnms"]: par_list.append(str(param[k])) pars = "/".join(par_list) - t0_dict: dict[str, Any] = {} - t0_dict[param["type"]] = {} - t0_dict[param["type"]][pars] = t0 - return t0_dict + t1_dict: dict[str, Any] = {} + t1_dict[param["type"]] = {} + t1_dict[param["type"]][pars] = t0 + return t1_dict