Commit 995e0e0d authored by nd-02110114's avatar nd-02110114
Browse files

🐛 fix test

parent 6663fdf4

deepchem/feat/graph_data.py

+2 −2
Original line number Diff line number Diff line
@@ -134,8 +134,8 @@ class GraphData: 
    g = DGLGraph()

    g.add_nodes(self.num_nodes)

    g.add_edges(

        torch.from_numpy(self.edge_index[0]),

        torch.from_numpy(self.edge_index[1]))

        torch.from_numpy(self.edge_index[0]).long(),

        torch.from_numpy(self.edge_index[1]).long())

    g.ndata['x'] = torch.from_numpy(self.node_features)



    if self.edge_features is not None:

CRA Git | Maintained and supported by SUSTech CRA and CCSE