From 45c5a924e537966a410681056b0b30ed8c0aff78 Mon Sep 17 00:00:00 2001
From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com>
Date: Mon, 8 Jul 2024 08:47:54 -0400
Subject: [PATCH] forgot a +1

---
 pokemonred_puffer/train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pokemonred_puffer/train.py b/pokemonred_puffer/train.py
index dd4981b..cb8dd0f 100644
--- 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,