diff --git a/src/basic_data_handling/control_flow_nodes.py b/src/basic_data_handling/control_flow_nodes.py index 1f8d183..88f343d 100644 --- a/src/basic_data_handling/control_flow_nodes.py +++ b/src/basic_data_handling/control_flow_nodes.py @@ -194,33 +194,6 @@ class SwitchCase(ComfyNodeABC): return (kwargs.get("default"),) -class ExecutionOrder(ComfyNodeABC): - """ - Force execution order in the workflow. - - This node is lightweight and does not affect the workflow. It is used to force - the execution order of nodes in the workflow. You only need to chain this node - with the other execution order nodes in the desired order and add any - output of the nodes you want to force execution order on. - """ - @classmethod - def INPUT_TYPES(cls): - return { - "optional": { - "E/O": ("E/O", {}), - "any node output": (IO.ANY, {}), - } - } - - RETURN_TYPES = ("E/O",) - CATEGORY = "Basic/flow control" - DESCRIPTION = cleandoc(__doc__ or "") - FUNCTION = "execute" - - def execute(self, **kwargs) -> tuple[Any]: - return (None,) - - class FlowSelect(ComfyNodeABC): """ Select the direction of the flow. @@ -252,6 +225,33 @@ class FlowSelect(ComfyNodeABC): return ExecutionBlocker(None), value +class ExecutionOrder(ComfyNodeABC): + """ + Force execution order in the workflow. + + This node is lightweight and does not affect the workflow. It is used to force + the execution order of nodes in the workflow. You only need to chain this node + with the other execution order nodes in the desired order and add any + output of the nodes you want to force execution order on. + """ + @classmethod + def INPUT_TYPES(cls): + return { + "optional": { + "E/O": ("E/O", {}), + "any node output": (IO.ANY, {}), + } + } + + RETURN_TYPES = ("E/O",) + CATEGORY = "Basic/flow control" + DESCRIPTION = cleandoc(__doc__ or "") + FUNCTION = "execute" + + def execute(self, **kwargs) -> tuple[Any]: + return (None,) + + NODE_CLASS_MAPPINGS = { "Basic data handling: IfElse": IfElse, "Basic data handling: IfElifElse": IfElifElse,