"removing enforce exploitatio"

scioip34 · May 20, 2024 · 6cb83f3 · 6cb83f3
1 parent bd68303
commit 6cb83f3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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