mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-23 12:40:08 +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))
|
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()))}
|
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
|
# Add images
|
||||||
for cam_name in robot.cameras:
|
for cam_name in robot.cameras:
|
||||||
|
|||||||
Reference in New Issue
Block a user