list aliasing bug

c3907460 · zzhang-cn · a3ce6e2f · c3907460
Commit c3907460 authored May 11, 2018 by zzhang-cn
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))]