flatland.envs.distance_map module¶

class flatland.envs.distance_map.DistanceMap(agents: List[flatland.envs.agent_utils.EnvAgent], env_height: int, env_width: int)[source]¶

Bases: object

get(self) → numpy.ndarray[source]¶: Get the distance map

reset(self, agents:List[flatland.envs.agent_utils.EnvAgent], rail:flatland.core.transition_map.GridTransitionMap)[source]¶: Reset the distance map

set(self, distance_map:numpy.ndarray)[source]¶: Set the distance map