jkuhl/corrlib
1
0
Fork 0
Code Issues 11 Pull requests Releases Packages Activity Actions

restructure: make code filter
Some checks failed
Pytest / pytest (3.12) (push) Successful in 1m18s
Details
Pytest / pytest (3.14) (push) Successful in 1m18s
Details
Ruff / ruff (push) Failing after 1m0s
Details
Mypy / mypy (push) Successful in 1m11s
Details
Pytest / pytest (3.13) (push) Successful in 1m13s
Details

Browse source
This commit is contained in:
Justus Kuhlmann 2026-03-27 11:53:39 +01:00
commit 1a1ac5121d
Signed by: jkuhl
GPG key ID: 00ED992DD79B85A6
1 changed files with 10 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

16
corrlib/find.py
View file

@ -276,6 +276,15 @@ def openQCD_filter(results:pd.DataFrame, **kwargs: Any) -> pd.DataFrame:
return results return results
def _code_filter(results: pd.DataFrame, code: str, **kwargs: Any) -> pd.DataFrame:
if code == "sfcf":
return sfcf_filter(results, **kwargs)
elif code == "openQCD":
return openQCD_filter(results, **kwargs)
else:
raise ValueError(f"Code {code} is not known.")
def find_record(path: Path, ensemble: str, correlator_name: str, code: str, project: Optional[str]=None, parameters: Optional[str]=None, def find_record(path: Path, ensemble: str, correlator_name: str, code: str, project: Optional[str]=None, parameters: Optional[str]=None,
created_before: Optional[str]=None, created_after: Optional[str]=None, updated_before: Optional[str]=None, updated_after: Optional[str]=None, created_before: Optional[str]=None, created_after: Optional[str]=None, updated_before: Optional[str]=None, updated_after: Optional[str]=None,
revision: Optional[str]=None, revision: Optional[str]=None,
@ -289,12 +298,7 @@ def find_record(path: Path, ensemble: str, correlator_name: str, code: str, proj
results = _db_lookup(db, ensemble, correlator_name,code, project, parameters=parameters) results = _db_lookup(db, ensemble, correlator_name,code, project, parameters=parameters)
if any([arg is not None for arg in [created_before, created_after, updated_before, updated_after]]): if any([arg is not None for arg in [created_before, created_after, updated_before, updated_after]]):
results = _time_filter(results, created_before, created_after, updated_before, updated_after) results = _time_filter(results, created_before, created_after, updated_before, updated_after)
if code == "sfcf": results = _code_filter(results, code, **kwargs)
results = sfcf_filter(results, **kwargs)
elif code == "openQCD":
results = openQCD_filter(results, **kwargs)
else:
raise ValueError(f"Code {code} is not known.")
if customFilter is not None: if customFilter is not None:
results = customFilter(results) results = customFilter(results)
print("Found " + str(len(results)) + " result" + ("s" if len(results)>1 else "")) print("Found " + str(len(results)) + " result" + ("s" if len(results)>1 else ""))