From 7bebce11d39baf6e04f11f5c94753ec9661f9d3c Mon Sep 17 00:00:00 2001 From: philtabor Date: Sat, 27 Jul 2024 08:46:27 -0400 Subject: [PATCH] fix bug related to issue #4 --- protorl/wrappers/vec_env.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/protorl/wrappers/vec_env.py b/protorl/wrappers/vec_env.py index 9c1a60a..9f006fd 100644 --- a/protorl/wrappers/vec_env.py +++ b/protorl/wrappers/vec_env.py @@ -129,8 +129,8 @@ def make_vec_envs(env_name, use_atari=False, seed=None, n_threads=2, **kwargs): mpi_rank = MPI.COMM_WORLD.Get_rank() if MPI else 0 seed = seed + 10000 * mpi_rank if seed is not None else None set_global_seeds(seed) - envs = [make_single_env(env_name, i, use_atari, **kwargs) - for i in range(n_threads)] + envs = [make_single_env(env_name, use_atari, **kwargs) + for _ in range(n_threads)] envs = SubprocVecEnv(envs, seed)