| | from leela_board import LeelaBoard |
| | import chess |
| | import torch |
| |
|
| |
|
| | def flip_move(move): |
| | from_square = chess.square_mirror(chess.parse_square(move[:2])) |
| | to_square = chess.square_mirror(chess.parse_square(move[2:4])) |
| | promotion = move[4:] if len(move) > 4 else "" |
| | return chess.square_name(from_square) + chess.square_name(to_square) + promotion |
| |
|
| |
|
| | def flip_board(fen, moves): |
| | temp_board = chess.Board(fen=fen) |
| | return temp_board.mirror().fen(), [flip_move(move) for move in moves] |
| |
|
| |
|
| | |
| | class ChessBoard: |
| | def __init__(self, fen): |
| | self.board = LeelaBoard(fen=fen) |
| | self.t = self.__t() |
| |
|
| | def move(self, move): |
| | self.board.push_uci(move) |
| | self.t = self.__t() |
| |
|
| | def __t(self): |
| | return torch.from_numpy(self.board.lcz_features()).float() |
| |
|
| | def __str__(self): |
| | return str(self.board) |
| |
|
