From 13ebd00961b28eb1ad8aa9237baf560ac14c36b6 Mon Sep 17 00:00:00 2001
From: Giacomo Spigler <spiglerg@gmail.com>
Date: Sun, 21 Apr 2019 05:17:14 +0200
Subject: [PATCH] fixed temporary_example
---
examples/temporary_example.py | 11 ++++++-----
1 file changed, 6 insertions(+), 5 deletions(-)
diff --git a/examples/temporary_example.py b/examples/temporary_example.py
index 97d41b53..1d13d2ce 100644
--- a/examples/temporary_example.py
+++ b/examples/temporary_example.py
@@ -27,7 +27,7 @@ transition_probability = [1.0, # empty cell - Case 0
0.1, # Case 5 - double slip
0.2, # Case 6 - symmetrical
0.01] # Case 7 - dead end
-"""
+
# Example generate a random rail
env = RailEnv(width=20,
height=20,
@@ -43,6 +43,7 @@ env_renderer.renderEnv(show=True)
specs = [[(0, 0), (0, 0), (0, 0), (0, 0), (7, 0), (0, 0)],
[(7, 270), (1, 90), (1, 90), (1, 90), (2, 90), (7, 90)]]
+"""
env = RailEnv(width=6,
height=2,
rail_generator=rail_from_manual_specifications_generator(specs),
@@ -55,16 +56,16 @@ env.agents_target[0] = [1, 1]
env.agents_direction[0] = 1
# TODO: watch out: if these variables are overridden, the obs_builder object has to be reset, too!
env.obs_builder.reset()
-"""
+#"""
env = RailEnv(width=7,
height=7,
rail_generator=random_rail_generator(cell_type_relative_proportion=transition_probability),
- number_of_agents=1)
+ number_of_agents=2)
# TODO: delete next line
-for i in range(4):
- print(env.obs_builder.distance_map[0, :, :, i])
+#for i in range(4):
+# print(env.obs_builder.distance_map[0, :, :, i])
obs, all_rewards, done, _ = env.step({0:0})
for i in range(env.number_of_agents):
--
GitLab