Make it possible to gather annotation classes for Union and Optional

2018-11-27 01:14:15 +01:00
parent eb27c64c71
commit 5bb88ca703
4 changed files with 13 additions and 2 deletions
--- a/jedi/evaluate/context/typing.py
+++ b/jedi/evaluate/context/typing.py
@@ -180,6 +180,11 @@ class TypingContextWithIndex(_WithIndexBase):
            self._context_of_index
        )])

+    def gather_annotation_classes(self):
+        return ContextSet.from_sets(
+            _iter_over_arguments(self._index_context, self._context_of_index)
+        )
+

 class TypingContext(_BaseTypingContext):
    index_class = TypingContextWithIndex