From 195b7773674e0f8a9a7edce94678b9095db29158 Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Mon, 20 Apr 2026 18:13:37 +0200 Subject: [PATCH] fix(rollout): features check --- src/lerobot/rollout/context.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/rollout/context.py b/src/lerobot/rollout/context.py index 7a7249e50..273198d56 100644 --- a/src/lerobot/rollout/context.py +++ b/src/lerobot/rollout/context.py @@ -305,7 +305,7 @@ def build_rollout_context( if not rename_map: expected_visuals = {k for k, v in full_config.input_features.items() if v.type == FeatureType.VISUAL} provided_visuals = { - f"observation.{k}" for k, v in robot.observation_features.items() if isinstance(v, tuple) + f"observation.images.{k}" for k, v in robot.observation_features.items() if isinstance(v, tuple) } policy_subset = expected_visuals.issubset(provided_visuals) hw_subset = provided_visuals.issubset(expected_visuals)