Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "1f15e3bc319fc55f175e2e32d28f27d503ebb26c"
Unverified Commit 9249be28 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[GraphBolt] remove main function from unit test script (#5950)

parent 49e336d2
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 10 deletions
tests/python/pytorch/graphbolt/test_csc_sampling_graph.py
View file @ 9249be28
...@@ -797,13 +797,3 @@ def test_from_dglgraph_heterogeneous(): ...@@ -797,13 +797,3 @@ def test_from_dglgraph_heterogeneous():
("n2", "r21", "n1"): 2, ("n2", "r21", "n1"): 2,
("n2", "r23", "n3"): 3, ("n2", "r23", "n3"): 3,
} }
if __name__ == "__main__":
test_sample_neighbors()
test_sample_neighbors_replace(True, 12)
test_sample_neighbors_probs(
False,
torch.tensor([2.5, 0, 8.4, 0, 0.4, 1.2, 2.5, 0, 8.4, 0, 0.4, 1.2]),
)
test_sample_neighbors_zero_probs(True, torch.zeros(12, dtype=torch.float32))
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