42b0c38f81fe90de8f709133c2a5183ddff0d4d1,tests/compute/test_propagate.py,,test_prop_nodes_topo,#,47

Before Change

    tree.add_edge(2, 0)
    tree.add_edge(3, 2)
    tree.add_edge(4, 2)
    tree.register_message_func(mfunc)
    tree.register_reduce_func(rfunc)
    // init node feature data
    tree.ndata["x"] = F.zeros((5, 2))
    // set all leaf nodes to be ones
    tree.nodes[[1, 3, 4]].data["x"] = F.ones((3, 2))

After Change

    tree.add_edge(2, 0)
    tree.add_edge(3, 2)
    tree.add_edge(4, 2)
    tree = dgl.graph(tree.edges())
    // init node feature data
    tree.ndata["x"] = F.zeros((5, 2))
    // set all leaf nodes to be ones
    tree.nodes[[1, 3, 4]].data["x"] = F.ones((3, 2))

Instances