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

only use position in dataset

Browse Source
This commit is contained in:
Pepijn
2025-12-30 12:01:26 +01:00
parent c07ab7e1fa
commit 1716d599c1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/rac/rac_data_collection_openarms.py
+2 -2
View File
@@ -250,8 +250,8 @@ def rac_rollout_loop(
was_intervention_active = True
obs = robot.get_observation()
# Filter out internal timing data before building frame
obs_filtered = {k: v for k, v in obs.items() if not k.startswith("_")}
# Filter to only include keys in observation_features (excludes .vel, .torque, _timing, etc.)
obs_filtered = {k: v for k, v in obs.items() if k in robot.observation_features}
obs_frame = build_dataset_frame(dataset.features, obs_filtered, prefix=OBS_STR)
if not events["intervention_active"]: