Commit 7d1ee39b authored by Erik Nygren's avatar Erik Nygren
Browse files

fixed tree observation data bug, where not the correct values where provided to agent

parent 20d93455
Pipeline #1082 passed with stages
in 13 minutes and 17 seconds
......@@ -234,6 +234,9 @@ class TreeObsForRailEnv(ObservationBuilder):
0 = no agent present other direction than myself
#8: possible conflict detected
1 = Other agent predicts to pass along this cell at the same time as the agent
0 = No other agent reserve the same cell at similar time
Missing/padding nodes are filled in with -inf (truncated).
......@@ -547,8 +550,7 @@ class TreeObsForRailEnv(ObservationBuilder):
child_size = (len(tree) - num_features_per_node) // 4
tree_data = tree[0:4].tolist()
distance_data = [tree[4]]
agent_data = tree[-3:].tolist()
agent_data = tree[num_features_per_node - 3:num_features_per_node].tolist()
for children in range(4):
child_tree = tree[(num_features_per_node + children * child_size):
(num_features_per_node + (children + 1) * child_size)]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment