Fix bug

fixie-ai · Oct 17, 2024 · 9bd8e4b · 9bd8e4b
1 parent 0f983af
commit 9bd8e4b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)