From e970218f6f591264f2b0512f46609c78e257e0b7 Mon Sep 17 00:00:00 2001 From: Jacob Walls Date: Sun, 30 Apr 2023 08:29:02 -0400 Subject: Yield directly from _explicit_inference in NodeNG.infer() (#2157) --- astroid/nodes/node_ng.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'astroid') diff --git a/astroid/nodes/node_ng.py b/astroid/nodes/node_ng.py index 52852bf9..de5dec77 100644 --- a/astroid/nodes/node_ng.py +++ b/astroid/nodes/node_ng.py @@ -136,11 +136,12 @@ class NodeNG: if self._explicit_inference is not None: # explicit_inference is not bound, give it self explicitly try: - # pylint: disable=not-callable - results = list(self._explicit_inference(self, context, **kwargs)) - if context is not None: - context.nodes_inferred += len(results) - yield from results + if context is None: + yield from self._explicit_inference(self, context, **kwargs) + return + for result in self._explicit_inference(self, context, **kwargs): + context.nodes_inferred += 1 + yield result return except UseInferenceDefault: pass -- cgit v1.2.1