fix the None issue

Jungyhuk · Jungyhuk · commit 4c5fe923dc69 · 2021-10-05T17:20:21.000-07:00
diff --git a/models/data_utils/data_utils.py b/models/data_utils/data_utils.py
@@ -310,7 +310,7 @@ def code_seq_transform(self, init_code_seq, reserved_dfs):
 				st = ed
 				continue
 			ast_tree = ast_tree['body'][0]['value']
-			if 'func' not in ast_tree:
+			if ast_tree is None or 'func' not in ast_tree:
 				st = ed
 				continue
 			func = ast_tree['func']
