diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py index 5a19deeb07513203a787b48c685b4c461b8c4e05..1f02d518a9714de91d8910b8cb1408f25eb3fe88 100644 --- a/flatland/envs/observations.py +++ b/flatland/envs/observations.py @@ -466,8 +466,6 @@ class TreeObsForRailEnv(ObservationBuilder): ] # ############################# # ############################# - - new_root_observation = observation[:] # Start from the current orientation, and see which transitions are available; # organize them as [left, forward, right, back], relative to the current orientation # Get the possible transitions diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index d2b47911ace505ee5ebf0f73e13749c9f810c57d..c70505505a4406e59d10472bc8e6eb89e9bd65a8 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -4,7 +4,6 @@ Definition of the RailEnv environment and related level-generation functions. Generator functions are functions that take width, height and num_resets as arguments and return a GridTransitionMap object. """ -# TODO: _ this is a global method --> utils or remove later from enum import IntEnum @@ -85,7 +84,6 @@ class RailEnv(Environment): a GridTransitionMap object rail_from_manual_sp ecifications_generator(rail_spec) : generate a rail from a rail specifications array - TODO: generate_rail_from_saved_list or from list of ndarray bitmaps --- width : int The width of the rail map. Potentially in the future, a range of widths to sample from. diff --git a/tests/test_flatland_core_transition_map.py b/tests/test_flatland_core_transition_map.py index 2f721ca24bd6c7f79e66c8cd75b75b5916a2f59d..8013e912d93ea1125f4f7bdc622f2f4e3d4b2333 100644 --- a/tests/test_flatland_core_transition_map.py +++ b/tests/test_flatland_core_transition_map.py @@ -20,6 +20,4 @@ def test_grid8_set_transitions(): grid8_map.set_transition((0, 0, Grid8TransitionsEnum.NORTH), Grid8TransitionsEnum.NORTH, 0) assert grid8_map.get_transitions((0, 0, Grid8TransitionsEnum.NORTH)) == (0, 0, 0, 0, 0, 0, 0, 0) - - # TODO GridTransitionMap