convert prediction_time to timestamp when meds_format is enabled (#22)

knatarajan-lab · Dec 2, 2024 · 26a65a6 · 26a65a6
1 parent d30b1ac
commit 26a65a6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/cehrbert_data/cohorts/spark_app_base.py b/src/cehrbert_data/cohorts/spark_app_base.py
@@ -534,6 +534,8 @@ def build(self):
                 "index_date", "prediction_time"
             ).withColumnRenamed(
                 "label", "boolean_value"
+            ).withColumn(
+                "prediction_time", F.to_timestamp("prediction_time")
             ).withColumn(
                 "boolean_value", F.col("boolean_value").cast("boolean")
             )