diff --git a/examples/demo.py b/examples/demo.py index 7dd358a8716484c06cc8f4a4a97d289c5d7c3716..5974e2aef7f923f3a0dcd224cfc67517fea83d67 100644 --- a/examples/demo.py +++ b/examples/demo.py @@ -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