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

fix seeding

This commit is contained in:
Alexander Soare 2024-03-20 08:20:56 +00:00
parent 18fa88475b
commit c5010fee9a
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lerobot/common/envs/factory.py
View File

@ -13,7 +13,6 @@ def make_env(cfg, transform=None):
"from_pixels": cfg.env.from_pixels,
"pixels_only": cfg.env.pixels_only,
"image_size": cfg.env.image_size,
"seed": cfg.seed,
"num_prev_obs": cfg.n_obs_steps - 1,
}
@ -59,10 +58,9 @@ def make_env(cfg, transform=None):
return SerialEnv(
cfg.rollout_batch_size,
create_env_fn=_make_env,
create_env_kwargs={
"seed": env_seed # noqa: B035
for env_seed in range(cfg.seed, cfg.seed + cfg.rollout_batch_size)
},
create_env_kwargs=[
{"seed": env_seed} for env_seed in range(cfg.seed, cfg.seed + cfg.rollout_batch_size)
],
)