add reset mode to vectorized_single_agent_race.py

axelbr · Dec 1, 2020 · 8cf0732 · 8cf0732
1 parent 3de64a5
commit 8cf0732
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/racecar_gym/envs/vectorized_single_agent_race.py b/racecar_gym/envs/vectorized_single_agent_race.py
@@ -66,7 +66,7 @@ def step(self, actions: Tuple[Dict]):
 
     def reset(self, mode: str = 'grid'):
         observations = []
-        for i, conn in self._env_connections:
+        for i, conn in enumerate(self._env_connections):
             conn.send('reset')
             conn.send(mode)
             obs = conn.recv()