diff --git a/comfy_extras/nodes_v1_test.py b/comfy_extras/nodes_v1_test.py deleted file mode 100644 index 14ce27896..000000000 --- a/comfy_extras/nodes_v1_test.py +++ /dev/null @@ -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", -} diff --git a/comfy_extras/nodes_v3_test.py b/comfy_extras/nodes_v3_test.py index 978c0575c..34c8b99c4 100644 --- a/comfy_extras/nodes_v3_test.py +++ b/comfy_extras/nodes_v3_test.py @@ -36,22 +36,6 @@ class V3TestNode(io.ComfyNode): 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("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=[ io.Int.Output(), diff --git a/nodes.py b/nodes.py index c8d99ddd8..da4a46366 100644 --- a/nodes.py +++ b/nodes.py @@ -2320,8 +2320,6 @@ async def init_builtin_extra_nodes(): "nodes_camera_trajectory.py", "nodes_edit_model.py", "nodes_tcfg.py", - "nodes_v3_test.py", # TODO: remove - "nodes_v1_test.py", # TODO: remove ] import_failed = []