Merge pull request rapidsai#852 from BradReesWork/branch-0.14

[REVIEW] Fix BFS Notebook
oorliu · May 6, 2020 · 5879040 · 5879040
2 parents 6113b7e + 55517aa
commit 5879040
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -33,7 +33,7 @@
 - PR #827 Fix indexing CI errors due to cudf updates
 - PR #844 Fixing tests, converting __getitem__ calls to .iloc
 - PR #851 Removed RMM from tests
-
+- PR #852 Fix BFS Notebook
 
 # cuGraph 0.13.0 (Date TBD)
 

diff --git a/notebooks/traversal/BFS.ipynb b/notebooks/traversal/BFS.ipynb
@@ -89,11 +89,11 @@
     "    \n",
     "    # Use the BFS predecessors and distance to trace the path \n",
     "    # from vertex id back to the starting vertex ( vertex 1 in this example)\n",
-    "    dist = df['distance'][id]\n",
+    "    dist = df['distance'].iloc[id]\n",
     "    lastVert = id\n",
     "    for i in range(dist):\n",
-    "        nextVert = df['predecessor'][lastVert]\n",
-    "        d = df['distance'][lastVert]\n",
+    "        nextVert = df['predecessor'].iloc[lastVert]\n",
+    "        d = df['distance'].iloc[lastVert]\n",
     "        print(\"Vertex: \" + str(lastVert) + \" was reached from vertex \" + str(nextVert) + \n",
     "        \" and distance to start is \" + str(d) )\n",
     "        lastVert = nextVert"
@@ -227,7 +227,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "df2[\"distance\"][0]"
+    "df2[\"distance\"].iloc[0]"
    ]
   },
   {