return the clazz!

2026-02-11 05:52:33 +08:00 · 2023-09-11 17:30:00 +10:00 · 2023-09-11 17:30:00 +10:00 · f076a1aefd
commit f076a1aefd
parent 1325933ebc
1 changed files with 1 additions and 0 deletions
--- a/comfy_extras/ui_decorator.py
+++ b/comfy_extras/ui_decorator.py
@ -35,6 +35,7 @@ def ui_signal(signals:str|list[str]):
        clazz.OUTPUT_NODE = True
        clazz.UI_OUTPUT = clazz.UI_OUTPUT+"," if hasattr(clazz, 'UI_OUTPUT') else ""
        clazz.UI_OUTPUT += ",".join(signals)
+        return clazz

    return decorator