diff --git a/comfy/api/paths/view/get/parameters/parameter_1/schema.py b/comfy/api/paths/view/get/parameters/parameter_1/schema.py index e406907e0..df880cbb0 100644 --- a/comfy/api/paths/view/get/parameters/parameter_1/schema.py +++ b/comfy/api/paths/view/get/parameters/parameter_1/schema.py @@ -13,19 +13,22 @@ from comfy.api.shared_imports.schema_imports import * # pyright: ignore [report class SchemaEnums: - - @schemas.classproperty - def OUTPUT(cls) -> typing.Literal["output"]: + @classmethod + def output(cls) -> typing.Literal["output"]: return Schema.validate("output") - @schemas.classproperty - def INPUT(cls) -> typing.Literal["input"]: + @classmethod + def input(cls) -> typing.Literal["input"]: return Schema.validate("input") - @schemas.classproperty - def TEMP(cls) -> typing.Literal["temp"]: + @classmethod + def temp(cls) -> typing.Literal["temp"]: return Schema.validate("temp") + OUTPUT = property(output) + INPUT = property(input) + TEMP = property(temp) + @dataclasses.dataclass(frozen=True) class Schema( diff --git a/comfy/api/schemas/__init__.py b/comfy/api/schemas/__init__.py index 1ea0c6bd4..887fd7859 100644 --- a/comfy/api/schemas/__init__.py +++ b/comfy/api/schemas/__init__.py @@ -14,7 +14,6 @@ import typing_extensions from .schema import ( get_class, none_type_, - classproperty, Bool, FileIO, Schema, @@ -104,7 +103,6 @@ def raise_if_key_known( __all__ = [ 'get_class', 'none_type_', - 'classproperty', 'Bool', 'FileIO', 'Schema',