summaryrefslogtreecommitdiff
path: root/rebuilder.py
diff options
context:
space:
mode:
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)