diff --git a/benchmarks/cugraph/pytest-based/bench_cugraph_uniform_neighbor_sample.py b/benchmarks/cugraph/pytest-based/bench_cugraph_uniform_neighbor_sample.py index 8c46095a7da..083acdde2f4 100644 --- a/benchmarks/cugraph/pytest-based/bench_cugraph_uniform_neighbor_sample.py +++ b/benchmarks/cugraph/pytest-based/bench_cugraph_uniform_neighbor_sample.py @@ -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}" ) @@ -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}