diff --git a/examples/custom_observation_example_02_SingleAgentNavigationObs.py b/examples/custom_observation_example_02_SingleAgentNavigationObs.py index 52a56b06dcdeb9bebeb9dbb70d5acc9ae0c350bb..c4e415b065effd519f8b769b464227bc9a5821dc 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 ac99835368fc4a5a709e72234396a286f1d622a7..b4830a957ebda30d061ccbd1f8233c8977e78707 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():