[+] A2 P2.2 e

2022-02-19 15:16:58 -05:00
parent a7a755da83
commit cc6058b239
1 changed files with 2 additions and 1 deletions
@@ -81,7 +81,8 @@ def generate_complete_game_tree(root_move: str, game_state: a2_minichess.Miniche
    >>> len(t.get_subtrees())
    0
    """
-    tree = a2_game_tree.GameTree(root_move, game_state.is_white_move())
+    tree = a2_game_tree.GameTree(root_move, game_state.is_white_move(),
+                                 int(game_state.get_winner() == 'White'))

    # Base case
    if d == 0: