Skip to content
Snippets Groups Projects
Commit 5a5d7980 authored by adrian_egli's avatar adrian_egli
Browse files

Merge branch 'feature/rendering_one_step_behind_fix' into 'master'

demo.py imports AgentRenderVariant

See merge request flatland/flatland!87
parents 2912e9f3 d36550c9
No related branches found
No related tags found
No related merge requests found
......@@ -7,7 +7,7 @@ import numpy as np
from flatland.envs.generators import complex_rail_generator
from flatland.envs.generators import random_rail_generator
from flatland.envs.rail_env import RailEnv
from flatland.utils.rendertools import RenderTool
from flatland.utils.rendertools import RenderTool, AgentRenderVariant
# ensure that every demo run behave constantly equal
random.seed(1)
......@@ -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, gl="PILSVG", agentRenderVariant=AgentRenderVariant.AGENT_SHOWS_OPTIONS)
handle = self.env.get_agent_handles()
return handle
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment