ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-31 19:01:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(features): allowing for sequence of shape (1,) when a names list is provided

Browse Source
This commit is contained in:
CarolinePascal
2026-03-11 11:02:07 +01:00
parent 09c93c4aa1
commit 92f08933ec
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -578,7 +578,7 @@ def get_hf_features_from_features(features: dict) -> datasets.Features:
continue
elif ft["dtype"] == "image":
hf_features[key] = datasets.Image()
elif ft["shape"] == (1,):
elif ft["shape"] == (1,) and ft["names"] is None:
hf_features[key] = datasets.Value(dtype=ft["dtype"])
elif len(ft["shape"]) == 1:
hf_features[key] = datasets.Sequence(