From d8c8f42bfdbd5f74db6471c9ac4625310f03b7ed Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Fri, 7 Jun 2019 10:40:27 +0200
Subject: [PATCH] typo and bugfinx in split_tree
---
flatland/envs/observations.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index bd095dba..4476bef5 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -491,7 +491,7 @@ class TreeObsForRailEnv(ObservationBuilder):
child_size = (len(tree) - num_features_per_node) // 4
tree_data = tree[0:num_features_per_node - 1].tolist()
distance_data = [tree[num_features_per_node - 3]]
- agent_data = tree[-2:]
+ agent_data = tree[-2:].tolist()
for children in range(4):
child_tree = tree[(num_features_per_node + children * child_size):
(num_features_per_node + (children + 1) * child_size)]
@@ -501,7 +501,7 @@ class TreeObsForRailEnv(ObservationBuilder):
if len(tmp_tree_data) > 0:
tree_data.extend(tmp_tree_data)
distance_data.extend(tmp_distance_data)
- agent_data.extrend(tmp_agent_data)
+ agent_data.extend(tmp_agent_data)
return tree_data, distance_data, agent_data
--
GitLab