From 7234b44abaefbfd545cf9d4625376f0ae9618fa2 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Thu, 29 Aug 2019 10:36:53 +0200
Subject: [PATCH] fix convergence issue
---
flatland/envs/observations.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 4158675c..c4fed2e0 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -383,8 +383,9 @@ class TreeObsForRailEnv(ObservationBuilder):
elif int_position in np.delete(self.predicted_pos[pre_step], handle, 0):
conflicting_agent = np.where(self.predicted_pos[pre_step] == int_position)
for ca in conflicting_agent[0]:
- if direction != self.predicted_dir[pre_step][ca] and cell_transitions[self._reverse_dir(
- self.predicted_dir[pre_step][ca])] == 1 and tot_dist < potential_conflict:
+ if direction != self.predicted_dir[pre_step][ca] \
+ and cell_transitions[self._reverse_dir(self.predicted_dir[pre_step][ca])] == 1 \
+ and tot_dist < potential_conflict: # noqa: E125
potential_conflict = tot_dist
if self.env.dones[ca] and tot_dist < potential_conflict:
potential_conflict = tot_dist
@@ -394,7 +395,8 @@ class TreeObsForRailEnv(ObservationBuilder):
conflicting_agent = np.where(self.predicted_pos[post_step] == int_position)
for ca in conflicting_agent[0]:
if direction != self.predicted_dir[post_step][ca] and cell_transitions[self._reverse_dir(
- self.predicted_dir[post_step][ca])] == 1 and tot_dist < potential_conflict:
+ self.predicted_dir[post_step][ca])] == 1 \
+ and tot_dist < potential_conflict: # noqa: E125
potential_conflict = tot_dist
if self.env.dones[ca] and tot_dist < potential_conflict:
potential_conflict = tot_dist
--
GitLab