diff --git a/abm/projects/madrl_foraging/madrl_agent/brain.py b/abm/projects/madrl_foraging/madrl_agent/brain.py index 291f897..2679863 100644 --- a/abm/projects/madrl_foraging/madrl_agent/brain.py +++ b/abm/projects/madrl_foraging/madrl_agent/brain.py @@ -130,8 +130,7 @@ def select_action(self, state): if len(self.legal_actions)==1: self.action_tensor = torch.LongTensor([[0]]).to(device) - if 1 in self.legal_actions: - self.action_tensor = torch.LongTensor([[1]]).to(device) + else: # Epsilon-greedy exploration