3 files changed, 8 insertions, 4 deletions
diff --git a/python/agent_AB.py b/python/agent_AB.py
index 4b1ace1..5564f11 100644
--- a/python/agent_AB.py
+++ b/python/agent_AB.py
@@ -1,10 +1,10 @@
+from python.alphaBeta import AlphaBeta
 import dotsandboxes.dotsandboxesagent as dba
 
 import sys
 import argparse
 import logging
 from GameState import GameState, DotsAndBoxesState
-from MCTS import MCTSNode, MCTSGameController
 import alphaBeta
 
 
@@ -18,7 +18,7 @@ class Agent(dba.DotsAndBoxesAgent):
         super(Agent, self).__init__(player, nb_rows, nb_cols, timelimit)
         self.GameStateClass = DotsAndBoxesState
         self.game_state = self.GameStateClass(nb_rows, nb_cols, player)
-        # self.controller = controller
+        self.controller = AlphaBeta()
 
     def register_action(self, row, column, orientation, player):
         super(Agent, self).register_action(row, column, orientation, player)
diff --git a/python/agent_MCTS.py b/python/agent_MCTS.py
index b0072cf..b60f5ec 100644
--- a/python/agent_MCTS.py
+++ b/python/agent_MCTS.py
@@ -5,8 +5,6 @@ import argparse
 import logging
 from GameState import GameState, DotsAndBoxesState
 from MCTS import MCTSNode, MCTSGameController
-import alphaBeta
-
 
 logger = logging.getLogger(__name__)
 games = {}
diff --git a/python/alphaBeta.py b/python/alphaBeta.py
index 31fa578..8e041fe 100644
--- a/python/alphaBeta.py
+++ b/python/alphaBeta.py
@@ -1,3 +1,9 @@
+from GameState import GameState
+class GameController(object):
+    def get_next_move(self, state):
+        # when you get a new move, it is assumed that the game is not ended yet
+        assert state.get_moves()
+
 
 def alpha_beta(node, alpha, beta):