From 67641a993848ddd01cac1d63a78b3097cd4681e1 Mon Sep 17 00:00:00 2001 From: Dan Davison Date: Tue, 24 Sep 2024 17:44:43 -0400 Subject: [PATCH] Fix return type of workflow.wait --- temporalio/workflow.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/temporalio/workflow.py b/temporalio/workflow.py index 11aa60a4..0c23ea1e 100644 --- a/temporalio/workflow.py +++ b/temporalio/workflow.py @@ -4486,7 +4486,7 @@ async def wait( *, timeout: Optional[float] = None, return_when: str = asyncio.ALL_COMPLETED, -) -> Tuple[List[asyncio.Task[AnyType]], set[asyncio.Task[AnyType]]]: ... +) -> Tuple[List[asyncio.Task[AnyType]], List[asyncio.Task[AnyType]]]: ... async def wait( @@ -4505,9 +4505,9 @@ async def wait( # but the "set" is changed out for a "list" and fixed up some typing/format if asyncio.isfuture(fs) or asyncio.iscoroutine(fs): - raise TypeError(f"expect a list of futures, not {type(fs).__name__}") + raise TypeError(f"Expect an iterable of Tasks/Futures, not {type(fs).__name__}") if not fs: - raise ValueError("Set of Tasks/Futures is empty.") + raise ValueError("Sequence of Tasks/Futures must not be empty.") if return_when not in ( asyncio.FIRST_COMPLETED, asyncio.FIRST_EXCEPTION, @@ -4534,7 +4534,7 @@ async def _wait( # https://github.com/python/cpython/blob/v3.12.3/Lib/asyncio/tasks.py#L522 # but the "set" is changed out for a "list" and fixed up some typing/format - assert fs, "Set of Futures is empty." + assert fs, "Sequence of Tasks/Futures must not be empty." waiter = loop.create_future() timeout_handle = None if timeout is not None: