mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-16 17:20:05 +00:00
add task to processor
This commit is contained in:
@@ -316,7 +316,7 @@ def run_ee_inference_loop(
|
||||
relative_state = convert_state_to_relative(ee_state_tensor.unsqueeze(0))
|
||||
ee_state = {k: float(relative_state[0, i]) for i, k in enumerate(sorted(ee_state.keys()))}
|
||||
|
||||
policy_obs = {"observation.state": ee_state}
|
||||
policy_obs = {"observation.state": ee_state, "task": task}
|
||||
|
||||
# Add images
|
||||
for cam_name in robot.cameras:
|
||||
|
||||
Reference in New Issue
Block a user