&`i$ddlZddlZddlmZddlmZddlmZmZm Z m Z ddl Z ddl m ZddlmZmZmZmZddlmZddlmZdZd Zd Zd Zd Zd ZdZdZeeeeeeeegZ ej!e"Z#dZ$edZ%ede&fdZ'de&de(fdZ)ede e&ee fde(fdZ*ed.de&de ee fde(fdZ+dZ,ede&d efdZ-dZ.ede&defd Z/d!Z0ede&de(fd"Z1ede&defd#Z2d$Z3d%Z4d&e&d'e&d(e&fd)Z5Gd*d+Z6e6Z7e j8j9j:;e7j<Gd,d-Z=dS)/N)partial) FunctionType)CallableOptionalTypeUnion)_internal_kv_del_internal_kv_get_internal_kv_initialized_internal_kv_put TuneError) DeveloperAPItrainable_class env_creator rllib_modelrllib_preprocessorrllib_action_dist rllib_inputrllib_connector__test__cBtt|SN)_global_registrycontainsTRAINABLE_CLASStrainable_names e/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/registry.py_has_trainabler (s  $ $_n E EEc`t|tt|Sr)validate_trainablergetrrs rget_trainable_clsr%,s&~&&&    @ @@r!rcjt|s!t|std|dSdS)NzUnknown trainable: )r _has_rllib_trainablerrs rr#r#2sS . ) )@2F~2V2V@>n>>???@@@@r!returnct ddlm}n#ttf$rYdSwxYw|t |S)Nr) _register_allF) ray.rllibr* ImportErrorModuleNotFoundErrorr )rr*s rr'r'8s_+++++++ , -uuMOOO . ) ))s   trainablecBddlm}t|trt |}t|t ot ||}|pNt|t  o8t|tp#t|tpt|S)zCheck if a given trainable is a function trainable. Either the trainable has been wrapped as a FunctionTrainable class already, or it's still a FunctionType/partial/callable.r)FunctionTrainable) ray.tune.trainabler0 isinstancestrr%type issubclassrrcallable)r.r0is_wrapped_funcs ris_function_trainabler8Cs 544444)S!!1%i00  D11j$77O   y$ ' '' y, / / #)W-- # "" r!Tnamewarncddlm}m}t|trt dnt|tst|tr&t d||}n4t|r%t d||}t||std|tt||dS)aRegister a trainable function or class. This enables a class or function to be accessed on every Ray process in the cluster. Args: name: Name to register. trainable: Function or tune.Trainable class. Functions must take (config, status_reporter) as arguments and will be automatically converted into a class during registration. r) Trainable wrap_functionzDetected class for trainable.z Detected function for trainable.z=Detected unknown callable for trainable. Converting to class.z0Second argument must be convertable to TrainableN)r1r<r=r2r4loggerdebugrrr6infor5 TypeErrorrregisterr)r9r.r:r<r=s rregister_trainablerCZs<;;;;;;;)T""- 45555 I| , ,- 9g0N0N- 7888!M),, )  - STTT!M),, i + +WJIVVVotY?????r!cDttdSr)runregister_allrr!r_unregister_trainablesrGxs##O44444r!ct|std|tt||dS)zRegister a custom environment for use with RLlib. This enables the environment to be accessed on every Ray process in the cluster. Args: name: Name to register. env_creator: Callable that creates an env. !Second argument must be callable.N)r6rArrB ENV_CREATOR)r9rs r register_envrK|sD K J;[IIIk4=====r!cDttdSr)rrErJrFr!r_unregister_envsrM##K00000r! input_creatorct|std|tt||dS)zRegister a custom input api for RLlib. Args: name: Name to register. input_creator: Callable that creates an input reader. rIN)r6rArrB RLLIB_INPUT)r9rOs rregister_inputrRsD M " "L;]KKKk4?????r!cDttdSr)rrErQrFr!r_unregister_inputsrTrNr!cBtt|Sr)rrrQr9s rregistry_contains_inputrWs  $ $[$ 7 77r!cBtt|Sr)rr$rQrVs rregistry_get_inputrYs    T 2 22r!cZtttdSr)rTrMrGrFr!r_unregister_allr[s,r!cHtt||dSr)rrBTEST)keyvalues r_check_serializabilityr`s dC/////r!prefixcategoryr^cd|dzdz|dzdz|dzS)zGenerate a binary key for the given category and key. Args: prefix: Prefix category: The category of the item key: The unique identifier for the item Returns: The key to use for storing a the value. s TuneRegistry:ascii:/)encode)rarbr^s r _make_keyrhs^  --  !   //' " " #    **W    r!c|eZdZd deefdZedZdZdZ dZ d deefd Z d Z d Z d ZdS) _RegistryNracVi|_||_t|_d|_dS)z2If no prefix is given, use runtime context job ID.FN) _to_flush_prefixset _registered_atexit_handler_registered)selfras r__init__z_Registry.__init__s* 55*/'''r!cr|js*tj|_|jSr)rmrayget_runtime_context get_job_idrqs rraz_Registry.prefixs1| B244??AADL|r!c|jrdStjjjjtjkrdStjtd|_dS)NT) rprt_privateworker global_workermode SCRIPT_MODEatexitrBr[rws r_register_atexitz_Registry._register_atexitsM  *  F <  , 1S_ D D F(((*.'''r!c|tvr*ddlm}|d|tt j||j||f<tr|dSdS)a#Registers the value with the global registry. Args: category: The category to register under. key: The key to register under. value: The value to register. Raises: PicklingError: If unable to pickle to provided file. rr z Unknown category {} not among {}N) KNOWN_CATEGORIESray.tunerformatpickle dumps_debugrlr flush_values)rqrbr^r_rs rrBz_Registry.registers + + + * * * * * *)299(DTUU +1*r~logging functoolsrtypesrtypingrrrrrtray.cloudpickle cloudpicklerray.experimental.internal_kvr r r r ray.tune.errorrray.util.annotationsrrrJ RLLIB_MODELRLLIB_PREPROCESSORRLLIB_ACTION_DISTrQRLLIB_CONNECTORr]r getLoggerrr>r r%r3r#boolr'r8rCrGrKrMrRrTrWrYr[r`rhrjrryrz_post_init_hooksappendrrrFr!rrs 222222222222  %$$$$$------#  )' #   8 $ $FFFAAA @s@@@@ ******U3$+>%?D,@@S@U8T>-B@$@@@@:555 >s > > > > > 111 @ @X @ @ @ @1118#8$88883S3X3333 000cSs*WWWWWWWWt9;; $++,<,IJJJr!