mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-13 21:16:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add comfytype decorator, convert all relevant v3_01 types to follow new convention, make v1 test node have xyz be optional

Browse Source
This commit is contained in:
kosinkadink1@gmail.com
2025-06-13 04:06:06 -07:00
parent cf7312d82c
commit 54e0d6b161
3 changed files with 243 additions and 209 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_v1_test.py
View File

@@ -9,13 +9,13 @@ class TestNode(ComfyNodeABC):
return {
"required": {
"image": (IO.IMAGE,),
"xyz": ("XYZ",),
"some_int": (IO.INT, {"display_name": "new_name",
"min": 0, "max": 127, "default": 42,
"tooltip": "My tooltip 😎", "display": "slider"}),
"combo": (IO.COMBO, {"options": ["a", "b", "c"], "tooltip": "This is a combo input"}),
},
"optional": {
"xyz": ("XYZ",),
"mask": (IO.MASK,),
}
}
@@ -29,7 +29,7 @@ class TestNode(ComfyNodeABC):
CATEGORY = "v3 nodes"
def do_thing(self, image: torch.Tensor, xyz, some_int: int, combo: str, mask: torch.Tensor=None):
def do_thing(self, image: torch.Tensor, some_int: int, combo: str, xyz=None, mask: torch.Tensor=None):
return (some_int, image)

17
comfy_extras/nodes_v3_01_test.py
View File

@@ -3,14 +3,14 @@ from comfy_api.v3_01 import io
import logging
@io.comfytype(io_type="XYZ")
class XYZ:
Type = tuple[int,str]
class Input(io.InputV3, io_type="XYZ"):
class Input(io.InputV3):
...
class Output(io.OutputV3, io_type="XYZ"):
class Output(io.OutputV3):
...
class V3TestNode(io.ComfyNodeV3):
def __init__(self):
@@ -28,9 +28,10 @@ class V3TestNode(io.ComfyNodeV3):
XYZ.Input("xyz", optional=True),
#CustomInput("xyz", "XYZ", optional=True),
io.Mask.Input("mask", optional=True),
io.Integer.Input("some_int", display_name="new_name", min=0, max=127, default=42,
io.Int.Input("some_int", display_name="new_name", min=0, max=127, default=42,
tooltip="My tooltip 😎", display_mode=io.NumberDisplay.slider),
io.ComboInput("combo", options=["a", "b", "c"], tooltip="This is a combo input"),
io.Combo.Input("combo", options=["a", "b", "c"], tooltip="This is a combo input"),
io.MultiCombo.Input("combo2", options=["a","b","c"]),
# ComboInput("combo", image_upload=True, image_folder=FolderType.output,
# remote=RemoteOptions(
# route="/internal/files/output",
@@ -49,7 +50,7 @@ class V3TestNode(io.ComfyNodeV3):
# ]]
],
outputs=[
io.Integer.Output("int_output"),
io.Int.Output("int_output"),
io.Image.Output("img_output", display_name="img🖼️", tooltip="This is an image"),
],
hidden=[
@@ -59,8 +60,8 @@ class V3TestNode(io.ComfyNodeV3):
)
@classmethod
def execute(cls, image: io.Image.Type, some_int: int, combo: io.ComboInput.Type, xyz: XYZ.Type=None, mask: io.Mask.Type=None):
some_int
def execute(cls, image: io.Image.Type, some_int: int, combo: io.Combo.Type, combo2: io.MultiCombo.Type, xyz: XYZ.Type=None, mask: io.Mask.Type=None):
#some_int
if hasattr(cls, "hahajkunless"):
raise Exception("The 'cls' variable leaked instance state between runs!")
if hasattr(cls, "doohickey"):