Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/pytorch/test_nn.py
View file @ e0fc038d
...@@ -1516,7 +1516,7 @@ def test_hgt(idtype, in_size, num_heads): ...@@ -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) sorted_y = m(sorted_g, sorted_x, sorted_ntype, sorted_etype, presorted=False)
assert sorted_y.shape == (g.num_nodes(), head_size * num_heads) assert sorted_y.shape == (g.num_nodes(), head_size * num_heads)
# mini-batch # mini-batch
train_idx = th.randint(0, 100, (10, ), dtype = idtype) train_idx = th.randperm(100, dtype=idtype)[:10]
sampler = dgl.dataloading.NeighborSampler([-1]) sampler = dgl.dataloading.NeighborSampler([-1])
train_loader = dgl.dataloading.DataLoader(g, train_idx.to(dev), sampler, train_loader = dgl.dataloading.DataLoader(g, train_idx.to(dev), sampler,
batch_size=8, device=dev, 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