o 1 iX@sddlZddlZddlZddlZddlZddlmZmZmZm Z m Z m Z m Z m Z mZddlZddlmZddlmZmZmZddlmZddlmZe dZeeZde ed e ee effd d Z d e e!e"ffd d Z#de e!effddZ$ded e"fddZ%  d deegefee e!efgeffde e e!efdede e!ded egeff ddZ&Gdddej'Z(GdddZ)dS)!N) AnyCallableContextManagerDictListOptionalTupleTypeVarUnion) ActorHandle)StartTracebackStartTracebackWithWorkerRankfind_free_port) RayActorError) ObjectRefT remote_valuesreturnc Cs|}t|dkr_t|\}}|D]E}zt|WqtyB}z||}td|dd|fWYd}~Sd}~wt yX}z ||}t |d|d}~wwt|dks dS)a$Check for actor failure when retrieving the remote values. Args: remote_values: List of object references from Ray actor methods. Returns: A tuple of (bool, Exception). The bool is True if evaluating all object references is successful, False otherwise. rzWorker z has failed.FN)Z worker_rank)TN) copylenraywaitgetrindexloggerinfo Exceptionr )r unfinishedfinishedZ object_refexcZfailed_actor_rankZfailed_worker_rankr e/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/train/_internal/utils.pycheck_for_failure!s,    r"cCstj}t}||fS)z4Returns the IP address and a free port on this node.)rutilZget_node_ip_addressr)addrportr r r!get_address_and_portGs r&env_varscCs"dd|D}tj|dS)zyUpdates the environment variables on this worker process. Args: env_vars: Environment variables to set. cSsi|] \}}|t|qSr )str).0kvr r r! Usz#update_env_vars..N)itemsosenvironupdate)r'Z sanitizedr r r!update_env_varsOsr1fncs<t|j}tjjtjjtjjhtfdd|DS)aCounts the number of required parameters of a function. NOTE: *args counts as 1 required parameter. Examples -------- >>> def fn(a, b, /, c, *args, d=1, e=2, **kwargs): ... pass >>> count_required_parameters(fn) 4 >>> fn = lambda: 1 >>> count_required_parameters(fn) 0 >>> def fn(config, a, b=1, c=2): ... pass >>> from functools import partial >>> count_required_parameters(partial(fn, a=0)) 1 cs(g|]}|jtjjkr|jvr|qSr )defaultinspect Parameteremptykind)r)pZpositional_param_kindsr r! xs z-count_required_parameters..) r4 signature parametersvaluesr5POSITIONAL_ONLYPOSITIONAL_OR_KEYWORDVAR_POSITIONALr)r2paramsr r9r!count_required_parametersYs rB train_funcFconfigtrain_func_context fn_arg_namediscard_returnsc st}|rtfdd}|n|dkr&|d|d}t||dkrAdur0intfdd}|Stfd d}|S) aValidates and constructs the training function to execute. Args: train_func: The training function to execute. This can either take in no arguments or a ``config`` dict. config (Optional[Dict]): Configurations to pass into ``train_func``. If None then an empty Dict will be created. train_func_context: Context manager for user's `train_func`, which executes backend-specific logic before and after the training function. fn_arg_name (Optional[str]): The name of training function to use for error messages. discard_returns: Whether to discard any returns from train_func or not. Returns: A valid training function. Raises: ValueError: if the input ``train_func`` is invalid. c s4z |i|WdSty}zt|d}~wwNrr )argskwargse)rCr r!discard_return_wrappers z4construct_train_func..discard_return_wrapperz: should take in 0 or 1 required arguments, but it accepts z required arguments instead.Nc sVz WdWS1swYWdSty*}zt|d}~wwrHrIrL)rDrEwrapped_train_funcr r!train_fns(z&construct_train_func..train_fnc sTz WdWS1swYWdSty)}zt|d}~wwrHrIrO)rErPr r!rQs()rB functoolswraps ValueError) rCrDrErFrGZnum_required_paramsrMerr_msgrQr )rDrCrErPr!construct_train_funcs(rVcs$eZdZdZiZfddZZS) SingletonzSingleton Abstract Base Class https://stackoverflow.com/questions/33364070/implementing -singleton-as-metaclass-but-for-abstract-classes cs0||jvrtt|j|i||j|<|j|SrH) _instancessuperrW__call__)clsrJrK __class__r r!rZs  zSingleton.__call__)__name__ __module__ __qualname____doc__rXrZ __classcell__r r r\r!rWsrWc@s&eZdZdZdefddZddZdS) ActorWrapperzDWraps an actor to provide same API as using the base class directly.actorcCs ||_dSrH)rd)selfrdr r r!__init__s zActorWrapper.__init__cst|j|fddS)Ncstj|i|SrH)rrremote)rJrKZ actor_methodr r!sz*ActorWrapper.__getattr__..)getattrrd)reitemr rhr! __getattr__s  zActorWrapper.__getattr__N)r^r_r`rar rfrlr r r r!rcs rc)rCF)*abcrRr4loggingr.typingrrrrrrrr r rZ ray.actorr Zray.air._internal.utilr r rZray.exceptionsrZ ray.typesrr getLoggerr^rboolrr"r(intr&r1rBrVABCMetarWrcr r r r!sL,     & +$  F