Resolved one error with the target test set; found another.

justin13601 · Aug 24, 2024 · 60c84f9 · 60c84f9
1 parent 9dd5062
commit 60c84f9
Showing 1 changed file with 53 additions and 53 deletions.
diff --git a/tests/test_other_meds.py b/tests/test_other_meds.py
@@ -161,61 +161,61 @@ def parse_labels_yaml(yaml_str: str) -> dict[str, pl.DataFrame]:
 # Data (input)
 MEDS_SHARDS = parse_shards_yaml(
     """
-  "0": |-
+  "0": |-2
     patient_id,time,code,numeric_value,text_value
     1,,GENDER//MALE,,
     1,,SNP//rs234567,,
-    1,12/18/60 11:03,MEDS_BIRTH,,
-    1,8/2/72 10:00,CLINIC_VISIT,,
-    1,8/2/72 10:00,ICD9CM//493.90,,
-    1,8/2/72 10:00,LOINC//8310-5,0.65,
-    1,8/2/72 10:00,VITALS//BP//SYSTOLIC,108,
-    1,1/14/20 15:14,ADMISSION//MEDICAL,,
-    1,1/14/20 15:18,VITALS//BP//SYSTOLIC,132,
-    1,1/14/20 15:18,VITALS//BP//DIASTOLIC,90,
-    1,1/14/20 15:18,VITALS//HR//BPM,121,
-    1,1/14/20 15:18,VITALS//WEIGHT//LBS,233.2,
-    1,1/15/20 10:04,VITALS//BP//SYSTOLIC,126,
-    1,1/15/20 10:04,VITALS//BP//DIASTOLIC,91,
-    1,1/15/20 10:04,VITALS//HR//BPM,85,
-    1,1/16/20 10:11,VITALS//BP//SYSTOLIC,135,
-    1,1/16/20 10:11,VITALS//BP//DIASTOLIC,88,
-    1,1/16/20 10:11,VITALS//HR//BPM,79,
-    1,1/16/20 13:02,LVEF//ECHO,0.24,
-    1,1/17/20 10:00,ICD9CM//428.9,,
-    1,1/17/20 10:00,DISCHARGE//HOME,,
-    1,1/18/22 4:46,ADMISSION//MEDICAL,,
-    1,1/20/22 8:00,DISCHARGE//HOME_AMA,,
-    1,1/20/22 8:00,ICD9CM//428.41,,
-    1,1/20/22 8:00,ICD9CM//451.1,,
-    1,1/24/22 8:11,ADMISSION//ED,,
-    1,1/25/22 10:04,VITALS//BP//SYSTOLIC,168,
-    1,1/25/22 10:04,VITALS//BP//DIASTOLIC,100,
-    1,1/25/22 10:04,VITALS//HR//BPM,56,
-    1,2/27/22 1:13,ICD9CM//428.41,,
-    1,2/27/22 1:13,ICD9CM//410.1,,
-    1,2/27/22 1:13,DEATH,,
+    1,12/18/1960 11:03,MEDS_BIRTH,,
+    1,08/02/1972 10:00,CLINIC_VISIT,,
+    1,08/02/1972 10:00,ICD9CM//493.90,,
+    1,08/02/1972 10:00,LOINC//8310-5,0.65,
+    1,08/02/1972 10:00,VITALS//BP//SYSTOLIC,108,
+    1,01/14/2020 15:14,ADMISSION//MEDICAL,,
+    1,01/14/2020 15:18,VITALS//BP//SYSTOLIC,132,
+    1,01/14/2020 15:18,VITALS//BP//DIASTOLIC,90,
+    1,01/14/2020 15:18,VITALS//HR//BPM,121,
+    1,01/14/2020 15:18,VITALS//WEIGHT//LBS,233.2,
+    1,01/15/2020 10:04,VITALS//BP//SYSTOLIC,126,
+    1,01/15/2020 10:04,VITALS//BP//DIASTOLIC,91,
+    1,01/15/2020 10:04,VITALS//HR//BPM,85,
+    1,01/16/2020 10:11,VITALS//BP//SYSTOLIC,135,
+    1,01/16/2020 10:11,VITALS//BP//DIASTOLIC,88,
+    1,01/16/2020 10:11,VITALS//HR//BPM,79,
+    1,01/16/2020 13:02,LVEF//ECHO,0.24,
+    1,01/17/2020 10:00,ICD9CM//428.9,,
+    1,01/17/2020 10:00,DISCHARGE//HOME,,
+    1,01/18/2022 04:46,ADMISSION//MEDICAL,,
+    1,01/20/2022 08:00,DISCHARGE//HOME_AMA,,
+    1,01/20/2022 08:00,ICD9CM//428.41,,
+    1,01/20/2022 08:00,ICD9CM//451.1,,
+    1,01/24/2022 08:11,ADMISSION//ED,,
+    1,01/25/2022 10:04,VITALS//BP//SYSTOLIC,168,
+    1,01/25/2022 10:04,VITALS//BP//DIASTOLIC,100,
+    1,01/25/2022 10:04,VITALS//HR//BPM,56,
+    1,02/27/2022 01:13,ICD9CM//428.41,,
+    1,02/27/2022 01:13,ICD9CM//410.1,,
+    1,02/27/2022 01:13,DEATH,,
 
   "1": |-2
     patient_id,time,code,numeric_value,text_value
     3,,GENDER//FEMALE,,
     3,,SNP//rs2345291,,
     3,,SNP//rs228192,,
-    3,2/28/82 0:00,MEDS_BIRTH,,
-    3,1/14/20 15:14,ADMISSION//MEDICAL,,
-    3,1/14/20 15:18,VITALS//BP//SYSTOLIC,132,
-    3,1/14/20 15:18,VITALS//BP//DIASTOLIC,90,
-    3,1/14/20 15:18,VITALS//HR//BPM,121,
-    3,1/17/20 10:00,ICD9CM//V30.00,,
-    3,1/17/20 10:00,DISCHARGE//HOME,,
-    3,1/18/20 18:18,ADMISSION//MEDICAL,,
-    3,1/20/22 15:18,DISCHARGE//HOME,,
-    3,3/18/24 16:54,ICD9CM//428.9,,
-    3,3/18/24 17:11,ADMISSION//SURGICAL,,
-    3,3/28/24 10:00,DISCHARGE//HOME,,
-    3,3/29/24 11:00,ADMISSION//SURGICAL,,
-    3,4/19/24 13:32,DISCHARGE//HOME,,
-    3,5/22/24 0:00,ICD9CM//428.9,,
+    3,02/28/1982 00:00,MEDS_BIRTH,,
+    3,01/14/2020 15:14,ADMISSION//MEDICAL,,
+    3,01/14/2020 15:18,VITALS//BP//SYSTOLIC,132,
+    3,01/14/2020 15:18,VITALS//BP//DIASTOLIC,90,
+    3,01/14/2020 15:18,VITALS//HR//BPM,121,
+    3,01/17/2020 10:00,ICD9CM//V30.00,,
+    3,01/17/2020 10:00,DISCHARGE//HOME,,
+    3,01/18/2020 18:18,ADMISSION//MEDICAL,,
+    3,01/20/2022 15:18,DISCHARGE//HOME,,
+    3,03/18/2024 16:54,ICD9CM//428.9,,
+    3,03/18/2024 17:11,ADMISSION//SURGICAL,,
+    3,03/28/2024 10:00,DISCHARGE//HOME,,
+    3,03/29/2024 11:00,ADMISSION//SURGICAL,,
+    3,04/19/2024 13:32,DISCHARGE//HOME,,
+    3,05/22/2024 00:00,ICD9CM//428.9,,
     """,
     text_value=pl.Utf8,
 )
@@ -305,22 +305,22 @@ def parse_labels_yaml(yaml_str: str) -> dict[str, pl.DataFrame]:
         """
   "0": |-2
     patient_id,prediction_time,boolean_value,integer_value,float_value,categorical_value
-    1,1/15/20 15:14,0,,,
-    1,1/19/22 4:46,0,,,
-    1,1/25/22 8:11,1,,,
+    1,1/15/2020 15:14,0,,,
+    1,1/19/2022 4:46,0,,,
+    1,1/25/2022 8:11,1,,,
 
   "1": |-2
     patient_id,prediction_time,boolean_value,integer_value,float_value,categorical_value
-    3,3/19/24 16:54,0,,,
-    3,3/30/24 11:00,0,,,
+    3,3/19/2024 16:54,0,,,
+    3,3/30/2024 11:00,0,,,
     """
     ),
     "HF_derived_readmission": parse_labels_yaml(
         """
   "0": |-2
     patient_id,prediction_time,boolean_value,integer_value,float_value,categorical_value
-    1,1/17/20 10:00,0,,,
-    1,1/20/22 8:00,1,,,
+    1,1/17/2020 10:00,0,,,
+    1,1/20/2022 8:00,1,,,
 
   "1": |-2
     patient_id,prediction_time,boolean_value,integer_value,float_value,categorical_value