diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py index 506afd686a0c5f5a5b35f2081698aa42f745e04d..dc0edc9b4960749a868635edb8df95a5608b629c 100644 --- a/flatland/envs/observations.py +++ b/flatland/envs/observations.py @@ -261,7 +261,7 @@ class TreeObsForRailEnv(ObservationBuilder): Missing values in present node are filled in with +inf (truncated). - In case of the root node, the values are [0, 0, 0, 0, distance from agent to target]. + In case of the root node, the values are [0, 0, 0, 0, distance from agent to target, own malfunction, own speed] In case the target node is reached, the values are [0, 0, 0, 0, 0]. """