update docstrings example

rapidsai · Nov 15, 2024 · 69afe17 · 69afe17
1 parent dc203dd
commit 69afe17
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 12 deletions.
diff --git a/python/pylibcugraph/pylibcugraph/heterogeneous_biased_neighbor_sample.pyx b/python/pylibcugraph/pylibcugraph/heterogeneous_biased_neighbor_sample.pyx
@@ -229,9 +229,9 @@ def heterogeneous_biased_neighbor_sample(ResourceHandle resource_handle,
     ...         resource_handle, G, start_vertices, starting_vertex_label_offsets,
     ...         h_fan_out, num_edge_types, False, True)
     >>> sampling_results
-    {'sources': array([2, 2, 2, 5, 5, 1, 1, 1, 1], dtype=int32),
-     'destinations': array([0, 1, 3, 3, 4, 0, 2, 3, 4], dtype=int32),
-     'indices': array([5.1, 3.1, 4.1, 7.2, 3.2, 0.1, 3.1, 2.1, 1.1], dtype=float32)}
+    {'majors': array([2, 2, 2, 5, 5, 1, 1, 1, 1], dtype=int32),
+     'minors': array([0, 1, 3, 3, 4, 0, 2, 3, 4], dtype=int32),
+     'weight': array([5.1, 3.1, 4.1, 7.2, 3.2, 0.1, 3.1, 2.1, 1.1], dtype=float32)}
 
     """
     cdef cugraph_resource_handle_t* c_resource_handle_ptr = (

diff --git a/python/pylibcugraph/pylibcugraph/heterogeneous_uniform_neighbor_sample.pyx b/python/pylibcugraph/pylibcugraph/heterogeneous_uniform_neighbor_sample.pyx
@@ -224,9 +224,9 @@ def heterogeneous_uniform_neighbor_sample(ResourceHandle resource_handle,
     ...         resource_handle, G, start_vertices, starting_vertex_label_offsets,
     ...         h_fan_out, num_edge_types, False, True)
     >>> sampling_results
-    {'sources': array([2, 2, 2, 5, 5, 1, 1, 1, 1], dtype=int32),
-     'destinations': array([0, 1, 3, 3, 4, 0, 2, 3, 4], dtype=int32),
-     'indices': array([5.1, 3.1, 4.1, 7.2, 3.2, 0.1, 3.1, 2.1, 1.1], dtype=float32)}
+    {'majors': array([2, 2, 2, 5, 5, 1, 1, 1, 1], dtype=int32),
+     'minors': array([0, 1, 3, 3, 4, 0, 2, 3, 4], dtype=int32),
+     'weight': array([5.1, 3.1, 4.1, 7.2, 3.2, 0.1, 3.1, 2.1, 1.1], dtype=float32)}
 
     """
     cdef cugraph_resource_handle_t* c_resource_handle_ptr = (

diff --git a/python/pylibcugraph/pylibcugraph/homogeneous_biased_neighbor_sample.pyx b/python/pylibcugraph/pylibcugraph/homogeneous_biased_neighbor_sample.pyx
@@ -224,9 +224,9 @@ def homogeneous_biased_neighbor_sample(ResourceHandle resource_handle,
     ...         resource_handle, G, start_vertices, starting_vertex_label_offsets,
     ...         h_fan_out, False, True)
     >>> >>> sampling_results
-    {'sources': array([2, 2, 5, 5, 1, 1], dtype=int32),
-     'destinations': array([1, 3, 3, 4, 3, 4], dtype=int32),
-     'indices': array([3.1, 4.1, 7.2, 3.2, 2.1, 1.1], dtype=float32)}
+    {'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)}
 
     """
     cdef cugraph_resource_handle_t* c_resource_handle_ptr = (

diff --git a/python/pylibcugraph/pylibcugraph/homogeneous_uniform_neighbor_sample.pyx b/python/pylibcugraph/pylibcugraph/homogeneous_uniform_neighbor_sample.pyx
@@ -219,9 +219,9 @@ def homogeneous_uniform_neighbor_sample(ResourceHandle resource_handle,
     ...         resource_handle, G, start_vertices, starting_vertex_label_offsets,
     ...         h_fan_out, False, True)
     >>> >>> sampling_results
-    {'sources': array([2, 2, 5, 5, 1, 1], dtype=int32),
-     'destinations': array([1, 3, 3, 4, 3, 4], dtype=int32),
-     'indices': array([3.1, 4.1, 7.2, 3.2, 2.1, 1.1], dtype=float32)}
+    {'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)}
 
     """
     cdef cugraph_resource_handle_t* c_resource_handle_ptr = (