diff options
Diffstat (limited to 'astroid/protocols.py')
-rw-r--r-- | astroid/protocols.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/astroid/protocols.py b/astroid/protocols.py index 8bd3813..21f0c29 100644 --- a/astroid/protocols.py +++ b/astroid/protocols.py @@ -414,6 +414,8 @@ def _infer_context_manager(self, mgr, context): return if not isinstance(enter, bases.BoundMethod): return + if not context.callcontext: + context.callcontext = contextmod.CallContext(args=[inferred]) for result in enter.infer_call_result(self, context): yield result |