1
0
forked from VimPlug/jedi

Fix generator issues for typing

This commit is contained in:
Dave Halter
2019-06-08 01:50:38 +02:00
parent 26951f5c18
commit eef02e5c56
4 changed files with 13 additions and 4 deletions

View File

@@ -76,7 +76,7 @@ class GeneratorBase(LazyAttributeOverwrite, IterableMixin):
@property
def name(self):
return compiled.CompiledContextName(self, 'generator')
return compiled.CompiledContextName(self, 'Generator')
class Generator(GeneratorBase):

View File

@@ -1,3 +1,4 @@
from jedi import debug
from jedi.evaluate.base_context import ContextSet, \
NO_CONTEXTS
from jedi.evaluate.utils import to_list
@@ -49,11 +50,13 @@ def _infer_from_stub(stub_module, qualified_names, ignore_compiled):
def try_stubs_to_actual_context_set(stub_contexts, prefer_stub_to_compiled=False):
return ContextSet.from_sets(
contexts = ContextSet.from_sets(
stub_to_actual_context_set(stub_context, ignore_compiled=prefer_stub_to_compiled)
or ContextSet([stub_context])
for stub_context in stub_contexts
)
debug.dbg('Stubs to actual: %s to %s', stub_contexts, contexts)
return contexts
@to_list