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: