From 4a986dcff74602193bf600136e0825014c3a265a Mon Sep 17 00:00:00 2001 From: Tony Date: Mon, 4 May 2026 14:19:41 -0400 Subject: [PATCH] Accept both ptb/ptb_xl and mimic/mimic_iv as splitter keys --- src/upload_hf.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/upload_hf.py b/src/upload_hf.py index b98215b..90d8d86 100644 --- a/src/upload_hf.py +++ b/src/upload_hf.py @@ -11,9 +11,11 @@ class Splitter: + _MIMIC_EXTRACTOR = lambda p: max(re.findall(r"p\d+", p), key=len, default=None) + _PTB_EXTRACTOR = lambda p: (m := re.search(r"_([0-9]+)_hr", p)) and m.group(1) _PATIENT_EXTRACTORS = { - "mimic": lambda p: max(re.findall(r"p\d+", p), key=len, default=None), - "ptb": lambda p: (m := re.search(r"_([0-9]+)_hr", p)) and m.group(1), + "mimic": _MIMIC_EXTRACTOR, "mimic_iv": _MIMIC_EXTRACTOR, + "ptb": _PTB_EXTRACTOR, "ptb_xl": _PTB_EXTRACTOR, } @staticmethod