diff options
Diffstat (limited to 'astroid/protocols.py')
-rw-r--r-- | astroid/protocols.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/astroid/protocols.py b/astroid/protocols.py index 0397595d..91fe5a34 100644 --- a/astroid/protocols.py +++ b/astroid/protocols.py @@ -477,8 +477,7 @@ def _infer_context_manager(self, mgr, context): return if not context.callcontext: context.callcontext = contextmod.CallContext(args=[inferred]) - for result in enter.infer_call_result(self, context): - yield result + yield from enter.infer_call_result(self, context) @decorators.raise_if_nothing_inferred |