From d3105e5223e9602ba3ef2dd27e343d044fe4bb5f Mon Sep 17 00:00:00 2001
From: William Hamilton <will@Williams-MacBook-Air-3.local>
Date: Sun, 24 Jun 2018 09:30:13 -0400
Subject: [PATCH] Small fix for type conversion.

---
 graphsage/encoders.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/graphsage/encoders.py b/graphsage/encoders.py
index 9094b1c..a4baa05 100644
--- a/graphsage/encoders.py
+++ b/graphsage/encoders.py
@@ -36,7 +36,7 @@ class Encoder(nn.Module):
 
         nodes     -- list of nodes
         """
-        neigh_feats = self.aggregator.forward(nodes, [self.adj_lists[node] for node in nodes], 
+        neigh_feats = self.aggregator.forward(nodes, [self.adj_lists[int(node)] for node in nodes], 
                 self.num_sample)
         if not self.gcn:
             if self.cuda: