From 13a429e5c799b55cbc12217cb177bcbc83aa814e Mon Sep 17 00:00:00 2001 From: Francesco Capuano Date: Thu, 6 Nov 2025 00:47:38 +0000 Subject: [PATCH] fix: DEFAULT FEATURES must be present when creating metadata. If not, then we raise. this is a first step towards standardazing the dataset format, or otherwise (as it is now) everything would be allowed --- src/lerobot/datasets/lerobot_dataset.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/lerobot/datasets/lerobot_dataset.py b/src/lerobot/datasets/lerobot_dataset.py index 48608a809..03e2ae44f 100644 --- a/src/lerobot/datasets/lerobot_dataset.py +++ b/src/lerobot/datasets/lerobot_dataset.py @@ -512,7 +512,9 @@ class LeRobotDatasetMetadata: obj.root.mkdir(parents=True, exist_ok=False) - features = {**features, **DEFAULT_FEATURES} + assert set(DEFAULT_FEATURES.keys()).issubset(set(features.keys())), ( + f"Default features must be a subset of the provided features! Got {set(features.keys())} and {set(DEFAULT_FEATURES.keys())}" + ) _validate_feature_names(features) obj.tasks = None