From d36550c95ccbf87b4f777963de88d1e9a4637933 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 2 Jul 2019 08:21:28 +0200
Subject: [PATCH] demo.py imports AgentRenderVariant
---
examples/demo.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/examples/demo.py b/examples/demo.py
index 7dd358a8..5974e2ae 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
--
GitLab