From a94209507dba8f6a12166129702316d202df1bbd Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Tue, 17 Feb 2026 15:35:08 +0530 Subject: [PATCH] introduce type ArgumentSet --- facefusion/args_store.py | 8 ++++---- facefusion/types.py | 7 ++++--- 2 files changed, 8 insertions(+), 7 deletions(-) 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 })