Change entangled nodes to only get actual types

guill · Aug 29, 2024 · b92f876 · b92f876
1 parent 02b13c7
commit b92f876
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/comfy_execution/graph.py b/comfy_execution/graph.py
@@ -154,7 +154,19 @@ def resolve_dynamic_definitions(self, node_id_set: Set[str]):
             if hasattr(class_def, "resolve_dynamic_types"):
                 entangled_types = {}
                 for (entangled_id, entangled_name) in entangled.get(node_id, []):
-                    input_types, output_types = self.get_input_output_types(entangled_id)
+                    entangled_def = self.get_node_definition(entangled_id)
+                    if entangled_def is None:
+                        continue
+                    input_types = {}
+                    output_types = {}
+                    for input_category, input_list in entangled_def["input"].items():
+                        for input_name, input_info in input_list.items():
+                            if isinstance(input_info, tuple) or input_category != "hidden":
+                                input_types[input_name] = input_info[0]
+                    for i in range(len(entangled_def["output"])):
+                        output_name = entangled_def["output_name"][i]
+                        output_types[output_name] = entangled_def["output"][i]
+
                     if entangled_name not in entangled_types:
                         entangled_types[entangled_name] = []
                     entangled_types[entangled_name].append({