Skip to content

Commit

Permalink
updated API argument
Browse files Browse the repository at this point in the history
  • Loading branch information
BradReesWork committed Dec 2, 2019
1 parent 953816a commit 8c0e177
Showing 2 changed files with 5 additions and 4 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -36,3 +36,4 @@ build/
.settings

datasets/*
.pydevproject
8 changes: 4 additions & 4 deletions python/cugraph/pregel/traversal/bfs_bsp.py
Original file line number Diff line number Diff line change
@@ -16,7 +16,7 @@
from collections import OrderedDict


def bfs_df_pregel(df, start, src_col='src', dst_col='dst', copy_data=True):
def bfs_df_pregel(_df, start, src_col='src', dst_col='dst', copy_data=True):
"""
This function executes an unwieghted Breadth-First-Search (BFS) traversal
to find the distances and predecessors from a specified starting vertex
@@ -26,7 +26,7 @@ def bfs_df_pregel(df, start, src_col='src', dst_col='dst', copy_data=True):
Parameters
----------
df : cudf.dataframe
_df : cudf.dataframe
a dataframe containing the source and destination edge list
start : same type as 'src' and 'dst'
@@ -61,9 +61,9 @@ def bfs_df_pregel(df, start, src_col='src', dst_col='dst', copy_data=True):

# extract the src and dst into a dataframe that can be modified
if copy_data:
coo_data = df[[src_col, dst_col]]
coo_data = _df[[src_col, dst_col]]
else:
coo_data = df
coo_data = _df

coo_data.rename(columns={src_col: 'src', dst_col: 'dst'}, inplace=True)

0 comments on commit 8c0e177

Please sign in to comment.