Catch StopIteration and reraise cleanly

Closes pylint-dev/pylint#4723
pylint-dev · Jul 19, 2021 · 1e32ea4 · 1e32ea4
1 parent 2738e93
commit 1e32ea4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/astroid/protocols.py b/astroid/protocols.py
@@ -490,8 +490,10 @@ def _infer_context_manager(self, mgr, context):
         else:
             # It doesn't interest us.
             raise InferenceError(node=func)
-
-        yield next(inferred.infer_yield_types())
+        try:
+            yield next(inferred.infer_yield_types())
+        except StopIteration as e:
+            raise InferenceError(node=func) from e
 
     elif isinstance(inferred, bases.Instance):
         try: