diff options
Diffstat (limited to 'astroid/tree/rebuilder.py')
| -rw-r--r-- | astroid/tree/rebuilder.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/astroid/tree/rebuilder.py b/astroid/tree/rebuilder.py index 8c6baa9a..90a02f73 100644 --- a/astroid/tree/rebuilder.py +++ b/astroid/tree/rebuilder.py @@ -792,8 +792,8 @@ class TreeRebuilder3(TreeRebuilder): elif node.handlers: return self.visit_tryexcept(node, parent) - def visit_with(self, node, parent): - newnode = nodes.With(node.lineno, node.col_offset, parent) + def visit_with(self, node, parent, constructor=nodes.With): + newnode = constructor(node.lineno, node.col_offset, parent) newnode.postinit([self.visit(item, newnode) for item in node.items], [self.visit(child, newnode) for child in node.body]) return newnode @@ -829,7 +829,7 @@ class TreeRebuilder3(TreeRebuilder): return newnode def visit_asyncwith(self, node, parent): - return self.visit_with(node, parent) + return self.visit_with(node, parent, constructor=nodes.AsyncWith) if sys.version_info >= (3, 0): |