admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-18 10:10:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(audio feature shape): fixing audio feature shape ordering (frames first, channels second)

Browse Source
This commit is contained in:
CarolinePascal
2025-04-22 17:06:10 +02:00
parent e714ff22e2
commit e4dd00c8f5
3 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/datasets/test_datasets.py
+1 -1
View File
@@ -82,7 +82,7 @@ def audio_dataset(tmp_path, empty_lerobot_dataset_factory):
features = {
"audio": {
"dtype": "audio",
"shape": (DUMMY_AUDIO_CHANNELS,),
"shape": (1, DUMMY_AUDIO_CHANNELS),
"names": [
"channels",
],
tests/fixtures/constants.py
Vendored
+2 -2
View File
@@ -41,8 +41,8 @@ DUMMY_VIDEO_INFO = {
"has_audio": False,
}
DUMMY_MICROPHONE_FEATURES = {
"laptop": {"dtype": "audio", "shape": (1,), "names": ["channels"], "info": None},
"phone": {"dtype": "audio", "shape": (1,), "names": ["channels"], "info": None},
"laptop": {"dtype": "audio", "shape": (1, 2), "names": ["channels"], "info": None},
"phone": {"dtype": "audio", "shape": (1, 2), "names": ["channels"], "info": None},
}
DEFAULT_SAMPLE_RATE = 48000
DUMMY_AUDIO_CHANNELS = 2