From 5464f4c3de3f38b4ba493147b69c3af965438a40 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 30 Jun 2025 20:03:28 +0200 Subject: [PATCH] Cosmetics on types --- facefusion_api/core.py | 6 +++--- facefusion_api/types.py | 5 ++--- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/facefusion_api/core.py b/facefusion_api/core.py index 766574c..38a35e5 100644 --- a/facefusion_api/core.py +++ b/facefusion_api/core.py @@ -11,12 +11,12 @@ from comfy_api_nodes.apinode_utils import bytesio_to_image_tensor, tensor_to_byt from httpx import Client as HttpClient, Headers from torch import Tensor -from .types import FaceSwapperModel, NodeInputTypes +from .types import FaceSwapperModel, InputTypes class SwapFaceImage: @classmethod - def INPUT_TYPES(s) -> NodeInputTypes: + def INPUT_TYPES(s) -> InputTypes: return\ { 'required': @@ -83,7 +83,7 @@ class SwapFaceImage: class SwapFaceVideo: @classmethod - def INPUT_TYPES(s) -> NodeInputTypes: + def INPUT_TYPES(s) -> InputTypes: return\ { 'required': diff --git a/facefusion_api/types.py b/facefusion_api/types.py index 42fc070..5d2869c 100644 --- a/facefusion_api/types.py +++ b/facefusion_api/types.py @@ -1,9 +1,8 @@ from typing import Any, Dict, Literal, TypeAlias -NodeInputTypes : TypeAlias = Dict[str, Any] +InputTypes : TypeAlias = Dict[str, Any] + NodeClassMapping : TypeAlias = Dict[str, Any] NodeDisplayNameMapping : TypeAlias = Dict[str, str] -Image : TypeAlias = Any - FaceSwapperModel = Literal['hyperswap_1a_256', 'hyperswap_1b_256', 'hyperswap_1c_256']