code: def draw_board(board, screen, height):
      for c in range(COLUMN_COUNT):
          for r in range(ROW_COUNT):
              pygame.draw.rect(
                  screen,
                  BLUE,
                  (
                      c * SQUARESIZE,
                      r * SQUARESIZE + SQUARESIZE,
                      SQUARESIZE,
                      SQUARESIZE,
                  ),
              )
              pygame.draw.circle(
                  screen,
                  BLACK,
                  (
                      int(c * SQUARESIZE + SQUARESIZE / 2),
                      int(r * SQUARESIZE + SQUARESIZE * 3 / 2),
                  ),
                  RADIUS,
              )
  
      for c in range(COLUMN_COUNT):
          for r in range(ROW_COUNT):
              if board[r][c] == PLAYER:
                  pygame.draw.circle(
                      screen,
                      RED,
                      (
                          int(c * SQUARESIZE + SQUARESIZE / 2),
                          int(r * SQUARESIZE + SQUARESIZE * 3 / 2),
                      ),
                      RADIUS,
                  )
              elif board[r][c] == AI:
                  pygame.draw.circle(
                      screen,
                      YELLOW,
                      (
                          int(c * SQUARESIZE + SQUARESIZE / 2),
                          int(r * SQUARESIZE + SQUARESIZE * 3 / 2),
                      ),
                      RADIUS,
                  )
      pygame.display.update()
  
file_path: PARL/examples/AlphaZero/connect4_aiplayer.py
imports: import math
  import random
  import sys
  import pygame
  from utils import *
  from MCTS import MCTS
  from Coach import Coach
  from parl.utils import logger
  from connect4_game import Connect4Game
label_desc: Use image node_id 1 for calling a global function with example usage: draw_board(board, screen, height) without return types