From d8e20bc047ea3f8eb69d7af0b349e495f91cf842 Mon Sep 17 00:00:00 2001
From: flaurent <florian.laurent@gmail.com>
Date: Thu, 11 Jun 2020 12:39:22 +0200
Subject: [PATCH] Fixed code examples
---
...ustom_observation_example_02_SingleAgentNavigationObs.py | 2 +-
.../custom_observation_example_03_ObservePredictions.py | 6 +++++-
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/examples/custom_observation_example_02_SingleAgentNavigationObs.py b/examples/custom_observation_example_02_SingleAgentNavigationObs.py
index 52a56b06..c4e415b0 100644
--- a/examples/custom_observation_example_02_SingleAgentNavigationObs.py
+++ b/examples/custom_observation_example_02_SingleAgentNavigationObs.py
@@ -82,7 +82,7 @@ def main(args):
number_of_agents=1, obs_builder_object=SingleAgentNavigationObs())
obs, info = env.reset()
- env_renderer = RenderTool(env, gl="PILSVG")
+ env_renderer = RenderTool(env)
env_renderer.render_env(show=True, frames=True, show_observations=True)
for step in range(100):
action = np.argmax(obs[0]) + 1
diff --git a/examples/custom_observation_example_03_ObservePredictions.py b/examples/custom_observation_example_03_ObservePredictions.py
index ac998353..b4830a95 100644
--- a/examples/custom_observation_example_03_ObservePredictions.py
+++ b/examples/custom_observation_example_03_ObservePredictions.py
@@ -128,7 +128,7 @@ def main(args):
number_of_agents=3, obs_builder_object=custom_obs_builder)
obs, info = env.reset()
- env_renderer = RenderTool(env, gl="PILSVG")
+ env_renderer = RenderTool(env)
# We render the initial step and show the obsered cells as colored boxes
env_renderer.render_env(show=True, frames=True, show_observations=True, show_predictions=False)
@@ -144,6 +144,10 @@ def main(args):
if sleep_for_animation:
time.sleep(0.5)
+ if done["__all__"]:
+ print("All done!")
+ break
+
if __name__ == '__main__':
if 'argv' in globals():
--
GitLab