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

fix(eval): prevent eval_policy crash when start_seed is None and num_envs>1 (#4203)

Browse Source 
* fix(eval): align seed list length with num_envs when unseeded

eval_policy appended a single None per batch to all_seeds on the unseeded path while the reward and success lists grew by num_envs. The per-episode zip(..., strict=True) then raised ValueError for num_envs > 1. Extend all_seeds by num_envs so the lists stay aligned.

* chore(tests): delete lerobot_eval test

---------

Co-authored-by: Devin Lai <markauto75@gmail.com>
This commit is contained in:
Steven Palma
2026-07-28 18:41:28 +02:00
committed by GitHub
parent 4d076845ac
commit f37be3edbe
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_eval.py
+1 -1
View File
@@ -564,7 +564,7 @@ def eval_policy(
if seeds: if seeds:
all_seeds.extend(seeds) all_seeds.extend(seeds)
else: else:
all_seeds.append(None) all_seeds.extend([None] * env.num_envs)
# FIXME: episode_data is either None or it doesn't exist # FIXME: episode_data is either None or it doesn't exist
if return_episode_data: if return_episode_data: