Use is_link for checking if an input is a link instead of looking for list status

This commit is contained in:
Jedrzej Kosinski
2026-02-14 19:13:25 -08:00
parent 962f62d7ca
commit 1768392db2

View File

@@ -6,6 +6,7 @@ from typing import TYPE_CHECKING, TypedDict
if TYPE_CHECKING:
from comfy_api.latest._node_replace import NodeReplace
from comfy_execution.graph_utils import is_link
from nodes import NODE_CLASS_MAPPINGS
class NodeStruct(TypedDict):
@@ -51,7 +52,7 @@ class NodeReplaceManager:
need_replacement.add(node_number)
# keep track of connections
for input_id, input_value in node_struct["inputs"].items():
if isinstance(input_value, list):
if is_link(input_value):
conn_number = input_value[0]
connections.setdefault(conn_number, []).append((node_number, input_id, input_value[1]))
if len(need_replacement) > 0: