Params changes.

mmcdermott · Aug 29, 2024 · 942613a · 942613a
1 parent 2f09440
commit 942613a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/MIMIC-IV_Example/configs/event_configs.yaml b/MIMIC-IV_Example/configs/event_configs.yaml
@@ -109,7 +109,7 @@ hosp/omr:
     time: col(chartdate)
     time_format: "%Y-%m-%d"
 
-hosp/subjects:
+hosp/patients:
   gender:
     code:
       - GENDER

diff --git a/MIMIC-IV_Example/slurm_runner.yaml b/MIMIC-IV_Example/slurm_runner.yaml
@@ -7,31 +7,32 @@ shard_events:
     launcher_params:
       timeout_min: 60
       cpus_per_task: 10
-      mem_gb: 25
+      mem_gb: 40
       partition: "short"
 
 split_and_shard_subjects:
   parallelize:
+    n_workers: 1
     launcher_params:
-      timeout_min: 60
+      timeout_min: 10
       cpus_per_task: 10
-      mem_gb: 10
+      mem_gb: 7
       partition: "short"
 
 convert_to_sharded_events:
   parallelize:
     launcher_params:
       timeout_min: 60
       cpus_per_task: 10
-      mem_gb: 10
+      mem_gb: 25
       partition: "short"
 
 merge_to_MEDS_cohort:
   parallelize:
     launcher_params:
       timeout_min: 60
       cpus_per_task: 10
-      mem_gb: 10
+      mem_gb: 25
       partition: "short"
 
 extract_code_metadata: