forgot a +1

thatguy11325 · Jul 8, 2024 · 45c5a92 · 45c5a92
1 parent 83ed6f7
commit 45c5a92
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pokemonred_puffer/train.py b/pokemonred_puffer/train.py
@@ -184,8 +184,8 @@ def train(
         vec = pufferlib.vector.Ray
 
     # TODO: Remove the +1 once the driver env doesn't permanently increase the env id
-    env_send_queues = [Queue() for _ in range(2 * args.train.num_envs)]
-    env_recv_queues = [Queue() for _ in range(2 * args.train.num_envs)]
+    env_send_queues = [Queue() for _ in range(2 * args.train.num_envs + 1)]
+    env_recv_queues = [Queue() for _ in range(2 * args.train.num_envs + 1)]
 
     vecenv = pufferlib.vector.make(
         env_creator,