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