wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-04-02 13:59:55 +00:00
parent b5439a276d
commit 998bd92774
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/eval.py
View File

@ -158,7 +158,7 @@ def rollout(
observation["task"] = env.call("task_description") observation["task"] = env.call("task_description")
elif hasattr(env.envs[0], "task"): elif hasattr(env.envs[0], "task"):
observation["task"] = env.call("task") observation["task"] = env.call("task")
else: #IIRC This was to deal with datasets without language instructions, i.e. aloha transfer cube and etc.? 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])] observation["task"] = ["" for _ in range(observation[list(observation.keys())[0]].shape[0])]
with torch.inference_mode(): with torch.inference_mode():
action = policy.select_action(observation) action = policy.select_action(observation)