Generating return statements.

2014-10-18 12:40:36 +02:00
parent 74d4fcf4e7
commit b2c95cb02f
2 changed files with 20 additions and 3 deletions
--- a/jedi/parser/pytree.py
+++ b/jedi/parser/pytree.py
@@ -76,9 +76,9 @@ def convert(grammar, raw_node):
        'import_from': pr.Import,
        'break_stmt': pr.KeywordStatement,
        'continue_stmt': pr.KeywordStatement,
-        'return_stmt': pr.KeywordStatement,
+        'return_stmt': pr.ReturnStmt,
        'raise_stmt': pr.KeywordStatement,
-        'yield_stmt': pr.KeywordStatement,
+        'yield_stmt': pr.ReturnStmt,
        'del_stmt': pr.KeywordStatement,
        'pass_stmt': pr.KeywordStatement,
        'global_stmt': pr.GlobalStmt,