diff --git a/examples/temporary_example.py b/examples/temporary_example.py index 28a94db3754fc21783447cf15057022476189a63..c2720a9324cbf2761bae9031c8b22f16abf7c4be 100644 --- a/examples/temporary_example.py +++ b/examples/temporary_example.py @@ -17,7 +17,11 @@ transition_probability = [1.0, # empty cell - Case 0 0.5, # Case 4 - single slip 0.5, # Case 5 - double slip 0.2, # Case 6 - symmetrical - 0.0] # Case 7 - dead end + 0.0, # Case 7 - dead end + 0.2, # Case 8 - turn left + 0.2, # Case 9 - turn right + 1.0] # Case 10 - mirrored switch + """ # Example generate a random rail env = RailEnv(width=20, diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py index 2c26076dd527c9e315366471274b20f021502881..de75b70b62145a911d64effa3dbad162c928289c 100644 --- a/flatland/envs/generators.py +++ b/flatland/envs/generators.py @@ -227,7 +227,7 @@ def generate_rail_from_list_of_manual_specifications(list_of_specifications) """ -def random_rail_generator(cell_type_relative_proportion=[1.0] * 8): +def random_rail_generator(cell_type_relative_proportion=[1.0] * 11): """ Dummy random level generator: - fill in cells at random in [width-2, height-2] @@ -266,7 +266,10 @@ def random_rail_generator(cell_type_relative_proportion=[1.0] * 8): transitions_templates_ = [] transition_probabilities = [] - for i in range(len(t_utils.transitions) - 4): # don't include dead-ends + for i in range(len(t_utils.transitions)): # don't include dead-ends + if t_utils.transitions[i] == int('0010000000000000', 2): + continue + all_transitions = 0 for dir_ in range(4): trans = t_utils.get_transitions(t_utils.transitions[i], dir_)