diff options
Diffstat (limited to 'rebuilder.py')
-rw-r--r-- | rebuilder.py | 4 |
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) |