diff --git a/examples/simple_example_2.py b/examples/simple_example_2.py index 1d2c1e6d72be9be840459cfed1faf1dad1d261e9..9b5f193856c96fe5aac7c06ad24e0f83f5b8b562 100644 --- a/examples/simple_example_2.py +++ b/examples/simple_example_2.py @@ -32,7 +32,7 @@ env = RailEnv(width=10, env.reset() -env_renderer = RenderTool(env, gl="PILSVG") +env_renderer = RenderTool(env, gl="PIL") env_renderer.renderEnv(show=True) env_renderer.renderEnv(show=True) diff --git a/examples/simple_example_3.py b/examples/simple_example_3.py index 1661ef65a9a33f3b44a098caaf83317919722398..eb7ac9ab820199f71db2d0a5af25a2dd636c614e 100644 --- a/examples/simple_example_3.py +++ b/examples/simple_example_3.py @@ -2,7 +2,7 @@ import random import numpy as np -from flatland.envs.generators import random_rail_generator +from flatland.envs.generators import random_rail_generator, complex_rail_generator from flatland.envs.observations import TreeObsForRailEnv from flatland.envs.rail_env import RailEnv from flatland.utils.rendertools import RenderTool @@ -12,7 +12,7 @@ np.random.seed(1) env = RailEnv(width=7, height=7, - rail_generator=random_rail_generator(), + rail_generator=complex_rail_generator(nr_start_goal=10, nr_extra=1, min_dist=8, max_dist=99999, seed=0), number_of_agents=2, obs_builder_object=TreeObsForRailEnv(max_depth=2)) @@ -21,7 +21,7 @@ obs, all_rewards, done, _ = env.step({0: 0}) for i in range(env.get_num_agents()): env.obs_builder.util_print_obs_subtree(tree=obs[i], num_features_per_node=7) -env_renderer = RenderTool(env, gl="PIL") +env_renderer = RenderTool(env, gl="PILSVG") env_renderer.renderEnv(show=True, frames=True) env_renderer.renderEnv(show=True, frames=True)