From ec7f8d2fa4408f044ccdd795c37d830c4287e369 Mon Sep 17 00:00:00 2001 From: bigcat88 Date: Thu, 4 Dec 2025 16:18:32 +0200 Subject: [PATCH] fix(api-nodes): the types returned by the "poll_op" function are now correct. --- comfy_api_nodes/util/client.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/comfy_api_nodes/util/client.py b/comfy_api_nodes/util/client.py index 71f34d7f5..bf37cba5f 100644 --- a/comfy_api_nodes/util/client.py +++ b/comfy_api_nodes/util/client.py @@ -89,7 +89,7 @@ async def sync_op( endpoint: ApiEndpoint, *, response_model: type[M], - price_extractor: Callable[[M], float | None] | None = None, + price_extractor: Callable[[M | Any], float | None] | None = None, data: BaseModel | None = None, files: dict[str, Any] | list[tuple[str, Any]] | None = None, content_type: str = "application/json", @@ -133,9 +133,9 @@ async def poll_op( poll_endpoint: ApiEndpoint, *, response_model: type[M], - status_extractor: Callable[[M], str | int | None], - progress_extractor: Callable[[M], int | None] | None = None, - price_extractor: Callable[[M], float | None] | None = None, + status_extractor: Callable[[M | Any], str | int | None], + progress_extractor: Callable[[M | Any], int | None] | None = None, + price_extractor: Callable[[M | Any], float | None] | None = None, completed_statuses: list[str | int] | None = None, failed_statuses: list[str | int] | None = None, queued_statuses: list[str | int] | None = None,