From f0d9afeb828ca62f6a3e9d9fabfd857331ed1652 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sun, 1 Sep 2019 16:39:53 -0400
Subject: [PATCH] fixed formatting
---
flatland/envs/rail_env.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 4926fb1f..ea9c16e1 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -339,7 +339,7 @@ class RailEnv(Environment):
agent.malfunction_data['malfunction'] -= 1
# Broken agents are stopped
- self.rewards_dict[i_agent] += step_penalty # * agent.speed_data['speed']
+ self.rewards_dict[i_agent] += step_penalty * agent.speed_data['speed']
self.agents[i_agent].moving = False
action_dict[i_agent] = RailEnvActions.DO_NOTHING
@@ -404,14 +404,14 @@ class RailEnv(Environment):
else:
# TODO: an invalid action was chosen after entering the cell. The agent cannot move.
self.rewards_dict[i_agent] += invalid_action_penalty
- self.rewards_dict[i_agent] += step_penalty #* agent.speed_data['speed']
+ self.rewards_dict[i_agent] += step_penalty * agent.speed_data['speed']
self.rewards_dict[i_agent] += stop_penalty
agent.moving = False
continue
else:
# TODO: an invalid action was chosen after entering the cell. The agent cannot move.
self.rewards_dict[i_agent] += invalid_action_penalty
- self.rewards_dict[i_agent] += step_penalty #* agent.speed_data['speed']
+ self.rewards_dict[i_agent] += step_penalty * agent.speed_data['speed']
self.rewards_dict[i_agent] += stop_penalty
agent.moving = False
continue
--
GitLab