From 3041fa0777b4c08a87eddd95b1b38311165de3a9 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Thu, 18 Jul 2019 14:44:39 -0400
Subject: [PATCH] Cleaned up examples folder and moved unused examles to
internal repo
---
examples/simple_example_1.py | 6 ++----
examples/simple_example_2.py | 4 +---
2 files changed, 3 insertions(+), 7 deletions(-)
diff --git a/examples/simple_example_1.py b/examples/simple_example_1.py
index 93d52688..7956c34f 100644
--- a/examples/simple_example_1.py
+++ b/examples/simple_example_1.py
@@ -1,5 +1,4 @@
from flatland.envs.generators import rail_from_manual_specifications_generator
-from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.rail_env import RailEnv
from flatland.utils.rendertools import RenderTool
@@ -13,12 +12,11 @@ specs = [[(0, 0), (0, 0), (0, 0), (0, 0), (0, 0), (0, 0)],
env = RailEnv(width=6,
height=4,
rail_generator=rail_from_manual_specifications_generator(specs),
- number_of_agents=1,
- obs_builder_object=TreeObsForRailEnv(max_depth=2))
+ number_of_agents=1)
env.reset()
env_renderer = RenderTool(env)
-env_renderer.render_env(show=True)
+env_renderer.render_env(show=True, show_predictions=False, show_observations=False)
input("Press Enter to continue...")
diff --git a/examples/simple_example_2.py b/examples/simple_example_2.py
index f8612a39..994c7ded 100644
--- a/examples/simple_example_2.py
+++ b/examples/simple_example_2.py
@@ -3,7 +3,6 @@ import random
import numpy as np
from flatland.envs.generators import random_rail_generator
-from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.rail_env import RailEnv
from flatland.utils.rendertools import RenderTool
@@ -27,8 +26,7 @@ transition_probability = [1.0, # empty cell - Case 0
env = RailEnv(width=10,
height=10,
rail_generator=random_rail_generator(cell_type_relative_proportion=transition_probability),
- number_of_agents=3,
- obs_builder_object=TreeObsForRailEnv(max_depth=2))
+ number_of_agents=3)
env.reset()
--
GitLab