Skip to content

Commit d7641e0

Browse files
authored
fix: remove deprecated code_mapping, dev, refresh_cache from examples (#935)
The v2.0 MIMIC3Dataset/MIMIC4Dataset (based on BaseDataset) no longer accepts code_mapping, dev, or refresh_cache parameters. These were part of the legacy BaseEHRDataset API. Update README.rst, example scripts, and leaderboard utilities to use the current v2.0 API. Note: task file docstrings and pyhealth/datasets/mimicextract.py still reference code_mapping but are left for separate PRs since mimicextract.py has not yet been migrated to v2.0. Fixes #535
1 parent e857c91 commit d7641e0

5 files changed

Lines changed: 0 additions & 17 deletions

File tree

README.rst

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -189,8 +189,6 @@ Module 1: <pyhealth.datasets>
189189
root="https://storage.googleapis.com/pyhealth/Synthetic_MIMIC-III/",
190190
# raw CSV table name
191191
tables=["DIAGNOSES_ICD", "PROCEDURES_ICD", "PRESCRIPTIONS"],
192-
# map all NDC codes to CCS codes in these tables
193-
code_mapping={"NDC": "CCSCM"},
194192
)
195193
196194
.. image:: figure/structured-dataset.png

examples/drug_recommendation/drug_recommendation_mimic4_gamenet.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,6 @@ def prepare_drug_task_data():
2525
mimicvi = MIMIC4Dataset(
2626
root="/srv/local/data/physionet.org/files/mimiciv/2.0/hosp",
2727
tables=["diagnoses_icd", "procedures_icd", "prescriptions"],
28-
code_mapping={"NDC": ("ATC", {"target_kwargs": {"level": 3}})},
29-
dev=_DEV,
30-
refresh_cache=False,
3128
)
3229

3330
print("stat")

examples/mortality_prediction/mortality_mimic3_grasp.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,6 @@
99
base_dataset = MIMIC3Dataset(
1010
root="/srv/local/data/physionet.org/files/mimiciii/1.4",
1111
tables=["DIAGNOSES_ICD", "PROCEDURES_ICD", "PRESCRIPTIONS"],
12-
code_mapping={"ICD9CM": "CCSCM", "ICD9PROC": "CCSPROC", "NDC": "ATC"},
13-
dev=False,
14-
refresh_cache=False,
1512
)
1613
base_dataset.stat()
1714

examples/patient_linkage_mimic3_medlink.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,6 @@
2929
base_dataset = MIMIC3Dataset(
3030
root="/srv/local/data/physionet.org/files/mimiciii/1.4",
3131
tables=["DIAGNOSES_ICD"],
32-
code_mapping={"ICD9CM": ("CCSCM", {})},
33-
dev=False,
34-
refresh_cache=False,
3532
)
3633
base_dataset.stat()
3734

leaderboard/utils.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -64,9 +64,6 @@ def get_dataset(dataset_name):
6464
mimic3dataset = MIMIC3Dataset(
6565
root="/srv/local/data/physionet.org/files/mimiciii/1.4",
6666
tables=["DIAGNOSES_ICD", "PROCEDURES_ICD", "PRESCRIPTIONS"],
67-
dev=False,
68-
code_mapping={"NDC": "ATC"},
69-
refresh_cache=False,
7067
)
7168
dataset = mimic3dataset
7269

@@ -83,9 +80,6 @@ def get_dataset(dataset_name):
8380
mimic4dataset = MIMIC4Dataset(
8481
root="/srv/local/data/physionet.org/files/mimiciv/2.0/hosp",
8582
tables=["diagnoses_icd", "procedures_icd", "prescriptions"],
86-
dev=False,
87-
code_mapping={"NDC": "ATC"},
88-
refresh_cache=False,
8983
)
9084
dataset = mimic4dataset
9185

0 commit comments

Comments
 (0)