mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-08-02 15:04:50 +08:00
Removed nodes_v1_test.py, removed both v1/v3 test nodes from being attempted to be loaded in nodes.py
This commit is contained in:
@@ -1,77 +0,0 @@
|
|||||||
import torch
|
|
||||||
from comfy.comfy_types.node_typing import ComfyNodeABC, IO
|
|
||||||
import asyncio
|
|
||||||
from comfy.utils import ProgressBar
|
|
||||||
import time
|
|
||||||
|
|
||||||
|
|
||||||
class TestNode(ComfyNodeABC):
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def INPUT_TYPES(cls):
|
|
||||||
return {
|
|
||||||
"required": {
|
|
||||||
"image": (IO.IMAGE,),
|
|
||||||
"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"}),
|
|
||||||
"combo2": (IO.COMBO, {"options": ["a", "b", "c"], "multi_select": True, "tooltip": "This is a combo input"}),
|
|
||||||
},
|
|
||||||
"optional": {
|
|
||||||
"xyz": ("XYZ",),
|
|
||||||
"mask": (IO.MASK,),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
RETURN_TYPES = (IO.INT, IO.IMAGE)
|
|
||||||
RETURN_NAMES = ("INT", "img🖼️")
|
|
||||||
OUTPUT_TOOLTIPS = (None, "This is an image")
|
|
||||||
FUNCTION = "do_thing"
|
|
||||||
|
|
||||||
OUTPUT_NODE = True
|
|
||||||
|
|
||||||
CATEGORY = "v3 nodes"
|
|
||||||
|
|
||||||
def do_thing(self, image: torch.Tensor, some_int: int, combo: str, combo2: list[str], xyz=None, mask: torch.Tensor=None):
|
|
||||||
return (some_int, image)
|
|
||||||
|
|
||||||
|
|
||||||
class TestSleep(ComfyNodeABC):
|
|
||||||
@classmethod
|
|
||||||
def INPUT_TYPES(cls):
|
|
||||||
return {
|
|
||||||
"required": {
|
|
||||||
"value": (IO.ANY, {}),
|
|
||||||
"seconds": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 9999.0, "step": 0.01, "tooltip": "The amount of seconds to sleep."}),
|
|
||||||
},
|
|
||||||
"hidden": {
|
|
||||||
"unique_id": "UNIQUE_ID",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
RETURN_TYPES = (IO.ANY,)
|
|
||||||
FUNCTION = "sleep"
|
|
||||||
|
|
||||||
CATEGORY = "_for_testing"
|
|
||||||
|
|
||||||
async def sleep(self, value, seconds, unique_id):
|
|
||||||
pbar = ProgressBar(seconds, node_id=unique_id)
|
|
||||||
start = time.time()
|
|
||||||
expiration = start + seconds
|
|
||||||
now = start
|
|
||||||
while now < expiration:
|
|
||||||
now = time.time()
|
|
||||||
pbar.update_absolute(now - start)
|
|
||||||
await asyncio.sleep(0.02)
|
|
||||||
return (value,)
|
|
||||||
|
|
||||||
|
|
||||||
NODE_CLASS_MAPPINGS = {
|
|
||||||
"V1TestNode1": TestNode,
|
|
||||||
"V1TestSleep": TestSleep,
|
|
||||||
}
|
|
||||||
|
|
||||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
||||||
"V1TestNode1": "V1 Test Node",
|
|
||||||
"V1TestSleep": "V1 Test Sleep",
|
|
||||||
}
|
|
@@ -36,22 +36,6 @@ class V3TestNode(io.ComfyNode):
|
|||||||
io.MultiCombo.Input("combo2", options=["a","b","c"]),
|
io.MultiCombo.Input("combo2", options=["a","b","c"]),
|
||||||
io.MultiType.Input(io.Int.Input("int_multitype", display_name="haha"), types=[io.Float]),
|
io.MultiType.Input(io.Int.Input("int_multitype", display_name="haha"), types=[io.Float]),
|
||||||
io.MultiType.Input("multitype", types=[io.Mask, io.Float, io.Int], optional=True),
|
io.MultiType.Input("multitype", types=[io.Mask, io.Float, io.Int], optional=True),
|
||||||
# ComboInput("combo", image_upload=True, image_folder=FolderType.output,
|
|
||||||
# remote=RemoteOptions(
|
|
||||||
# route="/internal/files/output",
|
|
||||||
# refresh_button=True,
|
|
||||||
# ),
|
|
||||||
# tooltip="This is a combo input"),
|
|
||||||
# IntegerInput("some_int", display_name="new_name", min=0, tooltip="My tooltip 😎", display=NumberDisplay.slider, ),
|
|
||||||
# ComboDynamicInput("mask", behavior=InputBehavior.optional),
|
|
||||||
# IntegerInput("some_int", display_name="new_name", min=0, tooltip="My tooltip 😎", display=NumberDisplay.slider,
|
|
||||||
# dependent_inputs=[ComboDynamicInput("mask", behavior=InputBehavior.optional)],
|
|
||||||
# dependent_values=[lambda my_value: IO.STRING if my_value < 5 else IO.NUMBER],
|
|
||||||
# ),
|
|
||||||
# ["option1", "option2". "option3"]
|
|
||||||
# ComboDynamicInput["sdfgjhl", [ComboDynamicOptions("option1", [IntegerInput("some_int", display_name="new_name", min=0, tooltip="My tooltip 😎", display=NumberDisplay.slider, ImageInput(), MaskInput(), String()]),
|
|
||||||
# CombyDynamicOptons("option2", [])
|
|
||||||
# ]]
|
|
||||||
],
|
],
|
||||||
outputs=[
|
outputs=[
|
||||||
io.Int.Output(),
|
io.Int.Output(),
|
||||||
|
2
nodes.py
2
nodes.py
@@ -2320,8 +2320,6 @@ async def init_builtin_extra_nodes():
|
|||||||
"nodes_camera_trajectory.py",
|
"nodes_camera_trajectory.py",
|
||||||
"nodes_edit_model.py",
|
"nodes_edit_model.py",
|
||||||
"nodes_tcfg.py",
|
"nodes_tcfg.py",
|
||||||
"nodes_v3_test.py", # TODO: remove
|
|
||||||
"nodes_v1_test.py", # TODO: remove
|
|
||||||
]
|
]
|
||||||
|
|
||||||
import_failed = []
|
import_failed = []
|
||||||
|
Reference in New Issue
Block a user