Update modeling_chessbot.py

modeling_chessbot.py

@@ -631,7 +631,7 @@ class ChessBotModel(ChessBotPreTrainedModel):
         # Map to legal moves
         legal_move_probs = {}
         for move in legal_moves:
-            idx = policy_index.index(move_trunc)
+            idx = policy_index.index(move)
             legal_move_probs[move] = probs[idx].item()
         
         # Select move based on probabilities