Documentation - Missing doc string

https://gitlab.aicrowd.com/flatland/flatland/blob/master/flatland/envs/distance_map.py#L58

Missing doc string of function _compute.