HaileyStorm
commited on
Update chess-gpt-eval/main.py
Browse files- 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)
|