dcf8c1f9cef00b2d961a80689a5fc772dabeea8f,torch_geometric/utils/convert.py,,from_networkx,#Any#,75

Before Change

        data[key] = torch.tensor(item)

    if not G.is_directed():
        edge_index = torch.cat((edge_index, edge_index[[1,0],:]), dim=1)
        
    data["edge_index"] = edge_index
    data = torch_geometric.data.Data.from_dict(data)

After Change

        data[key] = torch.tensor(item)

    if not G.is_directed():
        edge_index = to_undirected(edge_index, G.number_of_nodes())
        
    data["edge_index"] = edge_index
    data = torch_geometric.data.Data.from_dict(data)
    data.num_nodes = G.number_of_nodes()

Instances