Ng~ddlZddlZddlmZmZddlmZddlmZm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZddlmZdd lmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/dd l0m1Z1dd l2m3Z3dd l4m5Z5ee6e7e8e e9e:fZ;eGddZGdde>Z?Gdde?Z@Gdde ZAGdde?ZBdS)N)ABCabstractmethod) defaultdict) dataclassfield)Enum) AnyDictIteratorListOptionalSetTupleTypeUnion)ParserRuleContext) _DEFAULT_MARKER_NoneType ValueKind _get_value_is_interpolation_is_missing_value _is_specialformat_and_raiseget_value_kindis_union_annotationis_valid_value_annotation split_keytype_str) ConfigKeyErrorConfigTypeErrorInterpolationKeyErrorInterpolationResolutionError InterpolationToMissingValueErrorInterpolationValidationErrorMissingMandatoryValueUnsupportedInterpolationTypeValidationError)OmegaConfGrammarParser)parse)GrammarVisitorceZdZUeeeefed<eeeefed<eed<eed<dZe e e efed<dZ eed<e d Ze e efed <dd Zed eeeeffdZdS)Metadataref_type object_typeoptionalkeyNflagsF flags_rootc*ttSN)rdictJ/var/www/html/ai-engine/env/lib/python3.11/site-packages/omegaconf/base.pyzMetadata.As;tCTCTr9)default_factoryresolver_cachereturnc&|j i|_dSdSr6)r3selfs r: __post_init__zMetadata.__post_init__Cs : DJJJ  r9cB|jrt|jS|jS)zN)__name__ __module__ __qualname__rrr __annotations__boolr3r r strr4rr=rBpropertyrDr8r9r:r.r.,sDIsN####tCy#~&&&&NNN HHH (,E8DdO $+++J%*U;T;T%U%U%UNDcNUUU!5cC0!!!X!!!r9r.c4eZdZUdZeed<dZeed<ddZdS)ContainerMetadataNkey_type element_typer>c$|j t|_|jtust|jtsJ|j;t |js'tdt|jdd|j i|_ dSdS)NzUnsupported value type: 'T)include_module_name') r/r rO isinstancetyperPrr)r r3r@s r:rBzContainerMetadata.__post_init__Xs = DM}##z$-'F'F###   (,T->?? %h9J`d0e0e0ehhh : DJJJ  r9rE)rFrGrHrOr rIrPrBr8r9r:rNrNSsJHcL#      r9rNc eZdZUeed<eded<eeeeefed<deddefdZ deee ffd Z d eee fdd fd Z deddd fd Z d7dZdedfdZdedfdZdeeeefdeeeeeefddfdZdedeefdZdedeefdZdedeefdZ d8de de dedeede dd f dZedeeeefdefdZd9d Z d:d"ed#eeededfd$Z d;d"ed#eeededfd%Z dc>tj|j}|d=|SNr[)copyr`rA state_dicts r: __getstate__zNode.__getstate__qs!Yt}-- ~ &r9rfNcN|j|d|jd<dSrc)r`updateres r: __setstate__zNode.__setstate__ws* Z((((, n%%%r9ct|t|tsJ||jd<|dSNrZ)rTrYr`_invalidate_flags_cacherAr\s r: _set_parentzNode._set_parent{s=~FC!8!8~~~#) i  $$&&&&&r9cd|jd<dSrcr_r@s r:rmzNode._invalidate_flags_caches(, n%%%r9cR|jd}|t|tsJ|Srl)r`rTrYrns r: _get_parentzNode._get_parents*y)~FC!8!8~~~ r9 Containerc|jd}|t|tsJt|tr(|jd}|t|tsJ|S|t|tsJ|S)z{ Like _get_parent, but returns the grandparent in the case where `self` is wrapped by a UnionNode. rZ)r`rTrY UnionNoders)rAr\ grandparents r:_get_parent_containerzNode._get_parent_containers y)~FC!8!8~~~ fi ( (  /)4K&*[)*L*L&&& >Z %B%B>>>Mr9r3valuesct|tr|g}|t|tr|g}t|dkrt||z}t|t|krt dt |D]V\}}||}|*|jjJ||jjvr |jj|=9|jjJ||jj|<W||S)Nrz3Inconsistent lengths of input flag names and values) rTrKrJlen ValueError enumeraterXr3rm)rAr3rxidxflagvalues r: _set_flagzNode._set_flags eS ! ! GE >Z55>XF v;;!  ZZ&(F u::V $ $RSS S"5)) 3 3IC3KE}~+7774>///,T2~+777-2$T** $$&&& r9r~c\|jjJ|jj|S)zc :param flag: flag to inspect :return: the state of the flag on this node. )rXr3get)rAr~s r:_get_node_flagzNode._get_node_flags. ~#///~#''---r9c|jd}| ix}|jd<||t}|tur||}|||<|t |t sJ|Src)r`rr_get_flag_no_cacherTrJ)rAr~cacherets r: _get_flagzNode._get_flags| n- =46 6EDM.1ii.// " " "))$//CE$K{jd33{{{ r9c|jj}|J||vr||||S|rdS|}|dS||S)z Returns True if this config node flag is set A flag is set if node.set_flag(True) was called or one if it's parents is flag is set :return: N)rXr3_is_flags_rootrrr)rAr~r3r\s r:rzNode._get_flag_no_caches$   5==U4[4;      4!!## >4##D)) )r9r2rcausemsg type_overridecRt||||t|n|||J)N)noder2rrrr)rrK)rAr2rrrrs r:_format_and_raisezNode._format_and_raisesB !kE s'       r9cdSr6r8rAr2s r: _get_full_keyzNode._get_full_key r9c:|d}|J|S)NT)throw_on_resolution_failure_dereference_node_impl)rArs r:_dereference_nodezNode._dereference_nodes)**t*LL r9Frmemoc0|||S)Nrrr)rArrs r:_maybe_dereference_nodezNode._maybe_dereference_nodes' **(C+   r9c |s|S|}||rtddS|J|}||||t t |||S)Nz8Cannot resolve interpolation for a node without a parent)r\r2r parse_treerr)rrwr$_key&_resolve_interpolation_from_parse_treer+r)rArrr\r2s r:rzNode._dereference_node_impls %%'' K++-- >* 2N4!!!iikk<<Z--..(C =   r9cr|}|t|tsJ|S|t|tsJ|A|}|t|tsJ|A|t|tsJ|Sr6)rrrTrsrY)rAroots r: _get_rootzNode._get_roots"..00 <dI.. . . .KJtS$9$9  ,##%%D# 4(=(=###  ,JtY$?$? r9c t|S)zY Check if the node's value is `???` (does *not* resolve interpolations). )rr@s r: _is_missingzNode._is_missing*s!&&&r9c.|duS)zZ Check if the node's value is `None` (does *not* resolve interpolations). N)_valuer@s r:_is_nonez Node._is_none0s{{}}$$r9othercdSr6r8rArs r:__eq__z Node.__eq__6rr9cdSr6r8rs r:__ne__z Node.__ne__:rr9cdSr6r8r@s r:__hash__z Node.__hash__>rr9cdSr6r8r@s r:rz Node._valueBrr9cdSr6r8)rArr3s r: _set_valuezNode._set_valueFrr9cdSr6r8r@s r: _is_optionalzNode._is_optionalJrr9cdSr6r8r@s r:rzNode._is_interpolationNrr9c|jjSr6rXr2r@s r:rz Node._keyRs ~!!r9c||j_dSr6rrs r:_set_keyz Node._set_keyUs r9c|jjSr6)rXr4r@s r:rzNode._is_flags_rootXs ~((r9r4cj|jj|kr"||j_|dSdSr6)rXr4rm)rAr4s r:_set_flags_rootzNode._set_flags_root[s< > $ 2 2(2DN %  ( ( * * * * * 3 2r9c(|jjtuSr6)rXr/r r@s r: _has_ref_typezNode._has_ref_type`s~&c11r9rENN)r>rW)FNr6)r>rs)0rFrGrHr.rIr r rKrJrar rgrjrormrrrwrr rrrr Exceptionrr DictKeyTypeintrrrrrrrrrrrrrrrrrrrrr8r9r:rWrWfse e_4Xd^ 345555-x-(---- d38n -tCH~-$----'(5/'d'''' ----Xe_ x '< T#Y^$d8D>*HTN:;  :.3.8D>.... c htn    *s*x~****6"!    c]    $ % S0@*A!B s   ^ -2#'  %) s3x   &      $(  %) s3x   &      4    'T'''' %$%%%%  C D   ^  C D   ^  #   ^     ^    HT#t)_,E QU   ^  d   ^  4   ^ "c""""!C!D!!!!)))))+$+4++++ 2t222222r9rWcXeZdZUdZeed<deddeffd ZdefdZ d d Z xZ S) rYz Base class for nodes that can contain other nodes. Concrete subclasses include DictConfig, ListConfig, and UnionNode. _contentr\r]cbt||d|jd<dS)Nr\r]r)superrar`)rAr\r] __class__s r:raz Box.__init__ls2 :::$( j!!!r9r>c*tj|Sr6)rddeepcopyr@s r:__copy__z Box.__copy__ps}T"""r9NcVddlm}ddlm}t ||r|jd}t |t re|jdD]G\}}|||t |tr| FdSdSt ||rv|jd}t |trP|jdD]D}|||t |tr| CdSdSt |trb|jd}t |trB||t |tr| dSdSdSdS)Nr DictConfig ListConfigr) dictconfigr listconfigrrTr`r7itemsrorY _re_parentlistrurW)rArrcontentrritems r:rzBox._re_parentts************ dJ ' ' )mJ/G'4(( +#'=#<#B#B#D#D++KD%())$///!%--+((***  + +++ j ) ) )mJ/G'4(( * M*5**D'((...!$,,*)))  * *** i ( ( )mJ/G'4(( )##D)))gs++)&&((((( ) ) ) )))r9rE) rFrGrH__doc__r rIr r.rarr __classcell__rs@r:rYrYds MMM)x)())))))######))))))))r9rYceZdZUdZeed<e d,dedededed ed e e e e e e ff d Z e d,dedededed ed e e e e e e ff d Zeded d fdZededed d fdZed eefdZeded efdZded edeffdZ d-dededede eed ee de ee e ff dZde ddddedejdede eed e dfdZde ddddededed e df dZde d e d d fd!Zd"ede eed dfd#Zdede d$ed%eed&fd'eed&fd ef d(Z d-de ddede dede eed e e f d)Z! d.de"de de eede ed ef d*Z#d/d+Z$d S)0rsz% Container tagging interface rXTFr2validate_access validate_keythrow_on_missing_valuethrow_on_missing_keyr>cdSr6r8rAr2rrrrs r: _get_childzContainer._get_child  r9cdSr6r8rs r: _get_nodezContainer._get_noderr9NcdSr6r8rs r: __delitem__zContainer.__delitem__rr9rcdSr6r8)rAr2rs r: __setitem__zContainer.__setitem__rr9cdSr6r8r@s r:__iter__zContainer.__iter__rr9 key_or_indexcdSr6r8)rArs r: __getitem__zContainer.__getitem__rr9c4|}|ds||fS|}|dsJ |J|dd}|dsn*|}|td|dN||fS)N.TrzError resolving key 'rS) startswithrrwr!)rAr2origrs r:_resolve_key_and_rootzContainer._resolve_key_and_roots~~c"" >>##S( ((,D>>#&& & & & J'''!""g~~c**1133<()H)H)H)HIII J9 r9throw_on_missingrrc >ddlm}|dkr|d|fSt|}|}tt |dz D]}|n||} ||| ||\} } t | t r| ||} | Yt | tsDd |d|dz} ||dz} td |d | d | d | }|d S|d}|||||\}} |||dfS|7t|}||vrtd| | ||||||}|||n#|||wwxYw|||fS)zA Select a value using dot separated key sequence r) _select_oneN)cr2rthrow_on_type_errorrrrzError trying to access z: node `z.` is not a container and thus cannot contain ``)NNNz Recursive interpolation detected)r\r2rrr) omegaconfrrrangerzrTrWrrsjoinr"idr$add_maybe_resolve_interpolationremove)rAr2rrrrsplitrikr_ parent_key child_keylast_keyrvids r: _select_implzContainer._select_impls_ +***** "99T> !#$(s5zzA~&&  A|aA [!1$? FC #t$$ 110K2 z#y'A'A XXeAAI&677 !!a%L %PcPP:PPCLPPPDD <##9;- ;    q =4' '  U))Cd{{23UVVV HHSMMM !55,G 6E C    C     Xu$$s E==Fr\rWrc |||||}n#t$r|rYdSwxYw||||||S)a Resolve an interpolation. This happens in two steps: 1. The parse tree is visited, which outputs either a `Node` (e.g., for node interpolations "${foo}"), a string (e.g., for string interpolations "hello ${name}", or any other arbitrary value (e.g., or custom interpolations "${foo:bar}"). 2. This output is potentially validated and converted when the node being resolved (`value`) is typed. If an error occurs in one of the above steps, an `InterpolationResolutionError` (or a subclass of it) is raised, *unless* `throw_on_resolution_failure` is set to `False` (in which case the return value is `None`). :param parent: Parent of the node being resolved. :param value: Node being resolved. :param key: The associated key in the parent. :param parse_tree: The parse tree as obtained from `grammar_parser.parse()`. :param throw_on_resolution_failure: If `False`, then exceptions raised during the resolution of the interpolation are silenced, and instead `None` is returned. :return: A `Node` that contains the interpolation result. This may be an existing node in the config (in the case of a node interpolation "${foo}"), or a new node that is created to wrap the interpolated value. It is `None` if and only if `throw_on_resolution_failure` is `False` and an error occurs during resolution. )rrr2rN)r\rr2resolvedr)resolve_parse_treer$*_validate_and_convert_interpolation_result)rAr\rr2rrrrs r:rz0Container._resolve_interpolation_from_parse_tree!sL ..%Es/HH,   * 44  >>(C ?   s  --rc ddlm}m}m}t |t  } t ||rt ||st |} || } n?#t$r2} |r%| || | d|d| tYd} ~ dSd} ~ wwxYwt| t| krd} | }| r||||St |t sJ|S)Nr)AnyNodeInterpolationResultNode ValueNodez#While dereferencing interpolation 'z': )r2rrrrT)rr2r\) nodesrrrrTrWrvalidate_and_convertr)rr&rU) rAr\rr2rrrrr must_wrap res_value conv_valuees r:rz4Container._validate_and_convert_interpolation_resultXsU GFFFFFFFFF#8T222  eY ' ' & 5'0J0J &"8,,I "77 BB "   .**'O%OOAOO&B +ttttt J4 ??22 %  **s6RRR Rh-- - - -OsA'' B#1'BB#rtargetcd|}|+||urtd|}|)dSdS)Nz%Interpolation to parent node detected)r$rr)rArrr\s r:%_validate_not_dereferencing_to_parentz/Container._validate_not_dereferencing_to_parentsT!% 2;''))F      r9 inter_keyc  ||\}}nN#t$rA}td|t jdd}~wwxYw ||dd|\}}}nN#t$rA}td|t jdd}~wwxYw||td|d| || |S) z0A node interpolation is of the form `${foo.bar}`z.ConfigKeyError while resolving interpolation: NT)rrrz5MissingMandatoryValue while resolving interpolation: zInterpolation key 'z ' not found)rr) rr!r#with_tracebacksysexc_infor r'r%r)rArr root_nodeexcr\r rs r:_resolve_node_interpolationz%Container._resolve_node_interpolationsB 0#'#=#=i#H#H Iyy 0 0 0'FFFnS\^^A.// 0 0 0&/&<&<!%,0 '='' #FHee % 0 0 02MMMnS\^^A.// 0 0 >U]'(Ti(T(T(TUU U  6 6D 6 O O OLs, A&*>> r9c t|}|tjkr|Stt |}||||||||n t S)N)r\rr2rrr)rr INTERPOLATIONr+rrset)rAr\r2rrr value_kindrs r:rz&Container._maybe_resolve_interpolationsu$E** 0 0 0L:e,,-- ::!(C)suu ;   r9cdtdtttdtdffd }dtdtt dfd ttdfdt ffd }t ||| } ||S#t$rt$rU}tt|j d | tjd d}~wwxYw)z Resolve a given parse tree into its value. We make no assumption here on the type of the tree's root, so that the return value may be of any type. rrr>rWc2||S)N)rr)r')rrrAs r:node_interpolation_callbackzAContainer.resolve_parse_tree..node_interpolation_callbacks33id3SS Sr9nameargs.args_strc8|||S)N)r2rr(r)r*)r/)r7r8r9r2rrAs r:resolver_interpolation_callbackzEContainer.resolve_parse_tree..resolver_interpolation_callbacks111' 2 r9)r6r;rz' raised while resolving interpolation: r!N)rKr rrrr r,visitr$rrUrFr"r#r$) rArrrr2r6r;visitorr&s ` ` ` r:rzContainer.resolve_parse_treesX T T"*3s8"4 T f  T T T T T T  "38_ 8=c3h          !(C,K     0==,, ,+     0 0 0.99%SScSSnS\^^A.// 0 0s B!!D 5ADD cddlm}ddlm}|jdd|jd<t ||r\|jd}t |t r6|jdD]}|dSdSt ||rH|jd}t |tr(|jdD]}|dSdSdSdS)Nrrrr[r) rrrrr`rTr7rxrmr)rArrrrrs r:rmz!Container._invalidate_flags_caches=************ = ( 4,0DM. )$ ++ 7- 3gt,,8!%z!:!A!A!C!C885577778888D*-- 7- 3gt,,7 $ j 977446666 5 4 7 77777r9)TTFFr6rrE)%rFrGrHrrNrIrr rJrr rWr rrrrr rrrKrrrrr r*ConfigValueContextrrrr'r/rrrrmr8r9r:rsrss!   !%!',%*       !%  #   x~tHTN33 4   ^ !%!',%*       !%  #   x~tHTN33 4   ^  s t   ^  s 3 4   ^  (3-   ^      ^ {C7G1H.$( L%L% L%L%&* L% s3x L% x $hsmXd^C D L%L%L%L%\5 %5 5  5 += 5 &* 5 s3x 5  & 5 5 5 5 n)%)) )  ) &* ) & ))))V*$**QU****$,SX$6 :   #s(O  c3h  >$(   %    &*  s3x   $    4$(! +0+0%+0+0s3x +0 c] +0 +0+0+0+0Z777777r9rsceZdZdZdZdZdS)SCModerr!N)rFrGrHDICT DICT_CONFIG INSTANTIATEr8r9r:rArAs DKKKKr9rAc eZdZUdZeeed<eede fed< dde de de d e d ee d df fd Z d eeeefd e fd Zde d e fdZde d e fdZd efdZd eede ffdZdde deee e fd dfdZ dde deee e fd dfdZd e fdZd e fdZd e fdZd e fdZdeee fd dfdZxZS)rua This class handles Union type hints. The `_content` attribute is either a child node that is compatible with the given Union ref_type, or it is a special value (None or MISSING or interpolation). Much of the logic for e.g. value assignment and type validation is delegated to the child node. As such, UnionNode functions as a "pass-through" node. User apps and downstream libraries should not need to know about UnionNode (assuming they only use OmegaConf's public API). rZNrTrr/ is_optionalr2r\r>c  t|sd|ddz}t|t|ttfst dt |t|d||ddi| |dS#t$r,}td||t|| Yd}~dSd}~wwxYw) Nz"UnionNode got unexpected ref_type z. Please file a bugz3 report at https://github.com/omry/omegaconf/issuesz&Parent type is not omegaconf.ContainerconvertF)r/r0r1r2r3r)rr2rrr) rAssertionErrorrTrsrr"rrar.rrrrK) rArr/rGr2r\rexrs r:razUnionNode.__init__.s% W&x00 *VVVVKL%S)))fy(&;<< P%&NOOO GG  !% $($e,      OOG $ $ $ $ $ W W W $CwCGGSU V V V V V V V V V V WsBB## C-!CCc|}|'|jjdSt|jjS||jjS)Nr)rrrXr2rKr)rAr2r\s r:rzUnionNode._get_full_keyMsT!!## >~!)r4>-...''(:;; ;r9rcL|jd}t|tr||}n@t|tr||}n||}t|tt t fsJ|SNr)r`rTrWrrJrUNotImplemented)rArrrs r:rzUnionNode.__eq__Ws- + gt $ $ (..''CC t $ $ (,,w''CC..''C#d>&:&:;<<<<< r9cR||}|turtS| Sr6)rrO)rArxs r:rzUnionNode.__ne__bs+ KK     ! !u r9c6t|jdSrN)hashr`r@s r:rzUnionNode.__hash__hDM*-...r9ch|jd}t|tttfsJ|SrN)r`rTrWrrK)rArs r:rzUnionNode._valueks0- +'D(C#899999r9rr3c|jd}|jd} |||dS#t$r}||jd<||jd<|d}~wwxYw)NrrX)r`_set_value_implr)rArr3previous_contentprevious_metadatars r:rzUnionNode._set_valuepsy=4 M+6    . . . . .   (8DM* %):DM+ &G s4 AAAcLddlm}|jj}|jj}t |}t |r`t|ttfsJ|4| s tdt|d||j d<dSt|trtdt||jD]-} |||dd||j d<dS#t$rY*wxYwtd t|d) Nr) _node_wrapz/Value '$VALUE' is incompatible with type hint 'rSrz:Cannot assign container '$VALUE' of type '$VALUE_TYPE' to F)rr/rGr2r\zEValue '$VALUE' of type '$VALUE_TYPE' is incompatible with type hint ')omegaconf.omegaconfr[rXr/rDrrrTrKrrr)r r`rs__args__)rArr3r[r/rDcandidate_ref_types r:rWzUnionNode._set_value_implzs 322222>*N, 5!! u   ec8_55 5 5 5}((**)`(S\J]J]```).DM* % % % y ) ) !bXV_M`M`bb '/&7  " 0: #!3$) # 111DM*-EE&H&r\den\o\orrrsC66 DDc*|jdjduS)NrXT)r`r1r@s r:rzUnionNode._is_optionals}[)2d::r9c6t|jdSrN)rr`r@s r:rzUnionNode._is_interpolations z!:;;;r9c6t|jdSrN)rKr`r@s r:__str__zUnionNode.__str__s4=,---r9c6t|jdSrN)reprr`r@s r:__repr__zUnionNode.__repr__rTr9rctt|}|jD]'\}}|dvrt j|||j|<(|jd}t|trR|jd} d|jd<t j||}||jd<||jd<n#||jd<wxYw|}||jd<|jd|jd<|S)N)rrZ)rrrZ) object__new__rUr`rrdrrTrW)rArresr2r src_content old_parent content_copys r: __deepcopy__zUnionNode.__deepcopy__snnT$ZZ((---// D DJC111$(M%d$C$C$C S!mJ/ k4 ( ( '$-i8J =26 $Y/#}[tDDD 36 %i02< $Y//* $Y/<<<<'L#/ Z "&- ": Y s *C C)TNNr6)rFrGrHrr rsrIrrWrKr rJrYrarrrrrrrr rrWrrrbrermrrs@r:rurus  i    D$O$$$$ ! $ WWWW W  W  W WWWWWW><% S0@*A!BB%%%!)$sDy/!:% %%%%N;d;;;;<4<<<<...../#////c3hKr9ru)Crdr#abcrr collectionsr dataclassesrrenumrtypingr r r r r rrrrantlr4r_utilsrrrrrrrrrrrrr errorsr!r"r#r$r%r&r'r(r)"grammar.gen.OmegaConfGrammarParserr*grammar_parserr+grammar_visitorr,rKbytesrfloatrJrr.rNrWrYrsrArur8r9r:r{sw ##############((((((((OOOOOOOOOOOOOOOOOOOOOO$$$$$$                      GFFFFF!!!!!!++++++CT5$67  #!#!#!#!#!#!#! #!L  ${2{2{2{2{23{2{2{2|-)-)-)-)-)$-)-)-)`B7B7B7B7B7B7B7B7J T ccccccccccr9