fix return type (#146)

ModelSurge · Aug 17, 2023 · 0749265 · 0749265
1 parent 947813e
commit 0749265
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/comfyui_custom_nodes/webui_io.py b/comfyui_custom_nodes/webui_io.py
@@ -1,6 +1,19 @@
 from lib_comfyui import global_state
 
 
+class AnyType(str):
+    def __ne__(self, _) -> bool:
+        return False
+
+
+class AnyReturnTypes(tuple):
+    def __init__(self):
+        super().__init__()
+
+    def __getitem__(self, _):
+        return AnyType()
+
+
 class FromWebui:
     @classmethod
     def INPUT_TYPES(cls):
@@ -9,7 +22,7 @@ def INPUT_TYPES(cls):
                 "void": ("VOID", ),
             },
         }
-    RETURN_TYPES = ()
+    RETURN_TYPES = AnyReturnTypes()
     RETURN_NAMES = ()
     FUNCTION = "get_node_inputs"