diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py index a9a9b9dc664bd25b43b378b953b8f0fcd14de008..3cb29a3f2fac6755b59f0b12fe2906e3ea2efa60 100644 --- a/flatland/envs/observations.py +++ b/flatland/envs/observations.py @@ -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.