diff --git a/facefusion/args_store.py b/facefusion/args_store.py index 1af019cc..04000614 100644 --- a/facefusion/args_store.py +++ b/facefusion/args_store.py @@ -1,7 +1,7 @@ from argparse import Action from typing import List -from facefusion.types import Args, ArgsStore, ArgumentValue, Scope +from facefusion.types import Args, ArgsStore, ArgumentSet, Scope ARGS_STORE : ArgsStore =\ @@ -12,15 +12,15 @@ ARGS_STORE : ArgsStore =\ } -def get_api_set() -> ArgumentValue: +def get_api_set() -> ArgumentSet: return ARGS_STORE.get('api') -def get_cli_set() -> ArgumentValue: +def get_cli_set() -> ArgumentSet: return ARGS_STORE.get('cli') -def get_sys_set() -> ArgumentValue: +def get_sys_set() -> ArgumentSet: return ARGS_STORE.get('sys') diff --git a/facefusion/types.py b/facefusion/types.py index bfbf5eab..3fec4094 100755 --- a/facefusion/types.py +++ b/facefusion/types.py @@ -100,12 +100,13 @@ Resolution : TypeAlias = Tuple[int, int] Args : TypeAlias = Dict[str, Any] Scope : TypeAlias = Literal['api', 'cli', 'sys'] ArgumentValue : TypeAlias = Dict[str, Any] +ArgumentSet : TypeAlias = Dict[str, ArgumentValue] ArgsStore = TypedDict('ArgsStore', { - 'api' : Dict[str, ArgumentValue], - 'cli' : Dict[str, ArgumentValue], - 'sys' : Dict[str, ArgumentValue] + 'api' : ArgumentSet, + 'cli' : ArgumentSet, + 'sys' : ArgumentSet })