returns -> iter_return_stmts

2017-04-30 01:45:59 +02:00
parent b4039872bd
commit 606871eb62
2 changed files with 3 additions and 6 deletions
--- a/jedi/evaluate/representation.py
+++ b/jedi/evaluate/representation.py
@@ -321,7 +321,7 @@ class FunctionExecutionContext(context.TreeContext):
            types = set()
            returns = funcdef.yields
        else:
-            returns = funcdef.returns
+            returns = funcdef.iter_return_stmts()
            types = set(docstrings.infer_return_types(self.function_context))
            types |= set(pep0484.infer_return_types(self.function_context))