🐛 Fix minor bug in speech_recognition_dataset.py

hezarai · Feb 5, 2024 · cdea95d · cdea95d
1 parent b37cd26
commit cdea95d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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