KeyError: '7054' (Continuing with MIMIC3 question...)

/mnt/c/dev/mimic3-benchmarks$ python3 -m mimic3benchmark.scripts.create_phenotyping data/root/ data/phenotyping/
/mnt/c/dev/mimic3-benchmarks/mimic3benchmark/scripts/create_phenotyping.py:103: YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated, as the default Loader is unsafe. Please read https://msg.pyyaml.org/load for full details.
definitions = yaml.load(definitions_file)
Iterating over patients in test:   0%|                                                           | 0/15 [00:00<?, ?it/s]

Traceback (most recent call last):
  File "/usr/lib/python3.8/runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
  File "/usr/lib/python3.8/runpy.py", line 87, in _run_code
exec(code, run_globals)
  File "/mnt/c/dev/mimic3-benchmarks/mimic3benchmark/scripts/create_phenotyping.py", line 125, in <module>
main()
  File "/mnt/c/dev/mimic3-benchmarks/mimic3benchmark/scripts/create_phenotyping.py", line 120, in main
process_partition(args, definitions, code_to_group, id_to_group, group_to_id, "test")
  File "/mnt/c/dev/mimic3-benchmarks/mimic3benchmark/scripts/create_phenotyping.py", line 67, in process_partition
group = code_to_group[code]
KeyError: '7054'