def get_labels_as_list()

in src/exporters/coreml/convert.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

def get_labels_as_list(model):
    """Return the labels of a classifier model as a sorted list."""
    labels = []
    for i in range(len(model.config.id2label)):
        if i in model.config.id2label.keys():
            labels.append(model.config.id2label[i])
    return labels