gameplay stuff added

mahi97 · Oct 26, 2021 · 7a3669e · 7a3669e
1 parent 375b37b
commit 7a3669e
Show file tree

Hide file tree

Showing 4 changed files with 313 additions and 586 deletions.
diff --git a/pz_risk/arena.py b/pz_risk/arena.py
@@ -1,13 +1,14 @@
 from risk_env import env
-from agents.greedy import GreedyAgent
-from agents.random import RandomAgent
+from agents import GreedyAgent, RandomAgent, ModelAgent
 from loguru import logger
+import matplotlib.pyplot as plt
 
 e = env()
 e.reset()
 
 players = [GreedyAgent(i) for i in range(2)]
-players += [RandomAgent(2 + i) for i in range(4)]
+players += [GreedyAgent(2 + i) for i in range(2)]
+players += [RandomAgent(4 + i) for i in range(2)]
 winner = -1
 for agent in e.agent_iter():
     obs, rew, done, info = e.last()