Skip to content

Commit

Permalink
Remove Warnings and Timeout from `bench_cugraph_uniform_neighbor_samp…
Browse files Browse the repository at this point in the history
…le.py` (#4631)

This PR adds two arguments (`use_legacy_names=False`, `with_edge_properties=True`) to suppress warnings from `bench_cugraph_uniform_neighbor_sample.py` and reduces the number of fan-out combinations being run in the nightlies.

Authors:
  - Ralph Liu (https://github.com/nv-rliu)

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)

URL: #4631
  • Loading branch information
nv-rliu authored Aug 27, 2024
1 parent 73a2a84 commit f7e1cea
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -266,7 +266,7 @@ def uns_func(*args, **kwargs):
@pytest.mark.managedmem_off
@pytest.mark.poolallocator_on
@pytest.mark.parametrize("batch_size", params.batch_sizes.values())
@pytest.mark.parametrize("fanout", [params.fanout_10_25, params.fanout_5_10_15])
@pytest.mark.parametrize("fanout", [params.fanout_10_25])
@pytest.mark.parametrize(
"with_replacement", [False], ids=lambda v: f"with_replacement={v}"
)
Expand All @@ -287,6 +287,8 @@ def bench_cugraph_uniform_neighbor_sample(
start_list=uns_args["start_list"],
fanout_vals=uns_args["fanout"],
with_replacement=uns_args["with_replacement"],
use_legacy_names=False,
with_edge_properties=True,
)
"""
dtmap = {"int32": 32 // 8, "int64": 64 // 8}
Expand Down

0 comments on commit f7e1cea

Please sign in to comment.