Q&D

src/extra.py

@@ -339,7 +339,7 @@ class Extra(ObservationBuilder):
 
         return observation
 
-    def rl_agent_act(self, observation, info, eps=0.0):
+    def rl_agent_act_ADRIAN(self, observation, info, eps=0.0):
         self.loadAgent()
         action_dict = {}
         for a in range(self.env.get_num_agents()):
@@ -351,7 +351,7 @@ class Extra(ObservationBuilder):
 
         return action_dict
 
-    def rl_agent_act_ADRIAN(self, observation, info, eps=0.0):
+    def rl_agent_act(self, observation, info, eps=0.0):
         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()