Add networkx version checking to utils.py

2017-11-02 10:23:57 -04:00 · 2017-11-02 10:23:57 -04:00 · 9293301787
commit 9293301787
parent ca283f88b0
1 changed files with 3 additions and 0 deletions
--- a/graphsage/utils.py
+++ b/graphsage/utils.py
@ -6,7 +6,10 @@ import json
 import sys
 import os

+import networkx as nx
 from networkx.readwrite import json_graph
+major, minor = map(int, nx.__version__.split('.'))
+assert (major <= 1) and (minor <= 11), "networkx major version > 1.11"

 WALK_LEN=5
 N_WALKS=50