From 9293301787edba4210e4d0efb6b8d9e14524f9dd Mon Sep 17 00:00:00 2001
From: Ben Johnson <bkj.322@gmail.com>
Date: Thu, 2 Nov 2017 10:23:57 -0400
Subject: [PATCH] Add networkx version checking to utils.py

---
 graphsage/utils.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/graphsage/utils.py b/graphsage/utils.py
index 15e3f97..4b73c35 100644
--- 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