diff --git a/chesspp/engine.py b/chesspp/engine.py index da08b64..4a7242a 100644 --- a/chesspp/engine.py +++ b/chesspp/engine.py @@ -6,8 +6,8 @@ from torch import distributions as dist import chess import chess.engine -from chesspp.baysian_mcts import BayesianMcts -from chesspp.classic_mcts import ClassicMcts +from chesspp.mcts.baysian_mcts import BayesianMcts +from chesspp.mcts.classic_mcts import ClassicMcts from chesspp.i_strategy import IStrategy from typing import Dict diff --git a/chesspp/mcts/__init__.py b/chesspp/mcts/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/chesspp/baysian_mcts.py b/chesspp/mcts/baysian_mcts.py similarity index 99% rename from chesspp/baysian_mcts.py rename to chesspp/mcts/baysian_mcts.py index 9551bca..3de12f7 100644 --- a/chesspp/baysian_mcts.py +++ b/chesspp/mcts/baysian_mcts.py @@ -1,7 +1,7 @@ import math import torch.distributions as dist -from chesspp.i_mcts import * +from chesspp.mcts.i_mcts import * from chesspp.i_strategy import IStrategy from chesspp.util_gaussian import gaussian_ucb1, max_gaussian, min_gaussian diff --git a/chesspp/classic_mcts.py b/chesspp/mcts/classic_mcts.py similarity index 100% rename from chesspp/classic_mcts.py rename to chesspp/mcts/classic_mcts.py diff --git a/chesspp/i_mcts.py b/chesspp/mcts/i_mcts.py similarity index 100% rename from chesspp/i_mcts.py rename to chesspp/mcts/i_mcts.py diff --git a/main.py b/main.py index 549ae13..e2deef9 100644 --- a/main.py +++ b/main.py @@ -4,8 +4,8 @@ import time import chess import chess.engine import chess.pgn -from chesspp.classic_mcts import ClassicMcts -from chesspp.baysian_mcts import BayesianMcts +from chesspp.mcts.classic_mcts import ClassicMcts +from chesspp.mcts.baysian_mcts import BayesianMcts from chesspp.random_strategy import RandomStrategy from chesspp.stockfish_strategy import StockFishStrategy from chesspp import engine