From 9bd8e4bda7345ed6705f0f001f9173ec4d8f3525 Mon Sep 17 00:00:00 2001 From: juberti Date: Thu, 17 Oct 2024 16:02:12 -0700 Subject: [PATCH] Fix bug --- ultravox/data/datasets.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/ultravox/data/datasets.py b/ultravox/data/datasets.py index 0968fc6..51fdd4e 100644 --- a/ultravox/data/datasets.py +++ b/ultravox/data/datasets.py @@ -474,6 +474,9 @@ def __init__(self, args: VoiceDatasetArgs, config: DatasetConfig) -> None: ) dsets.append(ds) total_samples += split.num_samples + assert ( + len(dsets) > 0 + ), f"The {config.name} dataset has no {self._args.split} splits." dataset = ds if len(dsets) == 1 else hf_datasets.concatenate_datasets(dsets) super()._init_dataset(dataset, total_samples)