diff --git a/facefusion/exit_helper.py b/facefusion/exit_helper.py index 302949ef..1a37331a 100644 --- a/facefusion/exit_helper.py +++ b/facefusion/exit_helper.py @@ -3,25 +3,26 @@ import signal import sys from time import sleep from types import FrameType +from typing import NoReturn from facefusion import process_manager, state_manager from facefusion.temp_helper import clear_temp_directory from facefusion.types import ErrorCode -def fatal_exit(error_code : ErrorCode) -> None: +def fatal_exit(error_code : ErrorCode) -> NoReturn: os._exit(error_code) -def hard_exit(error_code : ErrorCode) -> None: +def hard_exit(error_code : ErrorCode) -> NoReturn: sys.exit(error_code) -def signal_exit(signum : int, frame : FrameType) -> None: +def signal_exit(signum : int, frame : FrameType) -> NoReturn: graceful_exit(0) -def graceful_exit(error_code : ErrorCode) -> None: +def graceful_exit(error_code : ErrorCode) -> NoReturn: signal.signal(signal.SIGINT, signal.SIG_IGN) process_manager.stop() diff --git a/facefusion/installer.py b/facefusion/installer.py index 553ed08c..51d0a452 100644 --- a/facefusion/installer.py +++ b/facefusion/installer.py @@ -6,6 +6,7 @@ import sys from argparse import ArgumentParser, HelpFormatter from functools import partial from types import FrameType +from typing import NoReturn from facefusion import metadata from facefusion.common_helper import is_linux, is_windows @@ -43,7 +44,7 @@ def cli() -> None: run(program) -def signal_exit(signum : int, frame : FrameType) -> None: +def signal_exit(signum : int, frame : FrameType) -> NoReturn: sys.exit(0)