Update return type extraction to return a list

instead of a tuple
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-23 11:49:04 -03:00
commit 49c9187e9c

View file

@ -22,11 +22,11 @@ def extract_inner_type_from_generic_alias(return_type: GenericAlias) -> Any:
return return_type return return_type
def extract_union_types_from_generic_alias(return_type: GenericAlias) -> tuple: def extract_union_types_from_generic_alias(return_type: GenericAlias) -> list:
""" """
Extracts the inner type from a type hint that is a Union. Extracts the inner type from a type hint that is a Union.
""" """
return return_type.__args__ return list(return_type.__args__)
def extract_union_types(return_type: str) -> list[str]: def extract_union_types(return_type: str) -> list[str]: