From 34c7e5f45e7fe3cfd3c5552e46855a1861025802 Mon Sep 17 00:00:00 2001 From: Chris Date: Mon, 11 Sep 2023 18:04:54 +1000 Subject: [PATCH] fix key --- comfy_extras/ui_decorator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy_extras/ui_decorator.py b/comfy_extras/ui_decorator.py index a9090ebca..03f84506a 100644 --- a/comfy_extras/ui_decorator.py +++ b/comfy_extras/ui_decorator.py @@ -27,7 +27,7 @@ def ui_signal(signals:str|list[str]): returns_tuple = returns_tuple[:-len(signals)] for i,key in enumerate(signals): - returns_ui['key'] = popped_returns[i] + returns_ui[key] = popped_returns[i] return { "ui":returns_ui, "result": returns_tuple } clazz._ui_signal_decorated_function = _ui_signal_decorated_function