From 8a6e86fff91390ea0ab69216850b15662aa554d6 Mon Sep 17 00:00:00 2001
From: Giacomo Spigler <spiglerg@gmail.com>
Date: Fri, 28 Jun 2019 15:56:25 +0200
Subject: [PATCH] updated examples
---
examples/simple_example_2.py | 2 +-
examples/simple_example_3.py | 6 +++---
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/examples/simple_example_2.py b/examples/simple_example_2.py
index 1d2c1e6d..9b5f1938 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 1661ef65..eb7ac9ab 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)
--
GitLab