Skip to content
Snippets Groups Projects
Commit 18cbc741 authored by u229589's avatar u229589
Browse files

Refactoring: bugfix of return type

parent ff83fdeb
No related branches found
No related tags found
No related merge requests found
......@@ -640,7 +640,7 @@ class LocalObsForRailEnv(ObservationBuilder):
direction = np.identity(4)[agent.direction]
return local_rail_obs, obs_map_state, obs_other_agents_state, direction
def get_many(self, handles: Optional[List[int]] = None) -> Dict[int, (np.ndarray, np.ndarray, np.ndarray, np.ndarray)]:
def get_many(self, handles: Optional[List[int]] = None) -> Dict[int, tuple]:
"""
Called whenever an observation has to be computed for the `env` environment, for each agent with handle
in the `handles` list.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment