From bd12fb341388e649bb26a0762f722c006a0dea0d Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Mon, 9 Nov 2020 14:13:58 +0100
Subject: [PATCH] FastTreeObs (fix) -> 0.8157
---
utils/fast_tree_obs.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/utils/fast_tree_obs.py b/utils/fast_tree_obs.py
index e8df61e..db22a8f 100755
--- a/utils/fast_tree_obs.py
+++ b/utils/fast_tree_obs.py
@@ -207,8 +207,10 @@ class FastTreeObs(ObservationBuilder):
possible_transitions = self.env.rail.get_transitions(*new_position, new_direction)
if agents_on_switch:
orientation = new_direction
- if fast_count_nonzero(possible_transitions) == 1:
+ possible_transitions_nonzero = fast_count_nonzero(possible_transitions)
+ if possible_transitions_nonzero == 1:
orientation = fast_argmax(possible_transitions)
+
for dir_loop, branch_direction in enumerate(
[(orientation + dir_loop) % 4 for dir_loop in range(-1, 3)]):
# branch the exploration path and aggregate the found information
--
GitLab