diff --git a/comfy/nodes/package_typing.py b/comfy/nodes/package_typing.py index 64a8cff11..21da741c5 100644 --- a/comfy/nodes/package_typing.py +++ b/comfy/nodes/package_typing.py @@ -11,6 +11,7 @@ class CustomNode(Protocol): RETURN_TYPES: ClassVar[Tuple[str]] RETURN_NAMES: ClassVar[Tuple[str]] = None OUTPUT_IS_LIST: ClassVar[Tuple[bool]] = None + INPUT_IS_LIST: ClassVar[bool] = None FUNCTION: ClassVar[str] CATEGORY: ClassVar[str] OUTPUT_NODE: ClassVar[bool] = None