From 037747da826857332c4645aba4531cb233262cca Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Thu, 8 Jan 2026 16:56:59 +0100
Subject: [PATCH] add task to processor

---
 examples/openarms/evaluate_ee.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/openarms/evaluate_ee.py b/examples/openarms/evaluate_ee.py
index ff62ba30f..50be357dd 100644
--- a/examples/openarms/evaluate_ee.py
+++ b/examples/openarms/evaluate_ee.py
@@ -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: