def get_column_counts(workbook_dict):
    """
    This expects workbook dictionaries that have been cleaned i.e. have
    consistent sheet names throughout. It returns a mapping from file ids to
    the number of columns in its workbook.
    """
    temp_column_name_map = {}
    for name, workbook in workbook_dict.items():
        temp_column_name_map[name] = []
        for sheet in workbook:
            temp_column_name_map[name].append(len(sheet['1']))

    return temp_column_name_map