From 989b8880b09b375ad1db379ad2f0ff9b3d401339 Mon Sep 17 00:00:00 2001
From: Dipam Chakraborty <dipam@aicrowd.com>
Date: Sat, 14 Aug 2021 03:34:00 +0530
Subject: [PATCH] always reset distance map in env.reset
---
flatland/envs/rail_env.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 4ee6dbae..ab0e1487 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -377,6 +377,8 @@ class RailEnv(Environment):
for agent_i, agent in enumerate(self.agents):
agent.earliest_departure = timetable.earliest_departures[agent_i]
agent.latest_arrival = timetable.latest_arrivals[agent_i]
+ else:
+ self.distance_map.reset(self.agents, self.rail)
# Agent Positions Map
self.agent_positions = np.zeros((self.height, self.width), dtype=int) - 1
--
GitLab