Add failsafe for MLS English dev set key alternate name as validation

This commit is contained in:
Kinan Martin 2025-06-11 09:18:28 +09:00
parent c8d932b0c2
commit c7c74b8658

View File

@ -74,9 +74,14 @@ def make_cutset_blueprints(
# Create dev dataset
logging.info("Creating dev cuts.")
cut_sets.append(
("dev", CutSet.from_huggingface_dataset(dataset["dev"], text_key="transcript"))
)
try:
cut_sets.append(
("dev", CutSet.from_huggingface_dataset(dataset["dev"], text_key="transcript"))
)
except KeyError:
cut_sets.append(
("dev", CutSet.from_huggingface_dataset(dataset["validation"], text_key="transcript"))
)
# Create train dataset
logging.info("Creating train cuts.")