diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py index 95e5aeaf3d969c22bcf66ee1138245745d6fe1d8..2187c598ed94684cf5380901baa6540e244dbb75 100644 --- a/flatland/envs/observations.py +++ b/flatland/envs/observations.py @@ -28,6 +28,9 @@ class TreeObsForRailEnv(ObservationBuilder): 'num_agents_malfunctioning ' 'speed_min_fractional ' 'childs') + + tree_explorted_actions_char = ['L', 'F', 'R', 'B'] + """ TreeObsForRailEnv object. @@ -46,7 +49,6 @@ class TreeObsForRailEnv(ObservationBuilder): self.location_has_agent_direction = {} self.predictor = predictor self.location_has_target = None - self.tree_explorted_actions_char = ['L', 'F', 'R', 'B'] def reset(self): self.location_has_target = {tuple(agent.target): 1 for agent in self.env.agents}