From 92f08933ec63450400a0e12de69659e13a32ca16 Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Wed, 11 Mar 2026 11:02:07 +0100 Subject: [PATCH] fix(features): allowing for sequence of shape (1,) when a names list is provided --- src/lerobot/datasets/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/datasets/utils.py b/src/lerobot/datasets/utils.py index 8bc56a1bd..d1c9d8c1b 100644 --- a/src/lerobot/datasets/utils.py +++ b/src/lerobot/datasets/utils.py @@ -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(