Include policy id in compute actions

agents/neural_baseline_agent.py

@@ -13,13 +13,12 @@ class NeuralBaselineAgent(NeuralMMOAgent):
 
     def register_reset(self, observations):
         obs = {0:observations}
-        actions,self.state,_ = self.trainer.compute_actions(obs)
+        actions,self.state,_ = self.trainer.compute_actions(obs,state={},policy_id='policy_0')
         # action = self.get_action(observations)
         return actions[0]
 
     def compute_action(self, observations, info=None):
         obs = {0:observations}
-        actions,self.state,_ = self.trainer.compute_actions(obs)
+        actions,self.state,_ = self.trainer.compute_actions(obs,state={},policy_id='policy_0')
         # action = self.get_action(observations)
         return actions[0]
-        
\ No newline at end of file