summaryrefslogtreecommitdiff
path: root/astroid/protocols.py
diff options
context:
space:
mode:
Diffstat (limited to 'astroid/protocols.py')
-rw-r--r--astroid/protocols.py3
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