Skip to content

Commit d48ca1c

Browse files
committed
Fix empty-graph return type: return set() instead of () for nodes/edges
Made-with: Cursor
1 parent 613a3d1 commit d48ca1c

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

cuda_core/cuda/core/_graph/_graph_def/_graph_def.pyx

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -319,7 +319,7 @@ cdef class GraphDef:
319319

320320
Returns
321321
-------
322-
tuple of GraphNode
322+
set of GraphNode
323323
All nodes in the graph.
324324
"""
325325
cdef size_t num_nodes = 0
@@ -328,7 +328,7 @@ cdef class GraphDef:
328328
HANDLE_RETURN(cydriver.cuGraphGetNodes(as_cu(self._h_graph), NULL, &num_nodes))
329329
330330
if num_nodes == 0:
331-
return ()
331+
return set()
332332
333333
cdef vector[cydriver.CUgraphNode] nodes_vec
334334
nodes_vec.resize(num_nodes)
@@ -342,7 +342,7 @@ cdef class GraphDef:
342342

343343
Returns
344344
-------
345-
tuple of tuple
345+
set of tuple
346346
Each element is a (from_node, to_node) pair representing
347347
a dependency edge in the graph.
348348
"""
@@ -355,7 +355,7 @@ cdef class GraphDef:
355355
HANDLE_RETURN(cydriver.cuGraphGetEdges(as_cu(self._h_graph), NULL, NULL, &num_edges))
356356
357357
if num_edges == 0:
358-
return ()
358+
return set()
359359
360360
cdef vector[cydriver.CUgraphNode] from_nodes
361361
cdef vector[cydriver.CUgraphNode] to_nodes

0 commit comments

Comments
 (0)