a998e7a1fa996edb4bcc1b34a0df5967ed6ec9e2,torch_geometric/utils/convert.py,,to_networkx,#Any#Any#Any#Any#Any#,53

Before Change

        G = nx.DiGraph()

    G.add_nodes_from(range(data.num_nodes))
    values = {key: data[key].squeeze().tolist() for key in data.keys}

    for i, (u, v) in enumerate(data.edge_index.t().tolist()):

After Change

    G.add_nodes_from(range(data.num_nodes))

    values = {}
    for key, item in data:
        if torch.is_tensor(item):
            values[key] = item.squeeze().tolist()
        else:
            values[key] = item
        if isinstance(values[key], (list, tuple)) and len(values[key]) == 1:
            values[key] = item[0]

    for i, (u, v) in enumerate(data.edge_index.t().tolist()):

        if to_undirected and v > u:
            continue

Instances