Update chess-gpt-eval/main.py

chess-gpt-eval/main.py

@@ -190,7 +190,8 @@ def record_results(
     player_two_failed_to_find_legal_move: bool,
     total_moves: int,
     illegal_moves: int,
-    opening_moves: int
+    opening_moves: int,
+    illegal_move_numbers: list[int]
 ):
     unique_game_id = generate_unique_game_id()
 
@@ -249,6 +250,8 @@ def record_results(
         "p1_avg_attempts_per_illegal": 0 if player_one_illegal_moves == 0 else player_one_illegal_attempts / float(player_one_illegal_moves),
         "p1_illegal_attemtps_pct": 1.0 if played_moves == 0 else player_one_illegal_attempts / float(player_one_illegal_attempts + player_one_legal_moves),
         "p1_illegal_moves_pct": 1.0 if played_moves == 0 else player_one_illegal_moves / float(played_moves),
+        "p1_first_illegal_move_num": illegal_move_numbers[0],
+        "p1_avg_illegal_move_num": np.average(illegal_move_numbers),
         "time_taken": total_time,
         "total_moves": total_moves,
         "illegal_moves": illegal_moves,
@@ -451,6 +454,7 @@ def play_game(
 
         total_moves = 0
         illegal_moves = 0
+        illegal_move_numbers = []
         print_for_human = isinstance(player_one, HumanPlayer) or isinstance(player_two, HumanPlayer)
 
         while not board.is_game_over():
@@ -482,6 +486,7 @@ def play_game(
             player_one_illegal_attempts += illegal_moves_one
             if illegal_moves_one != 0:
                 player_one_legal_moves -= 1
+                illegal_move_numbers.append(board.fullmove_number)
             if (
                 board.is_game_over()
                 or player_one_resignation
@@ -533,7 +538,8 @@ def play_game(
             player_two_failed_to_find_legal_move,
             total_moves,
             illegal_moves,
-            opening_moves
+            opening_moves,
+            illegal_move_numbers
         )
     if isinstance(player_one, StockfishPlayer):
         player_one.close()