Update graph.py

main
Alexandre B A Villares 2020-08-08 18:54:39 -03:00
rodzic 11730ddec3
commit 83747e7b08
1 zmienionych plików z 3 dodań i 3 usunięć

Wyświetl plik

@ -289,8 +289,8 @@ class Graph(object):
graph.add_vertex(v)
if random(1) < connect_rate:
if allow_loops:
names = list(vertices)
pool = list(vertices)
else:
names = list(vertices - set(v))
graph.add_edge({v, choice(names)})
pool = list(vertices - set((v,)))
graph.add_edge({v, choice(pool)})
return graph