mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-10 06:10:50 +08:00
30 lines
1.3 KiB
Python
30 lines
1.3 KiB
Python
from .execution_context import current_execution_context
|
|
from typing import Any, Generator, Sequence
|
|
|
|
|
|
def is_input_unscheduled_unexecuted(*arg_names: Sequence[str]) -> Generator[bool, Any, None]:
|
|
"""
|
|
returns true if the given argument in the context of an executing node is not scheduled nor executed
|
|
this will be true for inputs that are marked as lazy, and this method is more robust against nodes that return None
|
|
:param arg_names: each arg to evaluate
|
|
:return: True for each arg that is not scheduled nor executed
|
|
"""
|
|
context = current_execution_context()
|
|
if context is None or context.execution_list is None:
|
|
raise LookupError("Not executing a node")
|
|
assert context.execution_list is not None
|
|
dynprompt = context.execution_list.dynprompt
|
|
executed = context.executed
|
|
execution_list = context.execution_list
|
|
inputs = context.inputs
|
|
unscheduled_unexecuted = dynprompt.all_node_ids() - executed - set(execution_list.pendingNodes.keys())
|
|
for arg_name in arg_names:
|
|
if arg_name not in inputs:
|
|
raise ValueError(f"Input {arg_name} not found")
|
|
input_ = inputs[arg_name]
|
|
if isinstance(input_, list) or isinstance(input_, tuple) and len(input_) == 2:
|
|
node_id, *_ = input_
|
|
yield node_id in unscheduled_unexecuted
|
|
else:
|
|
yield False
|