fix docstrings with evaluator stuff

2013-12-25 18:31:42 +01:00
parent d86f180d2e
commit f039bc3fc9
3 changed files with 5 additions and 8 deletions
--- a/jedi/docstrings.py
+++ b/jedi/docstrings.py
@@ -111,6 +111,8 @@ def find_return_types(evaluator, func):
            if match:
                return match.group(1)

+    from jedi.evaluate import representation as er
+
    if isinstance(func, er.InstanceElement):
        func = func.var

@@ -125,4 +127,4 @@ def find_return_types(evaluator, func):
    if p.user_stmt is None:
        return []
    p.user_stmt.parent = func
-    return list(evaluate.follow_statement(p.user_stmt))
+    return list(evaluator.follow_statement(p.user_stmt))