wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-21 03:50:50 +08:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Figured out Switch situation
Some checks are pending
Python Linting / Run Ruff (push) Waiting to run
Details
Python Linting / Run Pylint (push) Waiting to run
Details

Browse Source
This commit is contained in:
Jedrzej Kosinski 2025-12-16 21:55:20 -08:00
parent 85a97cc5c8
commit 6630cd8135
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_api/latest/_io.py
View File

@ -1838,7 +1838,7 @@ class NodeOutput(_NodeOutputInternal):
return self.args if len(self.args) > 0 else None
@classmethod
def from_dict(cls, data: dict[str, Any]) -> "NodeOutput":
def from_dict(cls, data: dict[str, Any]) -> NodeOutput:
args = ()
ui = None
expand = None

15
comfy_extras/nodes_logic.py
View File

@ -7,13 +7,13 @@ from comfy_api.latest import _io
MISSING = object()
class SwitchNode(io.ComfyNode):
class SoftSwitchNode(io.ComfyNode):
@classmethod
def define_schema(cls):
template = io.MatchType.Template("switch")
return io.Schema(
node_id="ComfySwitchNode",
display_name="Switch",
node_id="ComfySoftSwitchNode",
display_name="Soft Switch",
category="logic",
is_experimental=True,
inputs=[
@ -58,14 +58,15 @@ class SwitchNode(io.ComfyNode):
return io.NodeOutput(on_true)
return io.NodeOutput(on_true if switch else on_false)
class SwitchNode2(io.ComfyNode):
class SwitchNode(io.ComfyNode):
@classmethod
def define_schema(cls):
template = io.MatchType.Template("switch")
return io.Schema(
node_id="ComfySwitchNode2",
display_name="Switch2",
node_id="ComfySwitchNode",
display_name="Switch",
category="logic",
is_experimental=True,
inputs=[
io.Boolean.Input("switch"),
io.MatchType.Input("on_false", template=template, lazy=True),
@ -214,7 +215,7 @@ class LogicExtension(ComfyExtension):
async def get_node_list(self) -> list[type[io.ComfyNode]]:
return [
SwitchNode,
SwitchNode2,
SoftSwitchNode,
CustomComboNode,
DCTestNode,
AutogrowNamesTestNode,