66 shortest-path-predictor: cleanup and unit test; not working yet

flatland/envs/predictions.py

@@ -31,7 +31,7 @@ class DummyPredictorForRailEnv(PredictionBuilder):
         Returns
         -------
         np.array
-            Returns a dictionary indexed by the agent handle and for each agent a vector of (max_depth + 1) x 5 elements:
+            Returns a dictionary indexed by the agent handle and for each agent a vector of (max_depth + 1)x5 elements:
             - time_offset
             - position axis 0
             - position axis 1
@@ -101,7 +101,7 @@ class ShortestPathPredictorForRailEnv(PredictionBuilder):
         Returns
         -------
         np.array
-            Returns a dictionary indexed by the agent handle and for each agent a vector of (max_depth + 1) x 5 elements:
+            Returns a dictionary indexed by the agent handle and for each agent a vector of (max_depth + 1)x5 elements:
             - time_offset
             - position axis 0
             - position axis 1

tests/test_env_prediction_builder.py

@@ -166,7 +166,7 @@ def test_shortest_path_predictor(rendering=False):
     if rendering:
         renderer = RenderTool(env, gl="PILSVG")
         renderer.renderEnv(show=True, show_observations=False)
-        # input("Continue?")
+        input("Continue?")
 
     agent = env.agents[0]
     assert agent.position == (5, 6)