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