diff --git a/examples/simple_example_1.py b/examples/simple_example_1.py index daf56a9bd54ecb553a707ddceab7900219f7d7d4..93d5268840c4980aea40ae3a18b35f2bed31260e 100644 --- a/examples/simple_example_1.py +++ b/examples/simple_example_1.py @@ -20,6 +20,5 @@ env.reset() env_renderer = RenderTool(env) env_renderer.render_env(show=True) -env_renderer.render_env(show=True) input("Press Enter to continue...") diff --git a/examples/simple_example_2.py b/examples/simple_example_2.py index be6fdef33fe058b938c69330cbde5f9925875c00..f8612a392d0c6d57f6077dc988379e927af1e1b1 100644 --- a/examples/simple_example_2.py +++ b/examples/simple_example_2.py @@ -34,6 +34,5 @@ env.reset() env_renderer = RenderTool(env, gl="PIL") env_renderer.render_env(show=True) -env_renderer.render_env(show=True) input("Press Enter to continue...") diff --git a/examples/simple_example_3.py b/examples/simple_example_3.py index 3a1c583fc05cb026c177b1a2dad3a055cea16f6d..5aa03d8f95a7079b708baea1e2ddce27e9a46554 100644 --- a/examples/simple_example_3.py +++ b/examples/simple_example_3.py @@ -23,7 +23,6 @@ for i in range(env.get_num_agents()): env_renderer = RenderTool(env) env_renderer.render_env(show=True, frames=True) -env_renderer.render_env(show=True, frames=True) print("Manual control: s=perform step, q=quit, [agent id] [1-2-3 action] \ (turnleft+move, move to front, turnright+move)") diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py index 1e1a7c157b061705bad62bc11a6da46ec983363e..bc24a214676e701a6159ff7d932bb5a289245fcf 100644 --- a/flatland/utils/rendertools.py +++ b/flatland/utils/rendertools.py @@ -57,6 +57,17 @@ class RenderTool(object): self.new_rail = True self.update_background() + def reset(self): + """ + Resets the environment + :return: + """ + self.set_new_rail() + self.frame_nr = 0 + self.start_time = time.time() + self.times_list = deque() + return + def update_background(self): # create background map targets = {}