From 81f6749c60e41085e01ec9754ba7b09610ef23a4 Mon Sep 17 00:00:00 2001
From: Comparing_Embeddings <38108440+vaudaine@users.noreply.github.com>
Date: Wed, 22 May 2019 13:51:31 +0200
Subject: [PATCH] Update graph_utils.py

---
 utils/graph_utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/utils/graph_utils.py b/utils/graph_utils.py
index e4ca997..d74df58 100755
--- a/utils/graph_utils.py
+++ b/utils/graph_utils.py
@@ -34,9 +34,9 @@ def __random_walk__(G, path_length, start, alpha=0, rand=random.Random()):
 
     while len(path) < path_length:
         cur = path[-1]
-        if len(G.neighbors(cur)) > 0:
+        if G.degree[cur] > 0:
             if rand.random() >= alpha:
-                path.append(rand.choice(G.neighbors(cur)))
+                path.append(rand.choice(list(G.neighbors(cur))))
             else:
                 path.append(path[0])
         else: