From 96f3d25c32aef2dbcbcbe66af01a0479de3d2a30 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marcin=20W=C4=85troba?= <markowanga@gmail.com>
Date: Thu, 12 Jan 2023 00:05:28 +0100
Subject: [PATCH] Add minds14 import
---
new_experiment/pipeline/import_datasets.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/new_experiment/pipeline/import_datasets.py b/new_experiment/pipeline/import_datasets.py
index 07bcb2a..a16bf91 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):
--
GitLab