From bbfdae22c91a39f8c21f29689fb454bc68720317 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Tue, 23 Jul 2019 15:44:01 -0400
Subject: [PATCH] initial changes to crossings. Now not detected anymore as
unusable switches
---
flatland/envs/observations.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 99139900..12801175 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -404,7 +404,8 @@ class TreeObsForRailEnv(ObservationBuilder):
# Check if crossing is found --> Not an unusable switch
if int(transition_bit, 2) == int('1000010000100001', 2):
- total_transitions = 1
+ # Treat the crossing as a straight rail cell
+ total_transitions = 2
num_transitions = np.count_nonzero(cell_transitions)
exploring = False
--
GitLab