Remove edge renumber map from the homogeneous sampling API

cpp/src/c_api/neighbor_sampling.cpp

@@ -1220,13 +1220,16 @@ struct neighbor_sampling_functor : public cugraph::c_api::abstract_functor {
 
         } else {  // heterogeneous renumbering
 
-          rmm::device_uvector<vertex_t> vertex_type_offsets(
-            graph_view.local_vertex_partition_range_size(), handle_.get_stream());
+          // FIXME: If no 'vertex_type_offsets' is provided, all vertices are assumed to have
+          // a vertex type of value 1. Update the API once 'vertex_type_offsets' is supported
+          rmm::device_uvector<vertex_t> vertex_type_offsets(2, handle_.get_stream());
+
+          cugraph::detail::stride_fill(handle_.get_stream(),
+                                       vertex_type_offsets.begin(),
+                                       vertex_type_offsets.size(),
+                                       vertex_t{0},
+                                       vertex_t{graph_view.local_vertex_partition_range_size()}
 
-          cugraph::detail::sequence_fill(handle_.get_stream(),
-                                         vertex_type_offsets.begin(),
-                                         vertex_type_offsets.size(),
-                                         vertex_t{0}  // FIXME: Update array
           );
 
           rmm::device_uvector<vertex_t> output_majors(0, handle_.get_stream());
@@ -1240,7 +1243,7 @@ struct neighbor_sampling_functor : public cugraph::c_api::abstract_functor {
                    edge_id,
                    label_type_hop_offsets,  // Contains information about the type and hop offsets
                    output_renumber_map,
-                   (*renumber_map_offsets),
+                   renumber_map_offsets,
                    renumbered_and_sorted_edge_id_renumber_map,
                    renumbered_and_sorted_edge_id_renumber_map_label_type_offsets) =
             cugraph::heterogeneous_renumber_and_sort_sampled_edgelist<vertex_t>(
@@ -1267,7 +1270,9 @@ struct neighbor_sampling_functor : public cugraph::c_api::abstract_functor {
 
               edge_label ? (*offsets).size() - 1 : size_t{1},
               hop ? fan_out_->size_ : size_t{1},
-              size_t{1},
+
+              vertex_type_offsets.size() -
+                1,  // num_vertex_type is by default 1 if 'vertex_type_offsets' is not provided
               num_edge_types_,
               src_is_major,
               do_expensive_check_);

python/pylibcugraph/pylibcugraph/homogeneous_biased_neighbor_sample.pyx

@@ -216,7 +216,7 @@ def homogeneous_biased_neighbor_sample(ResourceHandle resource_handle,
     >>> sampling_results = pylibcugraph.homogeneous_biased_neighbor_sample(
     ...         resource_handle, G, start_vertices, starting_vertex_label_offsets,
     ...         h_fan_out, False, True)
-    >>> >>> sampling_results
+    >>> sampling_results
     {'majors': array([2, 2, 5, 5, 1, 1], dtype=int32),
      'minors': array([1, 3, 3, 4, 3, 4], dtype=int32),
      'weight': array([3.1, 4.1, 7.2, 3.2, 2.1, 1.1], dtype=float32)}
@@ -383,8 +383,6 @@ def homogeneous_biased_neighbor_sample(ResourceHandle resource_handle,
     if renumber:
         cupy_renumber_map = result.get_renumber_map()
         cupy_renumber_map_offsets = result.get_renumber_map_offsets()
-        cupy_edge_renumber_map = result.get_edge_renumber_map()
-        cupy_edge_renumber_map_offsets = result.get_edge_renumber_map_offsets()
 
         sampling_results = {
             'major_offsets': cupy_major_offsets,
@@ -397,9 +395,7 @@ def homogeneous_biased_neighbor_sample(ResourceHandle resource_handle,
             'label_hop_offsets': cupy_label_hop_offsets,
             'hop_id': None,
             'renumber_map': cupy_renumber_map,
-            'renumber_map_offsets': cupy_renumber_map_offsets,
-            'edge_renumber_map' : cupy_edge_renumber_map,
-            'edge_renumber_map_offsets' : cupy_edge_renumber_map_offsets
+            'renumber_map_offsets': cupy_renumber_map_offsets
         }
 
     else:

python/pylibcugraph/pylibcugraph/homogeneous_uniform_neighbor_sample.pyx

@@ -211,7 +211,7 @@ def homogeneous_uniform_neighbor_sample(ResourceHandle resource_handle,
     >>> sampling_results = pylibcugraph.homogeneous_uniform_neighbor_sample(
     ...         resource_handle, G, start_vertices, starting_vertex_label_offsets,
     ...         h_fan_out, False, True)
-    >>> >>> sampling_results
+    >>> sampling_results
     {'majors': array([2, 2, 5, 5, 1, 1], dtype=int32),
      'minors': array([1, 3, 3, 4, 3, 4], dtype=int32),
      'weight': array([3.1, 4.1, 7.2, 3.2, 2.1, 1.1], dtype=float32)}
@@ -378,8 +378,6 @@ def homogeneous_uniform_neighbor_sample(ResourceHandle resource_handle,
     if renumber:
         cupy_renumber_map = result.get_renumber_map()
         cupy_renumber_map_offsets = result.get_renumber_map_offsets()
-        cupy_edge_renumber_map = result.get_edge_renumber_map()
-        cupy_edge_renumber_map_offsets = result.get_edge_renumber_map_offsets()
 
         sampling_results = {
             'major_offsets': cupy_major_offsets,
@@ -392,9 +390,7 @@ def homogeneous_uniform_neighbor_sample(ResourceHandle resource_handle,
             'label_hop_offsets': cupy_label_hop_offsets,
             'hop_id': None,
             'renumber_map': cupy_renumber_map,
-            'renumber_map_offsets': cupy_renumber_map_offsets,
-            'edge_renumber_map' : cupy_edge_renumber_map,
-            'edge_renumber_map_offsets' : cupy_edge_renumber_map_offsets
+            'renumber_map_offsets': cupy_renumber_map_offsets
         }
 
     else: