Start work on AB

1 files changed, 2 insertions, 2 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)