diff --git a/jedi/parser/tree.py b/jedi/parser/tree.py index df116820..a20030cb 100644 --- a/jedi/parser/tree.py +++ b/jedi/parser/tree.py @@ -879,9 +879,6 @@ class Function(ClassOrFunc): return bool(self.yields) def annotation(self): - if self.children[0] == "lambda": - # lambda functions have no annotation - return None try: if self.children[3] == "->": return _fix_forward_reference(self.children[4]) @@ -964,6 +961,10 @@ class Lambda(Function): def is_generator(self): return False + def annotation(self): + # lambda functions do not support annotations + return None + @property def yields(self): return []