From 8732fa09584942f7c369c795fee657eccdb79102 Mon Sep 17 00:00:00 2001
From: kamilest <stankeviciute.kamile@gmail.com>
Date: Mon, 16 Dec 2024 12:51:28 -0800
Subject: [PATCH] Extract task for MIMIC-IV demo

---
 demo/{workflow.py => extract_task_mimic.py} | 27 +--------------------
 1 file changed, 1 insertion(+), 26 deletions(-)
 rename demo/{workflow.py => extract_task_mimic.py} (56%)

diff --git a/demo/workflow.py b/demo/extract_task_mimic.py
similarity index 56%
rename from demo/workflow.py
rename to demo/extract_task_mimic.py
index 28d034a..5f65642 100644
--- a/demo/workflow.py
+++ b/demo/extract_task_mimic.py
@@ -1,6 +1,6 @@
 #%%
 # MEDS_DEV_DEMO_CONDA_ENV = "meds-dev-demo"
-MEDS_DEV_DEMO_CONDA_ENV = "meds-dev"
+# MEDS_DEV_DEMO_CONDA_ENV = "meds-dev"
 
 #%%
 # !conda init bash
@@ -10,8 +10,6 @@
 #%% [markdown]
 # Install MEDS-DEV and dependencies
 
-#%%
-!pwd
 #%%
 !git clone https://github.com/mmcdermott/MEDS-DEV.git
 !cd ./MEDS-DEV && pip install -e .
@@ -19,12 +17,6 @@
 #%% [markdown]
 # TODO Install any model-related dependencies.
 
-#%% [markdown]
-# TODO Install MEDS Evaluation
-#%%
-# !git clone https://github.com/kamilest/meds-evaluation.git
-# !cd ./meds-evaluation && pip install -e .
-
 # %%
 MEDS_DEV_PROJECT_ROOT = "."
 DATASET_NAME = "MIMIC-IV"
@@ -37,20 +29,3 @@
 
 # %%
 !./MEDS-DEV/src/MEDS_DEV/helpers/extract_task.sh {MEDS_DEV_PROJECT_ROOT} {DATASET_NAME} {TASK_NAME}
-
-# %% [markdown]
-# ## Training MEDS-TAB
-
-# %%
-!pip uninstall es-aces -y # TODO ???
-!pip install meds-tab
-
-# %%
-MIMICIV_MEDS_DIR = "./content/meds/"
-OUTPUT_TABULARIZATION_DIR="./content/tabularized/"
-TASK_DIR="./content/tasks/"
-TASK_NAME="in_hospital_3d_los_after_48h"
-OUTPUT_MODEL_DIR="./content/output/meds_tab/"
-
-# %%
-!meds-tab-describe input_dir={MIMICIV_MEDS_DIR}/data output_dir={OUTPUT_TABULARIZATION_DIR}