wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-15 13:02:35 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Add Boolean support to math expressions

Browse Source
This commit is contained in:
LaVie024 2026-03-29 22:40:59 +00:00 committed by GitHub
parent d113d1cc32
commit 43f2440972
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy_extras/nodes_math.py
View File

@ -63,7 +63,7 @@ class MathExpressionNode(io.ComfyNode):
@classmethod @classmethod
def define_schema(cls) -> io.Schema: def define_schema(cls) -> io.Schema:
autogrow = io.Autogrow.TemplateNames( autogrow = io.Autogrow.TemplateNames(
input=io.MultiType.Input("value", [io.Float, io.Int]), input=io.MultiType.Input("value", [io.Float, io.Int, io.Boolean]),
names=list(string.ascii_lowercase), names=list(string.ascii_lowercase),
min=1, min=1,
) )
@ -82,6 +82,7 @@ class MathExpressionNode(io.ComfyNode):
outputs=[ outputs=[
io.Float.Output(display_name="FLOAT"), io.Float.Output(display_name="FLOAT"),
io.Int.Output(display_name="INT"), io.Int.Output(display_name="INT"),
io.Boolean.Output(display_name="BOOL"),
], ],
) )
@ -97,7 +98,7 @@ class MathExpressionNode(io.ComfyNode):
result = simple_eval(expression, names=context, functions=MATH_FUNCTIONS) result = simple_eval(expression, names=context, functions=MATH_FUNCTIONS)
# bool check must come first because bool is a subclass of int in Python # bool check must come first because bool is a subclass of int in Python
if isinstance(result, bool) or not isinstance(result, (int, float)): if not isinstance(result, (int, float)):
raise ValueError( raise ValueError(
f"Math Expression '{expression}' must evaluate to a numeric result, " f"Math Expression '{expression}' must evaluate to a numeric result, "
f"got {type(result).__name__}: {result!r}" f"got {type(result).__name__}: {result!r}"
@ -106,7 +107,7 @@ class MathExpressionNode(io.ComfyNode):
raise ValueError( raise ValueError(
f"Math Expression '{expression}' produced a non-finite result: {result}" f"Math Expression '{expression}' produced a non-finite result: {result}"
) )
return io.NodeOutput(float(result), int(result)) return io.NodeOutput(float(result), int(result), bool(result))
class MathExtension(ComfyExtension): class MathExtension(ComfyExtension):