Skip to content

GitLab

"docs/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "f73d0b6becf8f6b985c237bdfc94e31f2d96e956"
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():
("n2", "r21", "n1"): 2,
("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