From c5010fee9a40c64d61e69918ee64691f69a4b4c8 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Wed, 20 Mar 2024 08:20:56 +0000 Subject: [PATCH] fix seeding --- lerobot/common/envs/factory.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/lerobot/common/envs/factory.py b/lerobot/common/envs/factory.py index 689f5869..e187d713 100644 --- a/lerobot/common/envs/factory.py +++ b/lerobot/common/envs/factory.py @@ -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) + ], )