o -wiJ@sddlmZddlZddlmZmZmZddlmZm Z er$ddl m Z edZ edZ Gdd d eZGd d d ejee ZGd d d ee ee ZdS)) annotationsN) TYPE_CHECKINGCallableGeneric)TypeVaroverride) interface_T_Sc@seZdZdZdS)HandleAbandonedErrorz2The handle has no response and has been abandoned.N)__name__ __module__ __qualname____doc__rr]/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/mailbox/mailbox_handle.pyr sr c@sjeZdZdZdddZejdd d Zejdd dZejdddZ ejdddZ ejdddZ dS) MailboxHandlezEA thread-safe handle that allows waiting for a response to a request.fnCallable[[_T], _S]returnMailboxHandle[_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)selfrrrrmaps zMailboxHandle.mapNonecCdS)z>Abandon the handle, indicating it will not receive a response.Nrrrrrabandon$zMailboxHandle.abandonifaceinterface.InterfaceBasecCr)aCancel the handle, requesting any associated work to not complete. This automatically abandons the handle, as a response is no longer guaranteed. Args: iface: The interface on which to publish the cancel request. Nrrrrrrcancel(rzMailboxHandle.cancel _T | NonecCr)z#Returns the response if it's ready.Nrrrrrcheck3rzMailboxHandle.checktimeout float | Noner cCr)aWait for a response or a timeout. 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. Nrrr%rrrwait_or7rzMailboxHandle.wait_orcsdS)aWait for a response or timeout. This must run in an `asyncio` event loop. 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. Nrr'rrr wait_asyncHszMailboxHandle.wait_asyncN)rrrrrrrr rr)rr#)r%r&rr ) r r rrrabcabstractmethodrr"r$r(r)rrrrrs     rc@s`eZdZdZddd Zedd d ZedddZedddZedddZ edddZ dS) rz=A mailbox handle whose result is derived from another handle.handleMailboxHandle[_T]rrrrcCs||_||_dSN)_handle_fn)rr.rrrr__init__]s z_MailboxMappedHandle.__init__cCs|jdSr0)r1rrrrrresz_MailboxMappedHandle.abandonrr cCs|j|dSr0)r1r"r!rrrr"isz_MailboxMappedHandle.cancel _S | NonecCs|j}r ||SdSr0)r1r$r2)rresponserrrr$ms z_MailboxMappedHandle.checkr%r&r cCs||jj|dSN)r%)r2r1r(r'rrrr(tsz_MailboxMappedHandle.wait_orcs |jj|dIdH}||Sr6)r1r)r2)rr%r5rrrr)xs z_MailboxMappedHandle.wait_asyncN)r.r/rrrrr*r+)rr4)r%r&rr ) r r rrr3rrr"r$r(r)rrrrrZs     r) __future__rr,typingrrrtyping_extensionsrrwandb.sdk.interfacerr r Exceptionr ABCrrrrrrs  E