From f4d3a70e1f856e35c2ccbfedb13b75cdda175f8e Mon Sep 17 00:00:00 2001 From: mshukor Date: Wed, 2 Apr 2025 17:34:05 +0200 Subject: [PATCH] Update lerobot/scripts/eval.py Co-authored-by: Remi --- lerobot/scripts/eval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index 4cad2dd1..a1d5fc24 100644 --- a/lerobot/scripts/eval.py +++ b/lerobot/scripts/eval.py @@ -156,8 +156,8 @@ def rollout( key: observation[key].to(device, non_blocking=device.type == "cuda") for key in observation } - # Infer "task" from envs. Works with AsyncVectorEnv. - observation = infer_envs_task(env, observation) + # Infer "task" from attributes of environments. Works with AsyncVectorEnv. + observation = addd_envs_task(env, observation) with torch.inference_mode(): action = policy.select_action(observation)