From f2fd71be53830fbaacf302c45d4241dce06d2700 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Wed, 29 May 2019 10:59:42 +0200
Subject: [PATCH] demo cleaned up
---
examples/demo.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/examples/demo.py b/examples/demo.py
index 8d128ea5..9311159b 100644
--- a/examples/demo.py
+++ b/examples/demo.py
@@ -71,7 +71,7 @@ class Scenario_Generator:
print("File does not exist:", filename, " Working directory: ", os.getcwd())
return env
-
+
class Demo:
def __init__(self, env):
@@ -106,6 +106,8 @@ class Demo:
# Action
for iAgent in range(self.env.get_num_agents()):
+
+
action = 2
if False:
@@ -125,9 +127,10 @@ class Demo:
trial += 1
if trial > 10:
break
+
action_dict.update({iAgent: action})
- self.renderer.renderEnv(show=True, action_dict=action_dict)
+ self.renderer.renderEnv(show=True)
# Environment step
next_obs, all_rewards, done, _ = self.env.step(action_dict)
--
GitLab