From 226300ca72bf66eb5621c8e00c2a0dc43a232972 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Wed, 9 Oct 2019 14:02:14 -0400
Subject: [PATCH] removed debugging print
---
flatland/envs/distance_map.py | 1 -
flatland/envs/rail_env.py | 1 +
tests/test_flaltland_rail_agent_status.py | 8 ++++----
3 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/flatland/envs/distance_map.py b/flatland/envs/distance_map.py
index c6e73b0b..2bc1a511 100644
--- a/flatland/envs/distance_map.py
+++ b/flatland/envs/distance_map.py
@@ -55,7 +55,6 @@ class DistanceMap:
self.env_width = rail.width
def _compute(self, agents: List[EnvAgent], rail: GridTransitionMap):
- print("computing distance map")
self.agents_previous_computation = self.agents
self.distance_map = np.inf * np.ones(shape=(len(agents),
self.env_height,
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 2543fc91..cda0dee9 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -459,6 +459,7 @@ class RailEnv(Environment):
agent.status = RailAgentStatus.ACTIVE
agent.position = agent.initial_position
self.rewards_dict[i_agent] += self.step_penalty * agent.speed_data['speed']
+ print(self.rewards_dict[i_agent])
return
else:
# TODO: Here we need to check for the departure time in future releases with full schedules
diff --git a/tests/test_flaltland_rail_agent_status.py b/tests/test_flaltland_rail_agent_status.py
index ec7635ba..099ccce6 100644
--- a/tests/test_flaltland_rail_agent_status.py
+++ b/tests/test_flaltland_rail_agent_status.py
@@ -50,7 +50,7 @@ def test_initial_status():
reward=env.step_penalty * 0.5, # running at speed 0.5
),
Replay(
- position=(3, 8),
+ position=(3, 9),
direction=Grid4TransitionsEnum.WEST,
status=RailAgentStatus.ACTIVE,
action=RailEnvActions.MOVE_FORWARD,
@@ -64,7 +64,7 @@ def test_initial_status():
reward=env.step_penalty * 0.5, # running at speed 0.5
),
Replay(
- position=(3, 7),
+ position=(3, 8),
direction=Grid4TransitionsEnum.WEST,
status=RailAgentStatus.ACTIVE,
action=RailEnvActions.MOVE_FORWARD,
@@ -79,7 +79,7 @@ def test_initial_status():
status=RailAgentStatus.ACTIVE
),
Replay(
- position=(3, 6),
+ position=(3, 7),
direction=Grid4TransitionsEnum.WEST,
action=RailEnvActions.MOVE_RIGHT,
reward=env.step_penalty * 0.5, # wrong action is corrected to forward without penalty!
@@ -93,7 +93,7 @@ def test_initial_status():
status=RailAgentStatus.ACTIVE
),
Replay(
- position=(3, 5),
+ position=(3, 6),
direction=Grid4TransitionsEnum.WEST,
action=None,
reward=env.global_reward, # already done
--
GitLab