diff --git a/comfy_extras/nodes/nodes_generators.py b/comfy_extras/nodes/nodes_generators.py index 1b538b076..b7ed9ecd7 100644 --- a/comfy_extras/nodes/nodes_generators.py +++ b/comfy_extras/nodes/nodes_generators.py @@ -1,5 +1,7 @@ +import typing from decimal import Decimal +from comfy.comfy_types import IO from comfy.node_helpers import export_custom_nodes from comfy.nodes.package_typing import CustomNode, InputTypes @@ -121,4 +123,21 @@ class StringSplit(CustomNode): return value.split(delimiter), +class IterateList(CustomNode): + @classmethod + def INPUT_TYPES(cls) -> InputTypes: + return { + "required": { + "value": (IO.ANY, {}) + } + } + + CATEGORY = "generators" + RETURN_TYPES = IO.ANY, + FUNCTION = "execute" + + def execute(self, value: typing.Any) -> tuple[typing.Any]: + return value, + + export_custom_nodes()