From 2d50a2b0f93927bdc1ee8bbebb709bd8224fd403 Mon Sep 17 00:00:00 2001 From: Vyas Ramasubramani Date: Thu, 27 Jul 2023 14:49:12 -0700 Subject: [PATCH] Fix compilation for pylibcugraph --- .../pylibcugraph/uniform_neighbor_sample.pyx | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/python/pylibcugraph/pylibcugraph/uniform_neighbor_sample.pyx b/python/pylibcugraph/pylibcugraph/uniform_neighbor_sample.pyx index d19162d503f..bc2aa9205f1 100644 --- a/python/pylibcugraph/pylibcugraph/uniform_neighbor_sample.pyx +++ b/python/pylibcugraph/pylibcugraph/uniform_neighbor_sample.pyx @@ -82,14 +82,14 @@ def uniform_neighbor_sample(ResourceHandle resource_handle, *, bool_t with_replacement, bool_t do_expensive_check, - bool_t with_edge_properties=False, + with_edge_properties=False, batch_id_list=None, label_list=None, label_to_output_comm_rank=None, prior_sources_behavior=None, - bool_t deduplicate_sources=False, - bool_t return_hops=False, - bool_t renumber=False, + deduplicate_sources=False, + return_hops=False, + renumber=False, random_state=None): """ Does neighborhood sampling, which samples nodes from a graph based on the @@ -177,6 +177,10 @@ def uniform_neighbor_sample(ResourceHandle resource_handle, resource_handle.c_resource_handle_ptr cdef cugraph_graph_t* c_graph_ptr = input_graph.c_graph_ptr + cdef bool_t c_deduplicate_sources = deduplicate_sources + cdef bool_t c_return_hops = return_hops + cdef bool_t c_renumber = renumber + assert_CAI_type(start_list, "start_list") assert_CAI_type(batch_id_list, "batch_id_list", True) assert_CAI_type(label_list, "label_list", True) @@ -271,10 +275,10 @@ def uniform_neighbor_sample(ResourceHandle resource_handle, assert_success(error_code, error_ptr, "cugraph_sampling_options_create") cugraph_sampling_set_with_replacement(sampling_options, with_replacement) - cugraph_sampling_set_return_hops(sampling_options, return_hops) - cugraph_sampling_set_dedupe_sources(sampling_options, deduplicate_sources) + cugraph_sampling_set_return_hops(sampling_options, c_return_hops) + cugraph_sampling_set_dedupe_sources(sampling_options, c_deduplicate_sources) cugraph_sampling_set_prior_sources_behavior(sampling_options, prior_sources_behavior_e) - cugraph_sampling_set_renumber_results(sampling_options, renumber) + cugraph_sampling_set_renumber_results(sampling_options, c_renumber) error_code = cugraph_uniform_neighbor_sample( c_resource_handle_ptr,