Skip to content

Commit cdbf1da

Browse files
author
Caterina Gratton
committed
Update watts-strogatz call to match new NetworkX API
1 parent fde7626 commit cdbf1da

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

brainx/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,13 +15,13 @@ def patch_nx():
1515

1616
nx._watts_strogatz_graph_ori = nx.watts_strogatz_graph
1717

18-
def patched_ws(n, k, p, create_using=None, seed=None):
18+
def patched_ws(n, k, p, seed=None):
1919
if k<2:
2020
g = nx.Graph()
2121
g.add_nodes_from(range(n))
2222
return g
2323
else:
24-
return nx._watts_strogatz_graph_ori(n, k, p, create_using, seed)
24+
return nx._watts_strogatz_graph_ori(n, k, p, seed)
2525

2626
patched_ws.__doc__ = nx._watts_strogatz_graph_ori.__doc__
2727

0 commit comments

Comments
 (0)