Merge pull request rapidsai#35 from afender/hotfix_idx_range_in_pr

Fixed the size of the pagerank array in Pagerank bindings
seunghwak · Jan 10, 2019 · b91a6bc · b91a6bc
2 parents 4af3d3e + 7995e8d
commit b91a6bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pagerank/pagerank_wrapper.pyx b/python/pagerank/pagerank_wrapper.pyx
@@ -191,7 +191,7 @@ cpdef pagerank(G,alpha=0.85, max_iter=100, tol=1.0e-5):
 
     cdef gdf_graph* g = <gdf_graph*>graph
     size = g.transposedAdjList.offsets.size
-    pagerank = cudf.Series(np.zeros(g.transposedAdjList.offsets.size,dtype=np.float64))
+    pagerank = cudf.Series(np.zeros(g.transposedAdjList.offsets.size-1,dtype=np.float64))
     cdef uintptr_t pagerank_ptr = create_column(pagerank)    
 
     gdf_pagerank(<gdf_graph*>graph, <gdf_column*>pagerank_ptr, <float> alpha, <float> tol, <int> max_iter, <bool> 0)