From ea38d404856f35be4a703bf2116d4f04d72c76e7 Mon Sep 17 00:00:00 2001
From: Matthew McDermott <mattmcdermott8@gmail.com>
Date: Thu, 29 Aug 2024 08:43:03 -0400
Subject: [PATCH] Updated to MEDS v0.3.3

---
 pyproject.toml       | 2 +-
 src/aces/__main__.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 3d74d7b..0a3399d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,7 +26,7 @@ dependencies = [
     "pytimeparse == 1.1.*",
     "networkx == 3.3.*",
     "pyarrow == 17.*",
-    "meds == 0.3.2",
+    "meds == 0.3.3",
 ]
 
 [tool.setuptools_scm]
diff --git a/src/aces/__main__.py b/src/aces/__main__.py
index c41fdfc..eb91cdb 100644
--- a/src/aces/__main__.py
+++ b/src/aces/__main__.py
@@ -8,7 +8,7 @@
 import pyarrow as pa
 import pyarrow.parquet as pq
 from loguru import logger
-from meds import label_schema, subject_id_field
+from meds import label_schema, prediction_time_field, subject_id_field
 from omegaconf import DictConfig
 
 config_yaml = files("aces").joinpath("configs/aces.yaml")
@@ -28,7 +28,7 @@
     "integer_value": pl.Int64,
     "float_value": pl.Float64,
     "categorical_value": pl.String,
-    "prediction_time": pl.Datetime("us"),
+    prediction_time_field: pl.Datetime("us"),
 }