diff --git a/new_experiment/pipeline/import_datasets.py b/new_experiment/pipeline/import_datasets.py index 07bcb2a228abad344b34caf6f7186814af40f8ea..a16bf91022687feafdbd397e6de604203a3730ea 100644 --- a/new_experiment/pipeline/import_datasets.py +++ b/new_experiment/pipeline/import_datasets.py @@ -24,7 +24,7 @@ def import_minds14_dataset(dataset_lang: str, experiment_dataset_name: str): get_experiment_repository(experiment_dataset_name), get_minio_audio_record_repository(), experiment_dataset_name, - ), load_dataset('PolyAI/minds14', dataset_lang)) + ), load_dataset('PolyAI/minds14', dataset_lang)['train']) def import_voxpopuli_dataset(dataset_lang: str, experiment_dataset_name: str):