Merge pull request #26 from PikachuHy/fix/MultiDiGraph_node

fix: 'MultiDiGraph' object has no attribute 'node'
spcl · Jul 3, 2020 · 6eb8748 · 6eb8748
2 parents fdeb882 + ff88895
commit 6eb8748
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inst2vec/inst2vec_preprocess.py b/inst2vec/inst2vec_preprocess.py
@@ -1094,7 +1094,7 @@ def basic_block_leaf(G, node, ids_in_basic_block):
     if G.out_degree(node) > 0:
         for n in G.successors(node):
             if n in ids_in_basic_block:
-                if G.node[n]['id'] != 'ad_hoc':
+                if G.nodes[n]['id'] != 'ad_hoc':
                     # Check whether the outgoing edge is a "store"
                     for e in G.out_edges(node, data=True):
                         if 'store' not in e[2]['stmt']: