Commit 8a9ba90d authored by u214892's avatar u214892
Browse files

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

parent c1057171
Pipeline #1120 passed with stage
in 11 minutes and 35 seconds
......@@ -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
......
......@@ -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)
......
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