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

remove debug

Browse Source
This commit is contained in:
Khalil Meftah
2026-04-26 18:08:13 +02:00
parent 58b6d844c4
commit 613d581f6c
1 changed files with 0 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rl/gym_manipulator.py
-14
View File
@@ -708,20 +708,6 @@ def control_loop(
terminated = transition.get(TransitionKey.DONE, False) terminated = transition.get(TransitionKey.DONE, False)
truncated = transition.get(TransitionKey.TRUNCATED, False) truncated = transition.get(TransitionKey.TRUNCATED, False)
# DEBUG: trace gripper value through the pipeline.
_action_t = transition[TransitionKey.ACTION]
_teleop_t = transition[TransitionKey.COMPLEMENTARY_DATA].get("teleop_action")
_info_teleop = transition[TransitionKey.INFO].get("teleop_action")
_is_intervention = transition[TransitionKey.INFO].get(TeleopEvents.IS_INTERVENTION, False)
print(
f"[grip-debug] use_gripper={use_gripper} "
f"is_intervention={_is_intervention} "
f"neutral={neutral_action.tolist()} "
f"ACTION={_action_t.squeeze().tolist() if hasattr(_action_t, 'squeeze') else _action_t} "
f"comp.teleop_action={_teleop_t.tolist() if hasattr(_teleop_t, 'tolist') else _teleop_t} "
f"info.teleop_action={_info_teleop.tolist() if hasattr(_info_teleop, 'tolist') else _info_teleop}"
)
if cfg.mode == "record": if cfg.mode == "record":
observations = { observations = {
k: v.squeeze(0).cpu() k: v.squeeze(0).cpu()