From 9a89ee6d6cf57cf02482b3b1068961fec8be1c19 Mon Sep 17 00:00:00 2001 From: Chao Pang Date: Sat, 11 Jan 2025 19:03:59 -0500 Subject: [PATCH] added original_visit_id to the ehrshot output --- src/cehrbert_data/tools/ehrshot_to_omop.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/cehrbert_data/tools/ehrshot_to_omop.py b/src/cehrbert_data/tools/ehrshot_to_omop.py index ac456e8..df45d4b 100644 --- a/src/cehrbert_data/tools/ehrshot_to_omop.py +++ b/src/cehrbert_data/tools/ehrshot_to_omop.py @@ -848,6 +848,7 @@ def main(args): f.col("ehr.unit").alias("unit"), f.col("ehr.omop_table").alias("omop_table"), f.col("ehr.visit_id").alias("visit_id"), + f.col("ehr.original_visit_id").alias("original_visit_id"), ) ehr_shot_data.write.mode("overwrite").parquet(ehr_shot_path)