diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index 16a7743d7167ed85241bf2357d03b24d902289b3..8cd482c08bea3f72630fed3ea97c59583cbd304c 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -25,6 +25,7 @@ class RailEnvActions(IntEnum): MOVE_RIGHT = 3 STOP_MOVING = 4 + class RailEnv(Environment): """ RailEnv environment class.