def get_canonical_mention()

in amazon_comprehend_events_tutorial/notebooks/events_graph.py [0:0]

• 9 lines of code
• 4 McCabe index (conditional complexity)

def get_canonical_mention(mentions, method="longest"):
    extents = enumerate([m['Text'] for m in mentions])
    if method == "longest":
        name = sorted(extents, key=lambda x: len(x[1]))
    elif method == "most_common": 
        name = [Counter(extents).most_common()[0][0]]
    else:
        name = [list(extents)[0]]
    return [mentions[name[-1][0]]]