Skip to content
10 changes: 6 additions & 4 deletions python/cugraph/cugraph/dask/traversal/bfs.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,12 +98,14 @@ def bfs(input_graph,
----------
input_graph : cugraph.Graph
cuGraph graph instance, should contain the connectivity information
as dask cudf edge list dataframe(edge weights are not used for this
as dask cudf edge list dataframe (edge weights are not used for this
algorithm).

start : Integer
Specify starting vertex for breadth-first search; this function
iterates over edges in the component reachable from this node.
start : Integer or list
The id of the graph vertex from which the traversal begins, or
if a list, the vertex from which the traversal begins in each
component of the graph. Only one vertex per connected
component of the graph is allowed.

depth_limit : Integer or None, optional (default=None)
Limit the depth of the search
Expand Down
7 changes: 5 additions & 2 deletions python/cugraph/cugraph/traversal/bfs.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,8 +183,11 @@ def bfs(G,
information. Edge weights, if present, should be single or double
precision floating point values.

start : Integer, optional (default=None)
The index of the graph vertex from which the traversal begins
start : Integer or list, optional (default=None)
The id of the graph vertex from which the traversal begins, or
if a list, the vertex from which the traversal begins in each
component of the graph. Only one vertex per connected
component of the graph is allowed.

depth_limit : Integer or None, optional (default=None)
Limit the depth of the search
Expand Down