removed chess.Board

app/attention_interface.py

@@ -7,6 +7,7 @@ import copy
 import chess
 import gradio as gr
 
+from lczerolens.board import LczeroBoard
 from demo import constants, utils, visualisation
 
 
@@ -43,9 +44,9 @@ def compute_cache(
         return None, None, None, state_boards, state_cache
 
     try:
-        board = chess.Board(board_fen)
+        board = LczeroBoard(board_fen)
     except ValueError:
-        board = chess.Board()
+        board = LczeroBoard()
         gr.Warning("Invalid FEN, using starting position.")
     state_boards = [board.copy()]
     if action_seq:

app/backend_interface.py

@@ -12,6 +12,7 @@ from demo import constants, utils, visualisation
 from lczerolens import move_encodings
 from lczerolens.model import lczero as lczero_utils
 from lczerolens.xai import PolicyLens
+from lczerolens.board import LczeroBoard
 
 
 def list_models():
@@ -52,9 +53,9 @@ def make_policy_plot(
             "",
         )
     try:
-        board = chess.Board(board_fen)
+        board = LczeroBoard(board_fen)
     except ValueError:
-        board = chess.Board()
+        board = LczeroBoard()
         gr.Warning("Invalid FEN, using starting position.")
     if action_seq:
         try:
@@ -62,7 +63,7 @@ def make_policy_plot(
                 board.push_uci(action)
         except ValueError:
             gr.Warning("Invalid action sequence, using starting position.")
-            board = chess.Board()
+            board = LczeroBoard()
     lczero_weights = Weights(f"{constants.LEELA_MODEL_DIRECTORY}/{model_name}")
     lczero_backend = Backend(lczero_weights)
     uci_moves = [move.uci() for move in board.move_stack]

app/board_interface.py

@@ -6,13 +6,14 @@ import chess
 import gradio as gr
 
 from demo import constants
+from lczerolens.board import LczeroBoard
 
 
 def make_board_plot(board_fen, arrows, square):
     try:
-        board = chess.Board(board_fen)
+        board = LczeroBoard(board_fen)
     except ValueError:
-        board = chess.Board()
+        board = LczeroBoard()
         gr.Warning("Invalid FEN, using starting position.")
     try:
         if arrows:

app/crp_interface.py

@@ -8,6 +8,8 @@ import chess
 import gradio as gr
 
 from demo import constants, utils, visualisation
+from lczerolens.board import LczeroBoard
+
 
 cache = None
 boards = None
@@ -44,9 +46,9 @@ def compute_cache(
         gr.Warning("No model selected.")
         return None, None, None, None, None
     try:
-        board = chess.Board(board_fen)
+        board = LczeroBoard(board_fen)
     except ValueError:
-        board = chess.Board()
+        board = LczeroBoard()
         gr.Warning("Invalid FEN, using starting position.")
     boards = [board.copy()]
     if action_seq:
@@ -127,7 +129,7 @@ def make_history_plot(
     if board.turn == chess.BLACK:
         heatmap = heatmap.view(8, 8).flip(0).view(64)
     if board_index - history_index + 1 < 0:
-        history_board = chess.Board(fen=None)
+        history_board = LczeroBoard(fen=None)
     else:
         history_board = boards[board_index - history_index + 1]
     svg_board, fig = visualisation.render_heatmap(history_board, heatmap, vmin=vmin, vmax=vmax)

app/encoding_interface.py

@@ -7,6 +7,7 @@ import gradio as gr
 
 from demo import constants, visualisation
 from lczerolens import board_encodings
+from lczerolens.board import LczeroBoard
 
 
 def make_encoding_plot(
@@ -16,9 +17,9 @@ def make_encoding_plot(
     color_flip,
 ):
     try:
-        board = chess.Board(board_fen)
+        board = LczeroBoard(board_fen)
     except ValueError:
-        board = chess.Board()
+        board = LczeroBoard()
         gr.Warning("Invalid FEN, using starting position.")
     if action_seq:
         try:
@@ -26,7 +27,7 @@ def make_encoding_plot(
                 board.push_uci(action)
         except ValueError:
             gr.Warning("Invalid action sequence, using starting position.")
-            board = chess.Board()
+            board = LczeroBoard()
     board_tensor = board_encodings.board_to_input_tensor(board)
     heatmap = board_tensor[plane_index]
     if color_flip and board.turn == chess.BLACK:

app/lrp_interface.py

@@ -8,6 +8,7 @@ import chess
 import gradio as gr
 
 from demo import constants, utils, visualisation
+from lczerolens.board import LczeroBoard
 
 cache = None
 boards = None
@@ -44,9 +45,9 @@ def compute_cache(
         gr.Warning("No model selected.")
         return None, None, None, None, None
     try:
-        board = chess.Board(board_fen)
+        board = LczeroBoard(board_fen)
     except ValueError:
-        board = chess.Board()
+        board = LczeroBoard()
         gr.Warning("Invalid FEN, using starting position.")
     boards = [board.copy()]
     if action_seq:
@@ -127,7 +128,7 @@ def make_history_plot(
     if board.turn == chess.BLACK:
         heatmap = heatmap.view(8, 8).flip(0).view(64)
     if board_index - history_index + 1 < 0:
-        history_board = chess.Board(fen=None)
+        history_board = LczeroBoard(fen=None)
     else:
         history_board = boards[board_index - history_index + 1]
     svg_board, fig = visualisation.render_heatmap(history_board, heatmap, vmin=vmin, vmax=vmax)

app/policy_interface.py

@@ -9,8 +9,10 @@ import torch
 
 from demo import constants, utils, visualisation
 from lczerolens import move_encodings
+from lczerolens.board import LczeroBoard
 from lczerolens.xai import PolicyLens
 
+
 current_board = None
 current_raw_policy = None
 current_policy = None
@@ -55,7 +57,7 @@ def compute_policy(
             "",
         )
     try:
-        board = chess.Board(board_fen)
+        board = LczeroBoard(board_fen)
     except ValueError:
         gr.Warning("Invalid FEN.")
         return (None, None, "", None)