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