Updated to MEDS v0.3.3

justin13601 · Aug 29, 2024 · ea38d40 · ea38d40
1 parent 504f48d
commit ea38d40
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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"),
 }