From 8e13228ec8ce332273e07302d5ca67f0a964a61c Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Fri, 14 Aug 2020 19:44:23 +0200
Subject: [PATCH] 14.08.2020

---
 src/extra.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/extra.py b/src/extra.py
index 5b6af80..b70830f 100644
--- a/src/extra.py
+++ b/src/extra.py
@@ -352,7 +352,7 @@ class Extra(ObservationBuilder):
         return action_dict
 
     def rl_agent_act(self, observation, info, eps=0.0):
-        if len(self.random_agent_starter) != len(self.env.get_num_agents()):
+        if len(self.random_agent_starter) != self.env.get_num_agents():
             self.random_agent_starter = np.random.random(self.env.get_num_agents()) * 1000.0
             self.loadAgent()
 
-- 
GitLab