def knight_amount(board_state, player):
    """
    Returns amount of knights the player has
    """
    board = board_state
    knight_amt = 0
    for row in board:
        for column in row:
            if player == 1 and column == "k":
                knight_amt += 1
            elif player == 0 and column == "K":
                knight_amt += 1
    return knight_amt