Upload folder using huggingface_hub

app.py

@@ -28,7 +28,7 @@ model.eval()
 os.makedirs(name=TEMP_DIR, exist_ok=True)
 
 
-@logger.catch(level="DEBUG")
+@logger.catch(level="DEBUG", reraise=True)
 def evaluate_board(board: chess.Board):
     """Evaluate the board.
 
@@ -53,7 +53,7 @@ def evaluate_board(board: chess.Board):
     return os.path.join(TEMP_DIR, "board.svg"), eval_board(model=model, board=board)
 
 
-@logger.catch(level="DEBUG")
+@logger.catch(level="DEBUG", reraise=True)
 def plot_beam_search(board: chess.Board, depth: int, beam_width: int):
     """Plot the beam search tree.
 
@@ -75,10 +75,10 @@ def plot_beam_search(board: chess.Board, depth: int, beam_width: int):
         temp_dir=TEMP_DIR,
     )
 
-    return os.path.join(TEMP_DIR, "beam_search.png")
+    return os.path.join(TEMP_DIR, "beam_search.svg")
 
 
-@logger.catch(level="DEBUG")
+@logger.catch(level="DEBUG", reraise=True)
 def test_graphviz(board: chess.Board, depth: int, beam_width: int):
     """Plot the beam search tree.
 

src/agents/viz_utils.py

@@ -19,6 +19,7 @@ graph_params = {
 }
 
 
+@logger.catch(level="DEBUG")
 def board_to_svg(board: chess.Board, is_pruned: bool = False, size: int = 360) -> str:
     """Convert a chess.Board object to an SVG string.
 
@@ -47,8 +48,9 @@ def board_to_svg(board: chess.Board, is_pruned: bool = False, size: int = 360) -
     return svg
 
 
-def save_svg(board: chess.Board, filename: str, is_pruned: bool = False):
-    """Save a chess.Board object to an SVG file.
+@logger.catch(level="DEBUG")
+def save_svg(board: chess.Board, filename: str, is_pruned: bool = False) -> None:
+    """Save a chess.Board object to an image file.
 
     Args:
         board (chess.Board): chess.Board object
@@ -61,9 +63,9 @@ def save_svg(board: chess.Board, filename: str, is_pruned: bool = False):
         f.write(xml_declaration + board_to_svg(board, is_pruned=is_pruned))
 
 
-@logger.catch(level="DEBUG")
+@logger.catch(level="DEBUG", reraise=True)
 def plot_save_beam_search(
-    beam: AnyNode, filename: str, temp_dir: str = "../reports/figures"
+        beam: AnyNode, filename: str, temp_dir: str = "../reports/figures"
 ) -> None:
     """Plot and save the beam search tree.
 
@@ -119,11 +121,11 @@ def plot_save_beam_search(
             tailport=graph_params["tailport"],
         )
 
-    dot.render(filename=filename, format="png")
+    dot.render(filename=filename, format="svg")
 
     logger.info(os.listdir(temp_dir))
 
-    #log render file
+    # log render file
     with open(filename, "r") as f:
         logger.info(f.read())
 

temp/beam_search

@@ -8,28 +8,4 @@ digraph "Beam Search" {
 	ROOT -> "Depth 0 Candidate 1" [fontsize=20 headlabel=e8d7 headport=n tailport=s]
 	"Depth 1 Candidate 1" [label="score = 0.2649" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 1 Candidate 1.svg" image_pos=tc labelloc=b shape=none]
 	"Depth 0 Candidate 1" -> "Depth 1 Candidate 1" [fontsize=20 headlabel=d1b3 headport=n tailport=s]
-	"Depth 2 Candidate 2" [label="score = -0.0108" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 2 Candidate 2.svg" image_pos=tc labelloc=b shape=none]
-	"Depth 1 Candidate 1" -> "Depth 2 Candidate 2" [fontsize=20 headlabel=a7a5 headport=n tailport=s]
-	"Depth 3 Candidate 0" [label="score = 0.3467" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 3 Candidate 0.svg" image_pos=tc labelloc=b shape=none]
-	"Depth 2 Candidate 2" -> "Depth 3 Candidate 0" [fontsize=20 headlabel=b1d2 headport=n tailport=s]
-	"Depth 0 Candidate 2" [label="score = 0.1044" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 0 Candidate 2.svg" image_pos=tc labelloc=b shape=none]
-	ROOT -> "Depth 0 Candidate 2" [fontsize=20 headlabel=a7a6 headport=n tailport=s]
-	"Depth 1 Candidate 2" [label="score = 0.4743" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 1 Candidate 2.svg" image_pos=tc labelloc=b shape=none]
-	"Depth 0 Candidate 2" -> "Depth 1 Candidate 2" [fontsize=20 headlabel=f2f3 headport=n tailport=s]
-	"Depth 2 Candidate 3" [label="score = 0.0125" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 2 Candidate 3.svg" image_pos=tc labelloc=b shape=none]
-	"Depth 1 Candidate 2" -> "Depth 2 Candidate 3" [fontsize=20 headlabel=c7c6 headport=n tailport=s]
-	"Depth 3 Candidate 1" [label="score = 0.1925" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 3 Candidate 1.svg" image_pos=tc labelloc=b shape=none]
-	"Depth 2 Candidate 3" -> "Depth 3 Candidate 1" [fontsize=20 headlabel=b1a3 headport=n tailport=s]
-	"Depth 0 Candidate 3" [label="score = 0.1071" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 0 Candidate 3.svg" image_pos=tc labelloc=b shape=none]
-	ROOT -> "Depth 0 Candidate 3" [fontsize=20 headlabel=g7g5 headport=n tailport=s]
-	"Depth 1 Candidate 3" [label="score = 0.2784" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 1 Candidate 3.svg" image_pos=tc labelloc=b shape=none]
-	"Depth 0 Candidate 3" -> "Depth 1 Candidate 3" [fontsize=20 headlabel=d1c2 headport=n tailport=s]
-	"Depth 2 Candidate 0" [label="score = -0.0356" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 2 Candidate 0.svg" image_pos=tc labelloc=b shape=none]
-	"Depth 1 Candidate 3" -> "Depth 2 Candidate 0" [fontsize=20 headlabel=b8d7 headport=n tailport=s]
-	"Depth 3 Candidate 2" [label="score = 0.2430" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 3 Candidate 2.svg" image_pos=tc labelloc=b shape=none]
-	"Depth 2 Candidate 0" -> "Depth 3 Candidate 2" [fontsize=20 headlabel=g1h3 headport=n tailport=s]
-	"Depth 2 Candidate 1" [label="score = -0.0332" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 2 Candidate 1.svg" image_pos=tc labelloc=b shape=none]
-	"Depth 1 Candidate 3" -> "Depth 2 Candidate 1" [fontsize=20 headlabel=b8c6 headport=n tailport=s]
-	"Depth 3 Candidate 3" [label="score = 0.1914" fontsize=20 image="/Users/10142632/PycharmProjects/explainable-chess-engine/demos/temp/Depth 3 Candidate 3.svg" image_pos=tc labelloc=b shape=none]
-	"Depth 2 Candidate 1" -> "Depth 3 Candidate 3" [fontsize=20 headlabel=c4d5 headport=n tailport=s]
 }