From a3a73f6822d6a2c7e6f6bc8fc43adada04ef5cb6 Mon Sep 17 00:00:00 2001 From: mshukor Date: Fri, 30 May 2025 13:10:56 +0200 Subject: [PATCH] add stuff related to real eval --- lerobot/common/robot_devices/control_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lerobot/common/robot_devices/control_utils.py b/lerobot/common/robot_devices/control_utils.py index 68f69487..103c4fdb 100644 --- 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