finalize code

Author: lechicuong2k3

Generate_CFG_dataset/cfg.py

@@ -9,7 +9,6 @@
 import io
 import linecache
 import random
-import ipdb
 
 # TODO later: graph
 '''
@@ -1131,7 +1130,6 @@ def removeCommentsAndDocstrings(self):
     parser.formatCode()
     cfg = CFGVisitor().build(filename, ast.parse(parser.script))
     cfg.clean()
-    # ipdb.set_trace()
     cfg.track_execution() 
     cfg.show(filepath)
 

model.py

@@ -159,11 +159,8 @@ def forward(self, x, edges, node_lens=None, token_lens=None, target=None):
 
         # B, max_node, H -> B, max_node
         output = torch.sigmoid(self.fc_output(h_f)) # 
-        # ipdb.set_trace()
         if self.opt.extra_aggregate:
-            # import ipdb; ipdb.set_trace()
             output = torch.bmm(matrix, output)
-        # import ipdb; ipdb.set_trace()
         return output
 
     @staticmethod