From 55eb1ccb905d64e72281e0eef99b6b3ac68444f8 Mon Sep 17 00:00:00 2001
From: John Wu <jhnwu3@gmail.com>
Date: Tue, 20 Jan 2026 18:43:39 -0600
Subject: [PATCH] init commit with additional visit ids for tables as
 attributes, although can be missing values in many cases

---
 pyhealth/datasets/configs/mimic4_ehr.yaml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pyhealth/datasets/configs/mimic4_ehr.yaml b/pyhealth/datasets/configs/mimic4_ehr.yaml
index 0a5f8c7a1..3dfb0e5e5 100644
--- a/pyhealth/datasets/configs/mimic4_ehr.yaml
+++ b/pyhealth/datasets/configs/mimic4_ehr.yaml
@@ -32,6 +32,7 @@ tables:
     patient_id: "subject_id"
     timestamp: "intime"
     attributes:
+      - "hadm_id"
       - "stay_id"
       - "first_careunit"
       - "last_careunit"
@@ -96,6 +97,7 @@ tables:
           - "category"
     timestamp: "charttime"
     attributes:
+      - "hadm_id"
       - "itemid"
       - "label"
       - "fluid"