diff --git a/comfy_api/v3/io.py b/comfy_api/v3/io.py index a7792acb2..c0b31cea9 100644 --- a/comfy_api/v3/io.py +++ b/comfy_api/v3/io.py @@ -140,7 +140,7 @@ def Custom(io_type: str) -> type[ComfyTypeIO]: ... return CustomComfyType -class IO_V3: +class _IO_V3: ''' Base class for V3 Inputs and Outputs. ''' @@ -157,7 +157,7 @@ class IO_V3: def Type(self): return self.Parent.Type -class InputV3(IO_V3): +class InputV3(_IO_V3): ''' Base class for a V3 Input. ''' @@ -206,7 +206,7 @@ class WidgetInputV3(InputV3): return self.widget_type if self.widget_type is not None else super().get_io_type() -class OutputV3(IO_V3): +class OutputV3(_IO_V3): def __init__(self, id: str=None, display_name: str=None, tooltip: str=None, is_output_list=False): self.id = id diff --git a/comfy_extras/nodes_v3_test.py b/comfy_extras/nodes_v3_test.py index 5d201740c..b8e5e2ae0 100644 --- a/comfy_extras/nodes_v3_test.py +++ b/comfy_extras/nodes_v3_test.py @@ -9,12 +9,8 @@ import asyncio @io.comfytype(io_type="XYZ") -class XYZ: +class XYZ(io.ComfyTypeIO): Type = tuple[int,str] - class Input(io.InputV3): - ... - class Output(io.OutputV3): - ... class V3TestNode(io.ComfyNode):