add stuff related to real eval

2025-05-30 13:10:56 +02:00
parent a9d425f470
commit a3a73f6822
1 changed files with 2 additions and 1 deletions
--- a/lerobot/common/robot_devices/control_utils.py
+++ b/lerobot/common/robot_devices/control_utils.py
@@ -256,7 +256,8 @@ def control_loop(
        else:
            observation = robot.capture_observation()
            action = None
-
+            observation['task'] = [single_task]
+            observation["robot_type"] = [policy.robot_type] if hasattr(policy, "robot_type") else [""]
            if policy is not None:
                pred_action = predict_action(
                    observation, policy, get_safe_torch_device(policy.config.device), policy.config.use_amp