summaryrefslogtreecommitdiff
path: root/rebuilder.py
diff options
context:
space:
mode:
authorClaudiu Popa <pcmanticore@gmail.com>2014-03-02 12:07:33 +0200
committerClaudiu Popa <pcmanticore@gmail.com>2014-03-02 12:07:33 +0200
commit2e0df240db0cf677b0821371af7c2dfc021452ab (patch)
treedceb3a67c05a44221177473b682b979359a736c0 /rebuilder.py
parent8506cd2a4577bd0a2a6cdff412201c628912c8d1 (diff)
downloadastroid-git-2e0df240db0cf677b0821371af7c2dfc021452ab.tar.gz
Add yield_from property for Yield nodes.
--HG-- branch : yield_from
Diffstat (limited to 'rebuilder.py')
-rw-r--r--rebuilder.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/rebuilder.py b/rebuilder.py
index b9a9adbb..24ce8027 100644
--- a/rebuilder.py
+++ b/rebuilder.py
@@ -954,7 +954,9 @@ class TreeRebuilder3k(TreeRebuilder):
return newnode
def visit_yieldfrom(self, node, parent):
- return self.visit_yield(node, parent)
+ newnode = self.visit_yield(node, parent)
+ newnode._from = True
+ return newnode
def visit_class(self, node, parent):
newnode = super(TreeRebuilder3k, self).visit_class(node, parent)