&`im" ddlZddlmZddlmZmZmZddlmZm Z m Z ddl m Z ddl mZdZdZd ZGd d ZGd d ZdZe ddededededef dZdededefdZededefdZdS)N) defaultdict)AnyDictTuple) CategoricalDomainFunction) assign_value) DeveloperAPIc*ttSN)rlistm/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/impl/placeholder.pycreate_resolvers_mapr s t  rctjt|ddt S)z>Compute a hash for the specific placeholder based on its path.zutf-8N)hashlibsha1strencode hexdigestID_HASH_LENGTH) path_tuples r_id_hashrs; <J..w77 8 8 B B D D_n_ UUrc4eZdZdZdZdZdefdZdefdZ dS) _FunctionResolverz*Replaced value for function typed objects.__fn_phc"||_||_dSr)hash_fn)selfr!fns r__init__z_FunctionResolver.__init__s rconfigc8|j|S)zSome functions take a resolved spec dict as input. Note: Function placeholders are independently sampled during resolution. Therefore their random states are not restored. )r&)r"sample)r#r&s rresolvez_FunctionResolver.resolves xf---rreturnc|j|jfSrTOKENr!r#s rget_placeholderz!_FunctionResolver.get_placeholder& DI&&rN) __name__ __module__ __qualname____doc__r-r%rr)rr/rrrrrsd44 E.d....'''''''rrc.eZdZdZdZdZdZdefdZdS) _RefResolverz3Replaced value for all other non-primitive objects.__ref_phc"||_||_dSr)r!_value)r#r!values rr%z_RefResolver.__init__/s  rc|jSr)r9r.s rr)z_RefResolver.resolve3s {rr*c|j|jfSrr,r.s rr/z_RefResolver.get_placeholder6r0rN) r1r2r3r4r-r%r)rr/rrrr6r6*sX== E'''''''rr6cZt|ttttfp|duS)zeReturns True if x is a primitive type. Primitive types are int, float, str, bool, and None. N) isinstanceintfloatrbool)xs r _is_primitiverC:s& a#uc40 1 1 >Q$Y>rrr& resolvers id_prefix path_prefixr*c.t|tr?d|vr;t|dkr(fdt|dD|d<|St|tr"fd|DSt|t rfdt|DSt|t r*t fdt|DSt|r|St|tr)fdt|j D|_ |St|trNt}t||} ||St|tsNt}t!||} ||S|S)aReplaces reference objects contained by a config dict with placeholders. Given a config dict, this function replaces all reference objects contained by this dict with placeholder strings. It recursively expands nested dicts and lists, and properly handles Tune native search objects such as Categorical and Function. This makes sure the config dict only contains primitive typed values, which can then be handled by different search algorithms. A few details about id_prefix and path_prefix. Consider the following config, where "param1" is a simple grid search of 3 tuples. config = { "param1": tune.grid_search([ (Cat, None, None), (None, Dog, None), (None, None, Fish), ]), } We will replace the 3 objects contained with placeholders. And after trial expansion, the config may look like this: config = { "param1": (None, (placeholder, hash), None) } Now you need 2 pieces of information to resolve the placeholder. One is the path of ("param1", 1), which tells you that the first element of the tuple under "param1" key is a placeholder that needs to be resolved. The other is the mapping from the placeholder to the actual object. In this case hash -> Dog. id and path prefixes serve exactly this purpose here. The difference between these two is that id_prefix is the location of the value in the pre-injected config tree. So if a value is the second option in a grid_search, it gets an id part of 1. Injected placeholders all get unique id prefixes. path prefix identifies a placeholder in the expanded config tree. So for example, all options of a single grid_search will get the same path prefix. This is how we know which location has a placeholder to be resolved in the post-expansion tree. Args: config: The config dict to replace references in. resolvers: A dict from path to replaced objects. id_prefix: The prefix to prepend to id every single placeholders. path_prefix: The prefix to prepend to every path identifying potential locations of placeholders in an expanded tree. Returns: The config with all references replaced. grid_searchcBg|]\}}t||fzSrinject_placeholders.0ichoicerErFrDs r z'inject_placeholders..~sD! ! ! 6  9t3C[ Q Q ! ! ! rc Li|] \}}|t||fz|fz!SrrK)rNkvrErFrDs r z'inject_placeholders..sN   1 "1iaT1A;RSQUCUVV   rc Jg|]\}}t||fz|fz SrrKrNrOelemrErFrDs rrQz'inject_placeholders..sK   4 iaT1A;RSQUCU V V   rc3RK|]!\}}t||fz|fzV"dSrrKrWs r z&inject_placeholders..sY  4 iaT1A;RSQUCU V V      rcBg|]\}}t||fzSrrKrMs rrQz'inject_placeholders..sD   6  9t3C[ Q Q    r)r>dictlen enumerateitemsrtuplerCr categoriesr rrappendr/rr6)r&rDrErFid_hashrTs ``` rrLrLBsv&$2MV$;$;F q@P@P! ! ! ! ! ! 'vm'<== ! ! ! } FD ! !)            FD ! !$      $V,,     FE " "      $V,,       v   FK ( (      'v'899     FH % %9%% gv . .+%%a(((  """  ' '9%% & ) )+%%a(((  """ rprefixpathc|s||fS|d}t|trb|dtjtjfvr||fS|t |kr*t ||||dfz|ddSnkt|tr||vs(t|tr=|t |kr*t ||||dfz|ddSdS)NrrI)rdre)NN) r>r`rr-r6r]_get_placeholderr\r)r&rdrekeys rrgrgs v~ q'C&%  X !9*0,2DE E E6> ! 3v;;  #s Fd1gZ$7d122h  VT " "Xsf}}64  (5%(3v;;%6%6 s Fd1gZ4GdSTSUSUhWWWW :rreplacedc`fd}|td|tfdS)zReplaces placeholders contained by a config dict with the original values. Args: config: The config to replace placeholders in. replaced: A dict from path to replaced objects. c D]g\}}|sJt|d|s td|\}}|s7|D]-}|j|dkrt ||j|.hdS)NrrrI)r_r>rgr!r r)) resolver_typeargsrerDrdphresolverr&ris r __resolvez'resolve_placeholders..__resolves'~~// F FOD)  9ilM:: )&"d;;JFB % F F=BqE))VV-=X-=t-DEEEE  F F Frr)rmN)r6r)r&rirps`` rresolve_placeholdersrqs_FFFFFF&Il$$$$Ivi000000r)rr)r collectionsrtypingrrrray.tune.search.samplerrr !ray.tune.search.variant_generatorr ray.util.annotationsr rrrrr6rCrLrgrqrrrrws################@@@@@@@@@@::::::------VVV ''''''''* ' ' ' ' ' ' ' ' ??? ll lll l  llll^ S % u    F11 111111r