diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index 1664c3155c61ad2ddde8244440edf30d48f3e410..4735f9ab44ec19fc544daa2c91cc9ba0533317c9 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -388,7 +388,7 @@ class RailEnv(Environment): def __init__(self, width, height, - rail_generator=random_rail_generator, + rail_generator=random_rail_generator(), number_of_agents=1, obs_builder_object=TreeObsForRailEnv(max_depth=2)): """