From aac40e4fca5bd2b87a3d709fce1023bf7f19b39b Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Thu, 12 Feb 2026 22:29:15 -0800 Subject: [PATCH] Skip replacements that lead to a missing node_id anyway --- app/node_replace_manager.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/app/node_replace_manager.py b/app/node_replace_manager.py index f6cdeceb4..26c7440ed 100644 --- a/app/node_replace_manager.py +++ b/app/node_replace_manager.py @@ -64,9 +64,13 @@ class NodeReplaceManager: # just use the first replacement replacement = replacements[0] new_node_id = replacement.new_node_id + # if replacement is not a valid node, skip trying to replace it as will only cause confusion + if new_node_id not in NODE_CLASS_MAPPINGS.keys(): + continue # first, replace node id (class_type) new_node_struct = copy_node_struct(node_struct, empty_inputs=True) new_node_struct["class_type"] = new_node_id + # TODO: consider replacing display_name in _meta as well for error reporting purposes; would need to query node schema # second, replace inputs if replacement.input_mapping is not None: for input_map in replacement.input_mapping: