diff --git a/README.rst b/README.rst index ad9eb84f9ccc33cbbe2d56a9d8da57329e8ce7e6..2b6b31022db7fae672f30d81078d3d2c63ce0796 100644 --- a/README.rst +++ b/README.rst @@ -147,7 +147,7 @@ Basic usage of the RailEnv environment used by the Flatland Challenge seed=0), number_of_agents=2) - env_renderer = RenderTool(env, gl="PILSVG") + env_renderer = RenderTool(env) for step in range(100): obs, all_rewards, done, _ = env.step( diff --git a/examples/demo.py b/examples/demo.py index 7dd358a8716484c06cc8f4a4a97d289c5d7c3716..768f44d368695a58f4041f9e465da30e53773133 100644 --- a/examples/demo.py +++ b/examples/demo.py @@ -75,7 +75,7 @@ class Demo: self.record_frames = record_frames def create_renderer(self): - self.renderer = RenderTool(self.env, gl="PILSVG") + self.renderer = RenderTool(self.env) handle = self.env.get_agent_handles() return handle diff --git a/examples/play_model.py b/examples/play_model.py index 6997b2c002582d35520362ac50555aae0641c9ec..c44118c2ce3702ad70a2a7612de18e1a831e7042 100644 --- a/examples/play_model.py +++ b/examples/play_model.py @@ -93,7 +93,7 @@ def max_lt(seq, val): return None -def main(render=True, delay=0.0, n_trials=3, n_steps=50, sGL="PILSVG"): +def main(render=True, delay=0.0, n_trials=3, n_steps=50): random.seed(1) np.random.seed(1) @@ -103,7 +103,7 @@ def main(render=True, delay=0.0, n_trials=3, n_steps=50, sGL="PILSVG"): number_of_agents=5) if render: - env_renderer = RenderTool(env, gl=sGL) + env_renderer = RenderTool(env) oPlayer = Player(env) diff --git a/examples/simple_example_1.py b/examples/simple_example_1.py index 89536edde93ba5d593d710290657c555bc542c5b..633c87b527814c7e13514aba17222350f9070d26 100644 --- a/examples/simple_example_1.py +++ b/examples/simple_example_1.py @@ -18,7 +18,7 @@ env = RailEnv(width=6, env.reset() -env_renderer = RenderTool(env, gl="PILSVG") +env_renderer = RenderTool(env) 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 ddb98e9979c059f3a08840a7d229448c83b839d2..9055dd4ccc5adf75e673e6d3685daec4304f3e9e 100644 --- a/examples/simple_example_3.py +++ b/examples/simple_example_3.py @@ -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="PILSVG") +env_renderer = RenderTool(env) env_renderer.renderEnv(show=True, frames=True) env_renderer.renderEnv(show=True, frames=True) diff --git a/examples/tkplay.py b/examples/tkplay.py index 7ab1c3234d31d5c083b068fec5aa71b47cbaa4e9..00a4fc70613a5f86faa6e65bc19ca9a988b2b7f0 100644 --- a/examples/tkplay.py +++ b/examples/tkplay.py @@ -1,4 +1,8 @@ -from examples.play_model import Player +try: + from examples.play_model import Player +except ImportError: + from play_model import Player + from flatland.envs.generators import complex_rail_generator from flatland.envs.rail_env import RailEnv from flatland.utils.rendertools import RenderTool