Update chess-gpt-eval/main.py

chess-gpt-eval/main.py

@@ -250,8 +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),
+        "p1_first_illegal_move_num": illegal_move_numbers[0] if illegal_move_numbers else 0,
+        "p1_avg_illegal_move_num": np.average(illegal_move_numbers) if illegal_move_numbers else 0,
         "time_taken": total_time,
         "total_moves": total_moves,
         "illegal_moves": illegal_moves,