Remove limited depth minimax and add ab pruning to roadmap

zacharyfmarion · Dec 29, 2018 · 2536a2e · 2536a2e
1 parent 41dd81e
commit 2536a2e
Show file tree

Hide file tree

Showing 7 changed files with 5 additions and 34 deletions.
diff --git a/README.md b/README.md
@@ -52,6 +52,7 @@ Build the project with `python3 setup.py sdist bdist_wheel`.
 ## Roadmap
 
 - [x] Add Monte Carlo Tree Search
+- [ ] Add Alpha Beta Pruning
 - [ ] Add Othello as a game
 - [ ] Add Chess as a game
 - [ ] Add a simple RL agent & trainer

diff --git a/gameai/agents/__init__.py b/gameai/agents/__init__.py
@@ -1,8 +1,7 @@
 from .agent import Agent
 from .trainable_agent import TrainableAgent
 
-from .human import HumanAgent
-from .limited_depth_minimax import LimitedDepthMinimaxAgent
-from .mcts import MCTSAgent
-from .minimax import MinimaxAgent
+from .human_agent import HumanAgent
+from .mcts_agent import MCTSAgent
+from .minimax_agent import MinimaxAgent
 from .random_agent import RandomAgent
diff --git a/gameai/agents/human.py → gameai/agents/human_agent.py b/gameai/agents/human.py → gameai/agents/human_agent.py
diff --git a/gameai/agents/mcts.py → gameai/agents/mcts_agent.py b/gameai/agents/mcts.py → gameai/agents/mcts_agent.py
diff --git a/gameai/agents/minimax.py b/gameai/agents/minimax.py
diff --git a/gameai/agents/limited_depth_minimax.py → gameai/agents/minimax_agent.py b/gameai/agents/limited_depth_minimax.py → gameai/agents/minimax_agent.py
@@ -1,12 +1,10 @@
 from .agent import Agent
 
 
-class LimitedDepthMinimaxAgent(Agent):
+class MinimaxAgent(Agent):
     '''
     Implementation of minimax which allows you to specify a cutoff horizon for the
     search.
-
-    TODO: Don't really need a separate agent for this
     '''
 
     def __init__(self, **kwargs):

diff --git a/sphinx/api/agents.rst b/sphinx/api/agents.rst
@@ -23,6 +23,4 @@ Classes
 
 .. autoclass:: agents.MinimaxAgent
 
-.. autoclass:: agents.LimitedDepthMinimaxAgent
-
 .. autoclass:: agents.MCTSAgent