diff --git a/flatland/utils/simple_rail.py b/flatland/utils/simple_rail.py index a12c26e66fdf5a9ff102bb79440bd4f4b805e819..fa5d8ef4a3388afd4d63c5ab400d9055f4762363 100644 --- a/flatland/utils/simple_rail.py +++ b/flatland/utils/simple_rail.py @@ -56,6 +56,7 @@ def make_disconnected_simple_rail() -> Tuple[GridTransitionMap, np.array]: # | # | # | + transitions = RailEnvTransitions() cells = transitions.transition_list empty = cells[0]