From cdea95d379ed543eb2ac437fc8645e3cfca855b1 Mon Sep 17 00:00:00 2001 From: arxyzan Date: Mon, 5 Feb 2024 23:47:29 +0330 Subject: [PATCH] :bug: Fix minor bug in `speech_recognition_dataset.py` --- hezar/data/datasets/speech_recognition_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hezar/data/datasets/speech_recognition_dataset.py b/hezar/data/datasets/speech_recognition_dataset.py index 082fd49e..0c68a0e9 100644 --- a/hezar/data/datasets/speech_recognition_dataset.py +++ b/hezar/data/datasets/speech_recognition_dataset.py @@ -51,7 +51,7 @@ def __init__(self, config: SpeechRecognitionDatasetConfig, split=None, **kwargs) ) def _load(self, split): - data = load_from_disk(self.config.path)[split] + data = load_dataset(self.config.path, split=split) data = data.cast_column(self.config.audio_column, Audio(sampling_rate=self.config.sampling_rate)) return data