jungyhuk/plotcoder

plotcoder's Issues

about train

Hi, Dear Author @Jungyhuk ，i try to replicate the experiments of poltcoder(python run.py --nl --use_comments --code_context --nl_code_linking --copy_mechanism --hierarchy --target_code_transform)，but it throw a TypeError error(TypeError: argument of type 'Nonetype' is not iterable) in line 313 of data_untils.py(if 'func' not in ast_tree:). I found when parse the cur_code(['for', 'curve', 'in', 'curvePoints', ':', 'ax', '.', 'plot', '(', 'curve', '[', ':', ',', '0', ']', ',', 'curve', '[', ':', ',', '1', ']', ',', 'curve', '[', ':', ',', '2', ']', ')', '\n']), ast_tree['body'][0]['value'] is a NoneType.
How should i go about resolving this issue?