bc9da61d8c8c2df4c3a31dad92d44ace8dc1c4c5,test/data/test_sampler.py,,test_cora,#,50

Before Change

    out_all = model.forward_all(data.x, data.edge_index)
    print(out_all[0])

    for data_flow in loader(torch.tensor([0])):
        x = data.x[data_flow[0].n_id]
        out = model.forward_batch(x, data_flow)[data_flow.batched_n_id]
        print(out[0])

After Change

        out = model(x, subdata.edge_index)[subdata.sub_b_id]
        assert torch.allclose(out_all[subdata.b_id], out)

    shutil.rmtree(root)

Instances