From 8ebdb4c64276a04185f453d3a6a5431aebcbc027 Mon Sep 17 00:00:00 2001 From: Mike Tian-Jian Jiang Date: Sat, 11 Dec 2021 21:43:15 +0900 Subject: [PATCH] fix: to be compatible with the new key name "cardData" close #689 --- promptsource/utils.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/promptsource/utils.py b/promptsource/utils.py index d3d02eb50..d8b8af092 100644 --- a/promptsource/utils.py +++ b/promptsource/utils.py @@ -105,9 +105,12 @@ def filter_english_datasets(): english_datasets.append(dataset_name) continue - if "card_data" not in dataset: + metadata_key_set = {"card_data", "cardData"} & dataset.keys() + if not metadata_key_set: continue - metadata = dataset["card_data"] + if len(metadata_key_set) == 2: + raise KeyError(f"A dataset should never have both keys of {metadata_key_set}") + metadata = dataset[metadata_key_set.pop()] if "languages" not in metadata: continue