From 05da3f611e0462852e7592f73d4add25f87e1e07 Mon Sep 17 00:00:00 2001 From: Chang Liu Date: Thu, 22 Feb 2024 17:29:44 -0800 Subject: [PATCH] Activate full-neighbor tests --- .../test_wholegraph_unweighted_sample_without_replacement.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholegraph_unweighted_sample_without_replacement.py b/python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholegraph_unweighted_sample_without_replacement.py index 776a94094..1953419f5 100644 --- a/python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholegraph_unweighted_sample_without_replacement.py +++ b/python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholegraph_unweighted_sample_without_replacement.py @@ -359,7 +359,7 @@ def routine_func(world_rank: int, world_size: int, **kwargs): @pytest.mark.parametrize("graph_node_count", [103]) @pytest.mark.parametrize("graph_edge_count", [1043]) -@pytest.mark.parametrize("max_sample_count", [11]) +@pytest.mark.parametrize("max_sample_count", [11, -1]) @pytest.mark.parametrize("center_node_count", [13]) @pytest.mark.parametrize("center_node_dtype", [torch.int32, torch.int64]) @pytest.mark.parametrize("col_id_dtype", [0, 1])