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

Make nested Type[T] annotations work

Browse Source
This commit is contained in:
Peter Law
2020-02-22 22:10:59 +00:00
parent bc53dabce3
commit c03ae0315e
1 changed files with 29 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
jedi/inference/gradual/annotation.py
View File

@@ -343,6 +343,27 @@ def _infer_type_vars(annotation_value, value_set, is_class_value=False):
if annotation_name == 'Type': if annotation_name == 'Type':
given = annotation_value.get_generics() given = annotation_value.get_generics()
if given: if given:
if is_class_value:
for element in value_set:
element_name = element.py__name__()
if annotation_name == element_name:
annotation_generics = annotation_value.get_generics()
actual_generics = element.get_generics()
for annotation_generics_set, actual_generic_set in zip(annotation_generics, actual_generics):
for nested_annotation_value in annotation_generics_set:
_merge_type_var_dicts(
type_var_dict,
_infer_type_vars(
nested_annotation_value,
actual_generic_set,
# This is a note to ourselves that we
# have already converted the instance
# representation to its class.
is_class_value=True,
),
)
else:
for nested_annotation_value in given[0]: for nested_annotation_value in given[0]:
_merge_type_var_dicts( _merge_type_var_dicts(
type_var_dict, type_var_dict,