VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-07 06:24:27 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix forward reference resolving

Browse Source
This commit is contained in:
Dave Halter
2018-09-26 09:18:04 +02:00
parent 4bb7a595e8
commit b5b0214c3c
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
jedi/evaluate/pep0484.py
View File

@@ -218,10 +218,12 @@ def infer_return_types(function_execution_context):
context = function_execution_context.function_context.get_default_param_context() context = function_execution_context.function_context.get_default_param_context()
unknown_type_vars = list(find_unknown_type_vars(context, annotation)) unknown_type_vars = list(find_unknown_type_vars(context, annotation))
# TODO this function should return the annotation context.
annotation = _fix_forward_reference(context, annotation)
annotation_contexts = context.eval_node(annotation)
if not unknown_type_vars: if not unknown_type_vars:
return context.eval_node(annotation).execute_annotation() return annotation_contexts.execute_annotation()
annotations_contexts = context.eval_node(annotation)
type_var_dict = infer_type_vars_for_execution(function_execution_context, all_annotations) type_var_dict = infer_type_vars_for_execution(function_execution_context, all_annotations)
def remap_type_vars(context, type_var_dict): def remap_type_vars(context, type_var_dict):
@@ -244,7 +246,7 @@ def infer_return_types(function_execution_context):
define_type_vars( define_type_vars(
annotation_context, annotation_context,
remap_type_vars(annotation_context, type_var_dict), remap_type_vars(annotation_context, type_var_dict),
) for annotation_context in annotations_contexts ) for annotation_context in annotation_contexts
).execute_annotation() ).execute_annotation()