danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor

Browse source
This commit is contained in:
Jesse Bannon 2023-12-06 23:51:24 -08:00
parent 2ffc53fc07
commit 81034af97a
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/ytdl_sub/script/types/function.py
View file

@ -107,7 +107,7 @@ class BuiltInFunction(Function, BuiltInFunctionType):
return arg.output_type() return arg.output_type()
return type(arg) return type(arg)
def _output_generic_type(self, union_args: List[Type[Argument]]) -> Type[Resolvable]: def _output_type(self, union_args: List[Type[Argument]]) -> Type[Resolvable]:
union_types_list = set() union_types_list = set()
for union_type in union_args: for union_type in union_args:
possible_output_type = union_type possible_output_type = union_type
@ -124,9 +124,9 @@ class BuiltInFunction(Function, BuiltInFunctionType):
def output_type(self) -> Type[Resolvable]: def output_type(self) -> Type[Resolvable]:
if is_union(self.function_spec.return_type): if is_union(self.function_spec.return_type):
return self._output_generic_type(self.function_spec.return_type.__args__) return self._output_type(self.function_spec.return_type.__args__)
return self._output_generic_type([self.function_spec.return_type]) return self._output_type([self.function_spec.return_type])
def _resolve_lambda_function( def _resolve_lambda_function(
self, self,