Added player visuals on replays

Kaggle · Nov 8, 2024 · 2b13dd5 · 2b13dd5
1 parent 02060c8
commit 2b13dd5
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/kaggle_environments/envs/chess/chess.js b/kaggle_environments/envs/chess/chess.js
diff --git a/kaggle_environments/envs/chess/chess.py b/kaggle_environments/envs/chess/chess.py
@@ -168,6 +168,7 @@ def square_str_to_int(square_str):
 game_one_complete = False
 game_start_position = math.floor(random.random() * len(OPENINGS))
 
+
 def interpreter(state, env):
     global seen_positions
     global game_one_complete
@@ -227,11 +228,9 @@ def interpreter(state, env):
         fen.split(" ")[4])  # fen keeps track of this
     # Check for game end conditions
     if pawn_or_capture_move_count == 100 or is_insufficient_material(
-            game.board):
-        active.status = terminal_state
-        inactive.status = terminal_state
-        game_one_complete = True
-    elif seen_positions[board_str] >= 3 or game.status == Game.STALEMATE:
+            game.board) or seen_positions[board_str] >= 3 or game.status == Game.STALEMATE:
+        active.reward += .5
+        inactive.reward += .5
         active.status = terminal_state
         inactive.status = terminal_state
         game_one_complete = True