demo.py imports AgentRenderVariant

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