flatland.envs.distance_map module

flatland.envs.distance_map module#

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

Bases: object

get() ndarray[source]#

Get the distance map

reset(agents: List[EnvAgent], rail: GridTransitionMap)[source]#

Reset the distance map

set(distance_map: ndarray)[source]#

Set the distance map