From 7a562a17b4375488dd405f91a2caf139dd813b1f Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Fri, 25 Oct 2019 11:56:02 -0400
Subject: [PATCH] removed == 0.0 comparison and replaced by np.isclose
This closes #198
---
flatland/envs/rail_env.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index cb4da95a..290cf96b 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -372,7 +372,8 @@ class RailEnv(Environment):
info_dict: Dict = {
'action_required': {
i: (agent.status == RailAgentStatus.READY_TO_DEPART or (
- agent.status == RailAgentStatus.ACTIVE and agent.speed_data['position_fraction'] == 0.0))
+ agent.status == RailAgentStatus.ACTIVE and np.isclose(agent.speed_data['position_fraction'], 0.0,
+ rtol=1e-03)))
for i, agent in enumerate(self.agents)},
'malfunction': {
i: self.agents[i].malfunction_data['malfunction'] for i in range(self.get_num_agents())
--
GitLab