dcf8c1f9cef00b2d961a80689a5fc772dabeea8f,torch_geometric/utils/convert.py,,to_networkx,#Any#Any#Any#,47

Before Change

            copied. (default: :obj:`None`)
    

    G = nx.DiGraph()
    G.add_nodes_from(range(data.num_nodes))

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

After Change

            copied. (default: :obj:`None`)
    

    G = nx.Graph() if data.is_undirected() else nx.DiGraph()
    G.add_nodes_from(range(data.num_nodes))

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

Instances