Merge pull request rapidsai#1279 from Iroy30/self_loop_check

[REVIEW] add self loop check variable
aschaffer · Nov 30, 2020 · b98175e · b98175e
2 parents 1df2a42 + dc6b6fc
commit b98175e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,7 @@
 - PR #1264 CuPy sparse matrix input support for WCC, SCC, SSSP, and BFS
 - PR #1265 Implement Hungarian Algorithm
 - PR #1274 Add generic from_edgelist() and from_adjlist() APIs
+- PR #1279 Add self loop check variable in graph
 
 ## Improvements
 - PR #1227 Pin cmake policies to cmake 3.17 version

diff --git a/python/cugraph/structure/graph.py b/python/cugraph/structure/graph.py
@@ -99,6 +99,7 @@ def __init__(
  self.multi = multi
  self.distributed = False
  self.dynamic = dynamic
+ self.self_loop = False
  self.edgelist = None
  self.adjlist = None
  self.transposedadjlist = None
@@ -403,6 +404,8 @@ def from_cudf_edgelist(
  if type(source) is list and type(destination) is list:
  raise Exception("set renumber to True for multi column ids")
 
+ if (elist[source] == elist[destination]).any():
+ self.self_loop = True
  source_col = elist[source]
  dest_col = elist[destination]