diff --git a/src/aces/__main__.py b/src/aces/__main__.py index 1345b37..9c97df7 100644 --- a/src/aces/__main__.py +++ b/src/aces/__main__.py @@ -45,6 +45,9 @@ def main(cfg: DictConfig): # query results result = query.query(task_cfg, predicates_df) + if cfg.data.standard.lower() == "meds": + result = result.rename(columns={"subject_id": "patient_id"}) + # save results to parquet os.makedirs(os.path.dirname(cfg.output_filepath), exist_ok=True) result.write_parquet(cfg.output_filepath, use_pyarrow=True)