From 1f50a25ceb35d902a7cd3ecc140bcbbb26d84cca Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Tue, 9 Jul 2019 14:23:37 -0400
Subject: [PATCH] fixed fomatting

---
 flatland/envs/observations.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index b6e6c91..b1f46ec 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.
-- 
GitLab