o xi@sddlmZddlZddlmZmZddlmZmZddl m Z edZ edZ Gdd d e ZGd d d e ZGd d d ejee ZGdddee ee ZdS)) annotationsN)CallableGeneric)TypeVaroverride)asyncio_manager_T_Sc@eZdZdZdS)HandleAbandonedErrorz2The handle has no response and has been abandoned.N__name__ __module__ __qualname____doc__rrT/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/mailbox/mailbox_handle.pyr r c@r )ServerResponseErrorzThe handle received a generic error response. In practice, this corresponds to receiving the ServerErrorResponse message. Nr rrrrrrrc@sbeZdZdZdddZeddd Zdd dZej dddZ ej dddZ ej dddZ dS) MailboxHandlez0A handle for waiting on a response to a request.asyncerasyncio_manager.AsyncioManagerreturnNonecCs ||_dSN_asyncer)selfrrrr__init__s zMailboxHandle.__init__cCs|jS)z{The asyncio thread to which the handle belongs. The handle's async methods must be run using this object. rrrrrrszMailboxHandle.asyncerfnCallable[[_T], _S]MailboxHandle[_S]cCs t||S)aOReturns a transformed handle. Methods on the returned handle call methods on this handle, but the response type is derived using the given function. Args: fn: A function to apply to this handle's result to get the new handle's result. The function should be pure and fast. )_MailboxMappedHandle)rr rrrmap's zMailboxHandle.mapcCdS)a#Cancel the handle, requesting any associated work to not complete. Any calls to `wait_or` or `wait_async` will raise `HandleAbandonedError` if they aren't resolved within a short time. Cancellation is best-effort. Most exceptions are logged and suppressed. Nrrrrrcancel3zMailboxHandle.canceltimeout float | NonercCr%)aWait for a response or a timeout. It is an error to call this from an async function. On error, including KeyboardInterrupt or a timeout, the handle cancels itself. Args: timeout: A finite number of seconds or None to never time out. If less than or equal to zero, times out immediately unless the response is available. Returns: The response if it arrives before the timeout or has already arrived. Raises: TimeoutError: If the timeout is reached. HandleAbandonedError: If the handle becomes abandoned. ServerResponseError: If the handle received an error response. Nrrr(rrrwait_or=r'zMailboxHandle.wait_orcsdS)abWait for a response or timeout. This must run in an `asyncio` event loop. On error, including asyncio cancellation, KeyboardInterrupt or a timeout, the handle cancels itself. Args: timeout: A finite number of seconds or None to never time out. Returns: The response if it arrives before the timeout or has already arrived. Raises: TimeoutError: If the timeout is reached. HandleAbandonedError: If the handle becomes abandoned. ServerResponseError: If the handle received an error response. Nrr*rrr wait_asyncSszMailboxHandle.wait_asyncN)rrrr)rr)r r!rr"rr)r(r)rr) r rrrrpropertyrr$abcabstractmethodr&r+r,rrrrrs     rcsLeZdZdZdfdd Zedd d ZedddZedddZZ S)r#z=A mailbox handle whose result is derived from another handle.handleMailboxHandle[_T]r r!rrcst|j||_||_dSr)superrr_handle_fn)rr1r  __class__rrrks z_MailboxMappedHandle.__init__cCs|jdSr)r4r&rrrrr&tsz_MailboxMappedHandle.cancelr(r)r cCs||jj|dSN)r()r5r4r+r*rrrr+xsz_MailboxMappedHandle.wait_orcs |jj|dIdH}||Sr8)r4r,r5)rr(responserrrr,|s z_MailboxMappedHandle.wait_async)r1r2r r!rrr-)r(r)rr ) r rrrrrr&r+r, __classcell__rrr6rr#hs  r#) __future__rr/typingrrtyping_extensionsrr wandb.sdk.librrr Exceptionr rABCrr#rrrrs  O