From 6b69bbc11676f2c3678cfaaff49f77cbd0a0f862 Mon Sep 17 00:00:00 2001
From: Nemo D'ACREMONT <ndacremont@enseirb-matmeca.fr>
Date: Sun, 18 May 2025 12:26:14 +0200
Subject: [PATCH] feat: make white player play

---
 go_player/myPlayer.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/go_player/myPlayer.py b/go_player/myPlayer.py
index 63139d2..72a0786 100644
--- a/go_player/myPlayer.py
+++ b/go_player/myPlayer.py
@@ -198,7 +198,10 @@ class myPlayer(PlayerInterface):
         with torch.no_grad():
             prediction = self.model(go_board)
 
-        return prediction
+        if color == Goban.Board._BLACK:
+            return prediction
+
+        return 1 - prediction
 
     def getPlayerMove(self):
         if self._board.is_game_over():