From 15a63e883abbbef1f37c5f8fa63e872aaebae530 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Tue, 1 Oct 2019 11:29:03 -0400
Subject: [PATCH] introduced depot for moving agents after they finished
---
flatland/envs/rail_env.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 303b271f..f79ab504 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -24,6 +24,7 @@ from flatland.envs.schedule_generators import random_schedule_generator, Schedul
m.patch()
+DEPOT_POSITION = (-10, -10)
class RailEnvActions(IntEnum):
DO_NOTHING = 0 # implies change of direction in a dead-end!
@@ -503,8 +504,8 @@ class RailEnv(Environment):
self.dones[i_agent] = True
agent.moving = False
# TODO: Moving agents to arbitrary position
- agent.position = (i_agent, 0)
- agent.target = (i_agent, 0)
+ agent.position = DEPOT_POSITION
+ agent.target = DEPOT_POSITION
else:
self.rewards_dict[i_agent] += self.step_penalty * agent.speed_data['speed']
else:
--
GitLab