From d35479d612badc29e24e886ea87b4348138657a9 Mon Sep 17 00:00:00 2001
From: "S.P. Mohanty" <spmohanty91@gmail.com>
Date: Thu, 24 Oct 2019 19:02:02 +0200
Subject: [PATCH] Fix small typo in previous commits
---
flatland/envs/rail_env.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index b5debcd4..cb4da95a 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -489,7 +489,7 @@ class RailEnv(Environment):
if (self._max_episode_steps is not None) and (self._elapsed_steps >= self._max_episode_steps):
self.dones["__all__"] = True
for i_agent in range(self.get_num_agents()):
- self.dones[i] = True
+ self.dones[i_agent] = True
return self._get_observations(), self.rewards_dict, self.dones, info_dict
--
GitLab