From 00201b71aed2c0f4f6f8ef3e795b75892c2fcaad Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Fri, 7 Jun 2019 10:54:51 +0200
Subject: [PATCH] minor bugfixes
---
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 4476bef5..dc5741b9 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -489,7 +489,7 @@ class TreeObsForRailEnv(ObservationBuilder):
depth += 1
pow4 *= 4
child_size = (len(tree) - num_features_per_node) // 4
- tree_data = tree[0:num_features_per_node - 1].tolist()
+ tree_data = tree[0:num_features_per_node - 3].tolist()
distance_data = [tree[num_features_per_node - 3]]
agent_data = tree[-2:].tolist()
for children in range(4):
--
GitLab