diff options
Diffstat (limited to 'protocols.py')
| -rw-r--r-- | protocols.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/protocols.py b/protocols.py index 1f4f15a6..a26986cb 100644 --- a/protocols.py +++ b/protocols.py @@ -310,10 +310,11 @@ nodes.ExceptHandler.assigned_stmts = raise_if_nothing_infered(excepthandler_assi def with_assigned_stmts(self, node, context=None, asspath=None): if asspath is None: - for lst in self.vars.infer(context): - if isinstance(lst, (nodes.Tuple, nodes.List)): - for item in lst.nodes: - yield item + for _, vars in self.items: + for lst in vars.infer(context): + if isinstance(lst, (nodes.Tuple, nodes.List)): + for item in lst.nodes: + yield item nodes.With.assigned_stmts = raise_if_nothing_infered(with_assigned_stmts) |