admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-23 04:30:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(rollout): features check

Browse Source
This commit is contained in:
Steven Palma
2026-04-20 18:13:37 +02:00
parent b49e4016f2
commit 195b777367
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rollout/context.py
+1 -1
View File
@@ -305,7 +305,7 @@ def build_rollout_context(
if not rename_map: if not rename_map:
expected_visuals = {k for k, v in full_config.input_features.items() if v.type == FeatureType.VISUAL} expected_visuals = {k for k, v in full_config.input_features.items() if v.type == FeatureType.VISUAL}
provided_visuals = { 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) policy_subset = expected_visuals.issubset(provided_visuals)
hw_subset = provided_visuals.issubset(expected_visuals) hw_subset = provided_visuals.issubset(expected_visuals)