add task to processor

This commit is contained in:
Pepijn
2026-01-08 16:56:59 +01:00
parent 71f3cf30cd
commit 037747da82
+1 -1
View File
@@ -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: