f5eb80d221fec8690e8cfb087256671545bb9a5a,examples/pytorch/jtnn/jtnn/jtnn_dec.py,DGLJTNNDecoder,run,#DGLJTNNDecoder#Any#Any#Any#Any#,115

Before Change

            )
            mol_tree_batch.edata.update(mol_tree_batch_lg.ndata)
            is_new = mol_tree_batch.nodes[v].data["new"]
            mol_tree_batch.pull(
                v,
                dec_tree_node_msg,
                dec_tree_node_reduce,
                dec_tree_node_update,
            )

            // Extract
            n_repr = mol_tree_batch.nodes[v].data
            h = n_repr["h"]

After Change

            mol_tree_batch.edata.update(mol_tree_batch_lg.ndata)

            is_new = mol_tree_batch.nodes[v].data["new"]
            mol_tree_batch.pull(v, DGLF.copy_e("m", "m"), DGLF.sum("m", "h"))
            mol_tree_batch.apply_nodes(dec_tree_node_update)

            // Extract

Instances