From 8cf073286dde4dca44e9b3869f35c74e7e212c2b Mon Sep 17 00:00:00 2001 From: axelbr Date: Tue, 1 Dec 2020 11:21:13 +0100 Subject: [PATCH] add reset mode to vectorized_single_agent_race.py --- racecar_gym/envs/vectorized_single_agent_race.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/racecar_gym/envs/vectorized_single_agent_race.py b/racecar_gym/envs/vectorized_single_agent_race.py index 8ac54f07..3a32cedd 100644 --- 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()