added original_visit_id to the ehrshot output

knatarajan-lab · Jan 12, 2025 · 9a89ee6 · 9a89ee6
1 parent 6215db4
commit 9a89ee6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)