o bi2@sddlZddlZddlmZddlmZddlmZdZej dddka t Z ej ddid Zead t_d d Zd efddZddZddZddZeddZeddZdefddZddZdd Zd!d"Zd#d$ZdS)%N)contextmanagerwraps) auto_init_ray__ray_client_mode_key__RAY_CLIENT_MODE01FTcCsttdsdt_tjS)zGet's the value of `_client_hook_status_on_thread`. Since `_client_hook_status_on_thread` is a thread-local variable, we may need to add and set the 'status' attribute. statusT)hasattr_client_hook_status_on_threadr r r Q/home/ubuntu/.local/lib/python3.10/site-packages/ray/_private/client_mode_hook.py!_get_client_hook_status_on_threads rvalcCs |t_dSN)r r rr r r_set_client_hook_status's rcCst}dt_|SNF)rr r )outr r r_disable_client_hook,srcCsdadS)zkForce client mode to be enabled. NOTE: This should not be used in tests, use `enable_client_mode`. TNis_client_mode_enabledr r r r_explicitly_enable_client_mode3srcCsdadSrrr r r r_explicitly_disable_client_mode;srccs(t}z dVWt|dSt|wr)rrrr r rdisable_client_hook@s rccs$tz dVWtdStwr)rrr r r renable_client_modeIs rfunccs&ddlmtfdd}|S)zDecorator for whether to use the 'regular' ray version of a function, or the Ray Client version of that function. Args: func: This function. This is set when this function is used as a decorator. rraycs8trjdks trtj|i|S|i|S)Ninit)client_mode_should_convert__name__!is_client_mode_enabled_by_defaultgetattr)argskwargsrrr rwrapper]sz!client_mode_hook..wrapper)ray.util.clientrrrr(r r'rclient_mode_hookRs  r+cCstptotS)z;Determines if functions should be converted to client mode.)rr#rr r r rr!msr!cstfdd}|S)aWraps a function called during client mode for execution as a remote task. Can be used to implement public features of ray client which do not belong in the main ray API (`ray.*`), yet require server-side execution. An example is the creation of placement groups: `ray.util.placement_group.placement_group()`. When called on the client side, this function is wrapped in a task to facilitate interaction with the GCS. csPddlm}ttr!|jdd}|j|i|}||S|i|S)Nrr)num_cpus)r)rrr!remoteget)r%r&rfrefrr rr(s  z!client_mode_wrap..wrapperrr*r r1rclient_mode_wrapws  r2cK^ddlm}t|td}|dus||s ||}t|t|||}|j||fi|S)aXRuns a preregistered ray RemoteFunction through the ray client. The common case for this is to transparently convert that RemoteFunction to a ClientRemoteFunction. This happens in circumstances where the RemoteFunction is declared early, in a library and only then is Ray used in client mode -- necessitating a conversion. rrN) r)rr$RAY_CLIENT_MODE_ATTR_converted_key_exists_convert_functionsetattr_get_converted_remote)func_clsin_args in_kwargsr&rkey client_funcr r rclient_mode_convert_functions     r?cKr3)aNRuns a preregistered actor class on the ray client The common case for this decorator is for instantiating an ActorClass transparently as a ClientActorClass. This happens in circumstances where the ActorClass is declared early, in a library and only then is Ray used in client mode -- necessitating a conversion. rrN) r)rr$r4r5_convert_actorr7r8r9) actor_clsr;r<r&rr= client_actorr r rclient_mode_convert_actors     rC) os threading contextlibr functoolsrray._private.auto_init_hookrr4environr.rr#updateis_init_calledlocalr r rboolrrrrrrcallabler+r!r2r?rCr r r rs4