From 5dd5d2f585b8c84729f9aab38638e07c9cf5ba4d Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Tue, 16 Jul 2019 16:30:29 -0400
Subject: [PATCH] minor updates to observation features. ALso reinstated the
dead-end branching to keep trees smaller
---
flatland/envs/observations.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 9dc397a0..3e8583d9 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -408,7 +408,7 @@ class TreeObsForRailEnv(ObservationBuilder):
nbits = total_transitions
if nbits == 1:
# Dead-end!
- last_is_dead_end = False
+ last_is_dead_end = True
if not last_is_dead_end:
# Keep walking through the tree along `direction'
--
GitLab