def convert_to_adj_list()

in source/sagemaker/sagemaker_graph_entity_resolution/dgl_entity_resolution/utils.py [0:0]

• 14 lines of code
• 5 McCabe index (conditional complexity)

def convert_to_adj_list(i, j):
    adj_list = {}
    for (a, b) in zip(i, j):
        if a == b:
            continue
        if a in adj_list:
            adj_list[a].append(b)
        else:
            adj_list[a] = [b]
        if b in adj_list:
            adj_list[b].append(a)
        else:
            adj_list[b] = [a]
    return adj_list