From 201c8691f07829b5659e8e536aca9e8b8c20db80 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Fri, 7 Jun 2019 10:38:12 +0200
Subject: [PATCH] bug fix in split_tree
---
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 5c4ed86b..bd095dba 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -479,7 +479,7 @@ class TreeObsForRailEnv(ObservationBuilder):
"""
if len(tree) < num_features_per_node:
- return [], []
+ return [], [], []
depth = 0
tmp = len(tree) / num_features_per_node - 1
--
GitLab