diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py index b6e6c916562ccedd45d5694ce1fd185579ef46d3..b1f46ec638d8a865a2ce36a9715d89b452c782f2 100644 --- a/flatland/envs/observations.py +++ b/flatland/envs/observations.py @@ -678,10 +678,10 @@ class LocalObsForRailEnv(ObservationBuilder): with dimensions (2*view_radius + 1, 2*view_radius + 1, 16), assuming 16 bits encoding of transitions. - - Two 2D arrays containing respectively, if they are in the agent's vision range, - its target position, the positions of the other targets. + - Two 2D arrays (2*view_radius + 1, 2*view_radius + 1, 2) containing respectively, + if they are in the agent's vision range, its target position, the positions of the other targets. - - A 3D array (map_height, map_width, 4) containing the one hot encoding of directions + - A 3D array (2*view_radius + 1, 2*view_radius + 1, 4) containing the one hot encoding of directions of the other agents at their position coordinates, if they are in the agent's vision range. - A 4 elements array with one hot encoding of the direction.