diff --git a/jedi/evaluate/docstrings.py b/jedi/evaluate/docstrings.py index 1b91ae57..f9c3ffb3 100644 --- a/jedi/evaluate/docstrings.py +++ b/jedi/evaluate/docstrings.py @@ -207,7 +207,7 @@ def follow_param(module_context, param): @memoize_default() -def find_return_types(module_context, func): +def infer_return_types(module_context, func): def search_return_in_docstr(code): for p in DOCSTRING_RETURN_PATTERNS: match = p.search(code) diff --git a/jedi/evaluate/pep0484.py b/jedi/evaluate/pep0484.py index d282642b..54f55903 100644 --- a/jedi/evaluate/pep0484.py +++ b/jedi/evaluate/pep0484.py @@ -99,7 +99,7 @@ def py__annotations__(funcdef): @memoize_default() -def find_return_types(context, func): +def infer_return_types(context, func): annotation = py__annotations__(func).get("return", None) return _evaluate_for_annotation(context, annotation) diff --git a/jedi/evaluate/representation.py b/jedi/evaluate/representation.py index 73a75670..57de5daa 100644 --- a/jedi/evaluate/representation.py +++ b/jedi/evaluate/representation.py @@ -330,8 +330,8 @@ class FunctionExecutionContext(context.TreeContext): returns = funcdef.yields else: returns = funcdef.returns - types = set(docstrings.find_return_types(self.get_root_context(), funcdef)) - types |= set(pep0484.find_return_types(self.get_root_context(), funcdef)) + types = set(docstrings.infer_return_types(self.get_root_context(), funcdef)) + types |= set(pep0484.infer_return_types(self.get_root_context(), funcdef)) for r in returns: check = flow_analysis.reachability_check(self, funcdef, r)