Unverified Commit 5ea7a250 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

Fix random BFS traversal failures (#621)

Essentially a redo of PR #612 which is erased by git merge somehow.
parent 7149650f
...@@ -31,8 +31,10 @@ def test_bfs(n=100): ...@@ -31,8 +31,10 @@ def test_bfs(n=100):
edges_nx.append(edge_frontier) edges_nx.append(edge_frontier)
frontier = set([v]) frontier = set([v])
edge_frontier = set([g.edge_id(u, v)]) edge_frontier = set([g.edge_id(u, v)])
layers_nx.append(frontier) # avoids empty successors
edges_nx.append(edge_frontier) if len(frontier) > 0 and len(edge_frontier) > 0:
layers_nx.append(frontier)
edges_nx.append(edge_frontier)
return layers_nx, edges_nx return layers_nx, edges_nx
g = dgl.DGLGraph() g = dgl.DGLGraph()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment