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(