Fix stub function inferrals

2019-05-28 01:51:37 +02:00
parent 6f41530a03
commit ec7b6b8d80
5 changed files with 65 additions and 27 deletions
--- a/jedi/evaluate/context/klass.py
+++ b/jedi/evaluate/context/klass.py
@@ -272,11 +272,11 @@ class ClassContext(use_metaclass(CachedMetaClass, ClassMixin, FunctionAndClassBa
                yield type_var_dict.get(type_var.py__name__(), NO_CONTEXTS)

        if type_var_dict:
-            return GenericClass(
+            return ContextSet([GenericClass(
                self,
                generics=tuple(remap_type_vars())
-            )
-        return self
+            )])
+        return ContextSet({self})

    def get_signatures(self):
        init_funcs = self.py__getattribute__('__init__')