From b5439a276d5225c78de57c9e990728b51a8cefc0 Mon Sep 17 00:00:00 2001 From: mshukor Date: Wed, 2 Apr 2025 15:59:40 +0200 Subject: [PATCH] Update lerobot/scripts/eval.py Co-authored-by: Dana Aubakirova <118912928+danaaubakirova@users.noreply.github.com> --- lerobot/scripts/eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index 6cfad233..20ace96e 100644 --- a/lerobot/scripts/eval.py +++ b/lerobot/scripts/eval.py @@ -158,7 +158,7 @@ def rollout( observation["task"] = env.call("task_description") elif hasattr(env.envs[0], "task"): observation["task"] = env.call("task") - else: + else: #IIRC This was to deal with datasets without language instructions, i.e. aloha transfer cube and etc.? observation["task"] = ["" for _ in range(observation[list(observation.keys())[0]].shape[0])] with torch.inference_mode(): action = policy.select_action(observation)