HaileyStorm commited on
Commit
ca35a8a
·
verified ·
1 Parent(s): ccb4a44

Update chess-gpt-eval/main.py

Browse files
Files changed (1) hide show
  1. chess-gpt-eval/main.py +1 -1
chess-gpt-eval/main.py CHANGED
@@ -336,7 +336,7 @@ def initialize_game_with_opening(
336
  move = token
337
 
338
  board.push_san(move)
339
- return game_state.rstrip(), board
340
 
341
 
342
  # Return is (move_san, move_uci, attempts, is_resignation, is_illegal_move)
 
336
  move = token
337
 
338
  board.push_san(move)
339
+ return game_state.rstrip(), board, len(tokens) // 2
340
 
341
 
342
  # Return is (move_san, move_uci, attempts, is_resignation, is_illegal_move)