link cugraph/issues/2863

rapidsai · Nov 8, 2022 · bf49417 · bf49417
1 parent 35872e6
commit bf49417
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/cugraph/cugraph/gnn/dgl_extensions/cugraph_service_store.py b/python/cugraph/cugraph/gnn/dgl_extensions/cugraph_service_store.py
@@ -379,9 +379,10 @@ def sample_neighbors(
         # of the seed dtype is not same as the node dtype
         self.set_sg_node_dtype(first_sg)
 
-        # TODO: Cant send dlpack or cupy arrays or numpys arrays
+        # Cant send dlpack or cupy arrays or numpys arrays
         # through  extensions
-        # Ask Rick
+        # See issue: https://github.com/rapidsai/cugraph/issues/2863
+
         if isinstance(nodes_cap, dict):
             nodes_ar = {
                 k: cp.from_dlpack(v).get().tolist() for k, v in nodes_cap.items()