random agent runs

aypan17 · Feb 15, 2024 · 0a4b3d2 · 0a4b3d2
1 parent b518aa2
commit 0a4b3d2
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/machiavelli/agent/random_agent.py b/machiavelli/agent/random_agent.py
@@ -0,0 +1,22 @@
+from machiavelli.agent.base_agent import BaseAgent
+import numpy as np
+
+
+class RandomAgent(BaseAgent):
+    def __init__(self):
+        super().__init__()
+
+    def get_action(self, obs, rew, done, info):
+        if done:
+            return None
+        return np.random.choice(info['game_state']['num_choices'])
+
+    def get_action_probabilities(self, obs, rew, done, info):
+        num_actions = info['game_state']['num_choices']
+        return [1 / num_actions for _ in range(num_actions)]
+
+    def train(self):
+        pass
+
+    def test(self):
+        pass
diff --git a/machiavelli/game/machiavelli_env.py b/machiavelli/game/machiavelli_env.py
@@ -260,7 +260,7 @@ def _get_info(self):
     @property
     def action_space(self):
         # We have N choices at each state, but N changes depending on the state
-        action_space = gym.spaces.Discrete(self.player.get_action_space())
+        action_space = gym.spaces.Discrete(max(self.player.get_action_space(), 1))
         return action_space
 
     @property