mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-27 06:29:47 +00:00
precommit
This commit is contained in:
@@ -672,11 +672,15 @@ def eval_policy_all(
|
|||||||
"""
|
"""
|
||||||
if max_parallel_tasks == 1:
|
if max_parallel_tasks == 1:
|
||||||
yield from _eval_monotask(
|
yield from _eval_monotask(
|
||||||
envs, policy, preprocessor=preprocessor,
|
envs,
|
||||||
postprocessor=postprocessor, n_episodes=n_episodes,
|
policy,
|
||||||
|
preprocessor=preprocessor,
|
||||||
|
postprocessor=postprocessor,
|
||||||
|
n_episodes=n_episodes,
|
||||||
max_episodes_rendered=max_episodes_rendered,
|
max_episodes_rendered=max_episodes_rendered,
|
||||||
videos_dir=videos_dir, return_episode_data=return_episode_data,
|
videos_dir=videos_dir,
|
||||||
start_seed=start_seed
|
return_episode_data=return_episode_data,
|
||||||
|
start_seed=start_seed,
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
yield from _eval_parallel(
|
yield from _eval_parallel(
|
||||||
|
|||||||
Reference in New Issue
Block a user