Merge pull request #6 from zzhang-cn/zz-play

list aliasing bug

Merge pull request #6 from zzhang-cn/zz-play
list aliasing bug
a347214f · Zheng Zhang · GitHub · a3ce6e2f · c3907460 · a347214f
Unverified Commit a347214f authored May 11, 2018 by Zheng Zhang Committed by GitHub May 11, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

model.py model.py +1 -1

No files found.
--- a/model.py
+++ b/model.py
@@ -88,7 +88,7 @@ class TreeGlimpsedClassifier(NN.Module):
                         {i: 'h%d' % i for i in range(len(G.nodes()))},
                         False
                         )
-        self.h_nodes_list = h_nodes_list = G.nodes()
+        self.h_nodes_list = h_nodes_list = list(G.nodes)[:]
        for h in h_nodes_list:
            G.node[h]['type'] = 'h'
        b_nodes_list = ['b%d' % i for i in range(len(h_nodes_list))]