Activate full-neighbor tests

rapidsai · Feb 23, 2024 · 05da3f6 · 05da3f6
1 parent b04cfc1
commit 05da3f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...graph/tests/wholegraph_torch/ops/test_wholegraph_unweighted_sample_without_replacement.py b/...graph/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])