fix (#3903)

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>

fix (#3903)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
a0bf5daa · Tong He · GitHub · c380f0b5 · a0bf5daa
Unverified Commit a0bf5daa authored Apr 13, 2022 by Tong He Committed by GitHub Apr 13, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pytorch/hilander/models/graphconv.py examples/pytorch/hilander/models/graphconv.py +1 -1

No files found.
--- a/examples/pytorch/hilander/models/graphconv.py
+++ b/examples/pytorch/hilander/models/graphconv.py
@@ -19,7 +19,7 @@ class GraphConvLayer(nn.Module):
            srcfeat, dstfeat = feat
        else:
            srcfeat = feat
-            dstfeat = feat[:graph.num_dst_nodes()]
+            dstfeat = feat[:bipartite.num_dst_nodes()]
        graph = bipartite.local_var()
        graph.srcdata['h'] = srcfeat