o bi@sZddlZddlZddlmZddlZeZejdddkZ ddZ dd Z d d Z dS) NwrapsRAY_ENABLE_AUTO_CONNECT0cCs^tr+ts-ttstWddSWddS1s$wYdSdSdSN)enable_auto_connectrayis_initializedauto_init_lockinitr r O/home/ubuntu/.local/lib/python3.10/site-packages/ray/_private/auto_init_hook.py auto_init_ray s  "rcstfdd}|S)Ncst|i|Sr)r)argskwargsfnr rauto_init_wrappersz)wrap_auto_init..auto_init_wrapperr)rrr rrwrap_auto_initsrcCs:|D]}tt|d}|dusJ|tt|t|qdS)z)Wrap public APIs with automatic ray.init.N)getattrr setattrr) api_namesapi_nameapir r rwrap_auto_init_for_all_apiss  r) os threading functoolsrr Lockr environgetrrrrr r r rs