From b13b719bbaaad72b1badcac170c7f1f247281c25 Mon Sep 17 00:00:00 2001
From: Nilabha <nilabha2007@gmail.com>
Date: Fri, 6 Aug 2021 16:47:59 +0530
Subject: [PATCH] update flatland example as per latest examples
---
...duction_flatland_2_1.py => introduction_flatland_3.py} | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
rename examples/{introduction_flatland_2_1.py => introduction_flatland_3.py} (97%)
diff --git a/examples/introduction_flatland_2_1.py b/examples/introduction_flatland_3.py
similarity index 97%
rename from examples/introduction_flatland_2_1.py
rename to examples/introduction_flatland_3.py
index d770fa43..3bb8c2b1 100644
--- a/examples/introduction_flatland_2_1.py
+++ b/examples/introduction_flatland_3.py
@@ -12,7 +12,7 @@ from flatland.envs.rail_env import RailEnv
from flatland.envs.rail_env import RailEnvActions
from flatland.envs.rail_generators import sparse_rail_generator
#from flatland.envs.sparse_rail_gen import SparseRailGen
-from flatland.envs.schedule_generators import sparse_schedule_generator
+from flatland.envs.line_generators import sparse_line_generator
# We also include a renderer because we want to visualize what is going on in the environment
from flatland.utils.rendertools import RenderTool, AgentRenderVariant
@@ -45,7 +45,7 @@ rail_generator = sparse_rail_generator(max_num_cities=cities_in_map,
seed=seed,
grid_mode=grid_distribution_of_cities,
max_rails_between_cities=max_rails_between_cities,
- max_rails_in_city=max_rail_in_cities,
+ max_rail_pairs_in_city=max_rail_in_cities,
)
#rail_generator = SparseRailGen(max_num_cities=cities_in_map,
@@ -68,7 +68,7 @@ speed_ration_map = {1.: 0.25, # Fast passenger train
# We can now initiate the schedule generator with the given speed profiles
-schedule_generator = sparse_schedule_generator(speed_ration_map)
+line_generator = sparse_line_generator(speed_ration_map)
# We can furthermore pass stochastic data to the RailEnv constructor which will allow for stochastic malfunctions
# during an episode.
@@ -87,7 +87,7 @@ observation_builder = GlobalObsForRailEnv()
env = RailEnv(width=width,
height=height,
rail_generator=rail_generator,
- schedule_generator=schedule_generator,
+ line_generator=line_generator,
number_of_agents=nr_trains,
obs_builder_object=observation_builder,
#malfunction_generator_and_process_data=malfunction_from_params(stochastic_data),
--
GitLab