ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-03 20:31:25 +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

5
src/lerobot/datasets/utils.py
View File

@@ -417,6 +417,11 @@ def load_image_as_numpy(
def load_audio_from_path(fpath: str | Path) -> np.ndarray:
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