admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-28 04:06:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(eval): catch AttributeError and NotImplementedError explicitly for task description

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Pepijn
2026-04-08 19:41:39 +02:00
parent 973bb7c7d7
commit c3fa286fdd
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_eval.py
+2 -2
View File
@@ -169,10 +169,10 @@ def rollout(
# env.call() works with both SyncVectorEnv and AsyncVectorEnv. # env.call() works with both SyncVectorEnv and AsyncVectorEnv.
try: try:
observation["task"] = list(env.call("task_description")) observation["task"] = list(env.call("task_description"))
except Exception: except (AttributeError, NotImplementedError):
try: try:
observation["task"] = list(env.call("task")) observation["task"] = list(env.call("task"))
except Exception: except (AttributeError, NotImplementedError):
observation["task"] = [""] * env.num_envs observation["task"] = [""] * env.num_envs
# Apply environment-specific preprocessing (e.g., LiberoProcessorStep for LIBERO) # Apply environment-specific preprocessing (e.g., LiberoProcessorStep for LIBERO)