From cd41b59e2cebe0b8e57ababd5cc8db4a9a077f54 Mon Sep 17 00:00:00 2001
From: Bhargavaram Krishnapur
 <132352169+Codex-Crusader@users.noreply.github.com>
Date: Tue, 16 Sep 2025 17:44:59 +0530
Subject: [PATCH] Refactor promotion handling in MCTS logic

---
 azlite_portfolio_clean.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/azlite_portfolio_clean.py b/azlite_portfolio_clean.py
index 2c0f5b1..3f6e242 100644
--- a/azlite_portfolio_clean.py
+++ b/azlite_portfolio_clean.py
@@ -187,7 +187,7 @@ def score_moves(self, state_embed: torch.Tensor,
         return logits
 
 
-# ----------------------------- MCTS -----------------------------------------
+# ----------------------------- MCTS -------------------------------
 
 @dataclass
 class MCTSNode:
@@ -344,8 +344,8 @@ def _simulate(self, board: chess.Board, root: MCTSNode) -> float:
                         if mv.promotion is None:
                             promo_idxs.append(0)
                         else:
-                            PROMOTION_MAP.get(mv.promotion, 0)
-                            promo_idxs.append(PROMOTION_MAP.get(mv.promotion, 0))
+                            promo_value = PROMOTION_MAP.get(mv.promotion, 0)
+                            promo_idxs.append(promo_value)
                     logits = self.net.score_moves(
                         state_embed.squeeze(0),
                         torch.tensor(from_idxs, dtype=torch.long,