Commit 1da5a9fe authored by nd-02110114's avatar nd-02110114
Browse files

💚 fix ci error

parent d064e857

deepchem/models/torch_models/cgcnn.py

+0 −2
Original line number Diff line number Diff line
@@ -98,8 +98,6 @@ class CGCNNLayer(nn.Module): 
    dgl_graph.edata['edge_attr'] = edge_feats

    dgl_graph.update_all(self.message_func, self.reduce_func)

    node_feats = dgl_graph.ndata.pop('new_x')

    if self.batch_norm is not None:

      node_feats = self.batch_norm(node_feats)

    return node_feats

CRA Git | Maintained and supported by SUSTech CRA and CCSE