"docs/vscode:/vscode.git/clone" did not exist on "b9e99654e1a08328f5e9365eb368994511d91ac2"
Unverified Commit e0fc038d authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[Test] Fix duplicate seed generation in unit test (#4961)

* fix unit test

* oops
parent 566d231a
......@@ -1516,7 +1516,7 @@ def test_hgt(idtype, in_size, num_heads):
sorted_y = m(sorted_g, sorted_x, sorted_ntype, sorted_etype, presorted=False)
assert sorted_y.shape == (g.num_nodes(), head_size * num_heads)
# mini-batch
train_idx = th.randint(0, 100, (10, ), dtype = idtype)
train_idx = th.randperm(100, dtype=idtype)[:10]
sampler = dgl.dataloading.NeighborSampler([-1])
train_loader = dgl.dataloading.DataLoader(g, train_idx.to(dev), sampler,
batch_size=8, device=dev,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment