diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index 89cae88f60195aa8752fe39607f8118678395e55..7694bb207a0610deb50777346ec4d00fcb55945a 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.