Skip to content

GitLab

"examples/distributed/graphsage/node_classification.py" did not exist on "a1472bcff6baf52303ba76a0230658a52d3be2d4"
Unverified Commit 764b5d05 authored by Serge Panev's avatar Serge Panev Committed by GitHub
Browse files

Fix test_module_heat_kernel test (#3941) 


Signed-off-by: default avatarSerge Panev <spanev@nvidia.com>
Co-authored-by: default avatarMufei Li <mufeili1996@gmail.com>
parent dc1c0ac2
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
tests/compute/test_transform.py
View file @ 764b5d05
...@@ -2241,9 +2241,6 @@ def test_module_heat_kernel(idtype): ...@@ -2241,9 +2241,6 @@ def test_module_heat_kernel(idtype):
assert new_g.idtype == g.idtype assert new_g.idtype == g.idtype
assert new_g.device == g.device assert new_g.device == g.device
assert new_g.num_nodes() == g.num_nodes() assert new_g.num_nodes() == g.num_nodes()
src, dst = new_g.edges()
eset = set(zip(list(F.asnumpy(src)), list(F.asnumpy(dst))))
assert eset == {(0, 2), (0, 4), (1, 3), (1, 5), (2, 3), (2, 4), (3, 5), (4, 5)}
assert F.allclose(g.ndata['h'], new_g.ndata['h']) assert F.allclose(g.ndata['h'], new_g.ndata['h'])
assert 'w' in new_g.edata assert 'w' in new_g.edata
......
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