From b2b08ab4320a0622ac664f360af72fffe5bf8470 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Wed, 22 May 2019 20:33:24 +0200 Subject: [PATCH] Better annotation signature string for classes --- jedi/evaluate/signature.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/jedi/evaluate/signature.py b/jedi/evaluate/signature.py index 321fc7a5..ffd1ea80 100644 --- a/jedi/evaluate/signature.py +++ b/jedi/evaluate/signature.py @@ -16,7 +16,7 @@ class AbstractSignature(object): s = self.name.string_name + '(' + param_code + ')' annotation = self.annotation if annotation is not None: - s += ' -> ' + annotation.get_code(include_prefix=False) + s += ' -> ' + annotation.get_code(include_prefix=False) return s def bind(self, context): @@ -39,6 +39,10 @@ class TreeSignature(AbstractSignature): @property def annotation(self): + # Classes don't need annotations, even if __init__ has one. They always + # return themselves. + if self.context.is_class(): + return None return self._function_context.tree_node.annotation