mctsagent.py

def search(self, time_budget: int) -> None:
    """
    Search and update the search tree for a
    specified amount of time in seconds.

    """
    start_time = clock()
    num_rollouts = 0

    # do until we exceed our time budget
    while clock() - start_time < time_budget:
        node, state = self.select_node()
        turn = state.turn()
        outcome = self.roll_out(state)
        self.backup(node, turn, outcome)
        num_rollouts += 1
    run_time = clock() - start_time
    node_count = self.tree_size()
    self.run_time = run_time
    self.node_count = node_count
    self.num_rollouts = num_rollouts