admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 09:46:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(audio load file): adding missing dimension when loading mono audio data

Browse Source
This commit is contained in:
CarolinePascal
2025-04-22 18:31:38 +02:00
parent 2962330bb1
commit 6d726266fd
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/utils.py
+5
View File
@@ -417,6 +417,11 @@ def load_image_as_numpy(
def load_audio_from_path(fpath: str | Path) -> np.ndarray: def load_audio_from_path(fpath: str | Path) -> np.ndarray:
audio_data, _ = read(fpath, dtype="float32") audio_data, _ = read(fpath, dtype="float32")
# Fill missing channel dimension when loading mono audio data
if audio_data.ndim == 1:
audio_data = np.expand_dims(audio_data, axis=1)
return audio_data return audio_data