?i~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?ZBy)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 eeeeffdZy)Metadataref_type object_typeoptionalkeyNflagsF flags_rootc ttSN)rdictc/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/omegaconf/base.pyzMetadata.As ;tCTr9)default_factoryresolver_cachereturnc,|ji|_yyr6)r3selfs r: __post_init__zMetadata.__post_init__Cs :: DJ r9cX|jrt|jS|jS)zN)__name__ __module__ __qualname__rrr __annotations__boolr3r r strr4rr=rBpropertyrDr8r9r:r.r.,sDIsN##tCy#~&&N H (,E8DdO $+J%*;T%UNDcNU!5cC0!!r9r.c2eZdZUdZeed<dZeed<ddZy)ContainerMetadataNkey_type element_typec@|j t|_|jtust|jtsJ|j 9t |j s$tdt|j dd|ji|_ yy)NzUnsupported value type: 'T)include_module_name') r/r rO isinstancetyperPrr)r r3r@s r:rBzContainerMetadata.__post_init__Xs == DM}}#z$--'FFF    (,T->->?%/9J9J`d0e/ffgh :: DJ r9rE)rFrGrHrOr rIrPrBr8r9r:rNrNSsHcL# r9rNc &eZdZUeed<eded<eeeeefed<deddefdZ deee ffd Z d eee fdd fd Z deddd fd Z d6dZdedfdZdedfdZdeeeefdeeeeeefddfdZdedeefdZdedeefdZdedeefdZ d7de de dedeede dd f dZedeeeefdefdZd8d Z d9d!ed"eeededfd#Z d:d!ed"eeededfd$Z d;d%Z!defd&Z"defd'Z#ed(e defd)Z$ed(e defd*Z%edefd+Z&ede fd,Z'ed:de deeeefdd fd-Z(edefd.Z)edefd/Z*de fd0Z+de dd fd1Z,defd2Z-d3edd fd4Z.defd5Z/y )<Node _metadataBox_parent _flags_cacheparentmetadatac^||jd<||jd<d|jd<y)NrXrZr[__dict__)rAr\r]s r:__init__z Node.__init__ls,%- k"#) i (, n%r9r>cJtj|j}|d=|SNr[)copyr`rA state_dicts r: __getstate__zNode.__getstate__qs"YYt}}- ~ &r9rfNcX|jj|d|jd<yrc)r`updateres r: __setstate__zNode.__setstate__ws" Z((, n%r9cj|t|tsJ||jd<|jyNrZ)rTrYr`_invalidate_flags_cacherAr\s r: _set_parentzNode._set_parent{s0~FC!888#) i  $$&r9c"d|jd<yrcr_r@s r:rmzNode._invalidate_flags_caches(, n%r9cL|jd}|t|tsJ|Srl)r`rTrYrns r: _get_parentzNode._get_parents)y)~FC!888 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_containersr y)~FC!888 fi ( //)4K&*[)*L LL >Z %B BBMr9r3valuesct|tr|g}|t|tr|g}t|dk(rt||z}t|t|k7r t dt |D]\}}||}|I|j jJ||j jvs>|j j|=V|j jJ||j j|<|j|S)Nrz3Inconsistent lengths of input flag names and values) rTrKrJlen ValueError enumeraterXr3rm)rAr3rxidxflagvalues r: _set_flagzNode._set_flags eS !GE >Z5XF v;! Z&(F u:V $RS S"5)IC3KE}~~++7774>>///,,T2~~++777-2$$T** $$& r9r~c||jjJ|jjj|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_flags5 ~~##///~~##''--r9c|jd}|ix}|jd<|j|t}|tur|j|}|||<|t |t sJ|Src)r`rr_get_flag_no_cacherTrJ)rAr~cacherets r: _get_flagzNode._get_flagst n- =46 6EDMM.1ii./ " "))$/CE$K{jd333 r9c|jj}|J||vr ||||S|jry|j}|y|j |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_cachesq$$   5=U4[4;     !!# >##D) )r9r2rcausemsg type_overridecBt|||| t|n|||J)N)noder2rrrr)rrK)rAr2rrrrs r:_format_and_raisezNode._format_and_raises/ !kE s'   ur9cyr6r8rAr2s r: _get_full_keyzNode._get_full_key r9c2|jd}|J|S)NT)throw_on_resolution_failure_dereference_node_impl)rArs r:_dereference_nodezNode._dereference_nodes&**t*L r9rmemoc(|j||S)Nrrr)rArrs r:_maybe_dereference_nodezNode._maybe_dereference_nodes" **(C+  r9c |js|S|j}||r tdy|J|j}|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++- >*2N!!!iik<<Z-.(C =  r9c,|j}|t|tsJ|S|t|tsJ|j5|j}|t|tsJ|j5|t|tsJ|Sr6)rrrTrsrY)rAroots r: _get_rootzNode._get_roots"..0 <dI. ..KJtS$999 ,##%D# 4(= == ,JtY$??? r9ct|S)zY Check if the node's value is `???` (does *not* resolve interpolations). )rr@s r: _is_missingzNode._is_missing*s!&&r9c&|jduS)zZ Check if the node's value is `None` (does *not* resolve interpolations). N)_valuer@s r:_is_nonez Node._is_none0s{{}$$r9othercyr6r8rArs r:__eq__z Node.__eq__6rr9cyr6r8rs r:__ne__z Node.__ne__:rr9cyr6r8r@s r:__hash__z Node.__hash__>rr9cyr6r8r@s r:rz Node._valueBrr9cyr6r8)rArr3s r: _set_valuezNode._set_valueFrr9cyr6r8r@s r: _is_optionalzNode._is_optionalJrr9cyr6r8r@s r:rzNode._is_interpolationNrr9c.|jjSr6rXr2r@s r:rz Node._keyRs~~!!!r9c&||j_yr6rrs r:_set_keyz Node._set_keyUs r9c.|jjSr6)rXr4r@s r:rzNode._is_flags_rootXs~~(((r9r4cz|jj|k7r"||j_|jyyr6)rXr4rm)rAr4s r:_set_flags_rootzNode._set_flags_root[s1 >> $ $ 2(2DNN %  ( ( * 3r9c:|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:rWrWfsr e_4Xd^ 3455-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+ 2t2r9rWcReZdZUdZeed<deddeffd ZdefdZ d dZ xZ S) rYz Base class for nodes that can contain other nodes. Concrete subclasses include DictConfig, ListConfig, and UnionNode. _contentr\r]cFt|||d|jd<y)Nr\r]r)superrar`)rAr\r] __class__s r:raz Box.__init__ls# :$( j!r9r>c,tj|Sr6)rddeepcopyr@s r:__copy__z Box.__copy__ps}}T""r9cddlm}ddlm}t ||rz|j d}t |t rZ|j djD]9\}}||j|t |ts*|j;yyt ||ri|j d}t |trI|j dD]6}||j|t |ts'|j8yyt |trS|j d}t |tr3|j|t |tr|jyyyy)Nr DictConfig ListConfigr) dictconfigr listconfigrrTr`r7itemsrorY _re_parentlistrurW)rArrcontentrritems r:rzBox._re_parentts9** dJ 'mmJ/G'4(#'==#<#B#B#DKD%())$/!%-((* $E) j )mmJ/G'4( MM*5D'((.!$,) 6) i (mmJ/G'4(##D)gs+&&(,))r9rE) rFrGrH__doc__r rIr r.rarr __classcell__rs@r:rYrYds7 M)x)()###)r9rYceZdZUdZeed<e d*dededededede e e e e e ff d Z e d*dededededede e e e e e ff d Zededd fd Zeded edd fdZedeefdZededefdZdededeffdZ d+dededede eedee de ee e ff dZde dd ddedej6dede eede dfdZde dd ddededede df dZde de dd fdZd ede eeddfd!Zdede d"ed#eed$fd%eed$fdef d&Z d+de dded e dede eede e f d'Z! d,de"de de eede edef d(Z#d-d)Z$y ).rsz% Container tagging interface rXr2validate_access validate_keythrow_on_missing_valuethrow_on_missing_keyr>cyr6r8rAr2rrrrs r: _get_childzContainer._get_child r9cyr6r8rs r: _get_nodezContainer._get_noderr9Ncyr6r8rs r: __delitem__zContainer.__delitem__rr9rcyr6r8)rAr2rs r: __setitem__zContainer.__setitem__rr9cyr6r8r@s r:__iter__zContainer.__iter__rr9 key_or_indexcyr6r8)rArs r: __getitem__zContainer.__getitem__rr9c|}|jds|j|fS|}|jdsJ |J|dd}|jds ||fS|j}|td|dA)N.rzError resolving key 'rS) startswithrrwr!)rAr2origrs r:_resolve_key_and_rootzContainer._resolve_key_and_roots~~c">>#S( ((,D>>#& &&'''!"g~~c* 9  113<(+@a)HIIr9throw_on_missingrrc ddlm}|dk(r|d|fSt|}|}tt |dz D]}|n||} ||| ||\} } t | t r| j||} | Dt | ts4dj|d|dz} ||dz} td |d | d | d | }|y |d}|||||\}} |||dfS|+t|}||vr td|j| |j|||||}||j |||fS#||jwwxYw)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 + "9T> !#$(s5zA~&A|aA !1$? FC #t$110K2 z#y'A XXeAA&67 !!a%L %-cU(:,GCCL+QPD1'4 <#9- ;  q =4' '  U)Cd{23UVV HHSM !55,G 6E C Xu$$  C  s D88Er\rWrc~ |j||||}|j|||||S#t$r|rYywxYw)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!siL ..%Es/H>>(C ?  , * s - <<rc ddlm}m}m}t |t  } t ||rDt ||s8t |} |j| } t| t| k7rd} | }| r ||||St |t sJ|S#t$r,} |r |j|| | d|d| tYd} ~ yd} ~ wwxYw)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_resultXs GF#8T22  eY ' 5'0J"8,I "77 B J4 ?2 % *s6R Rh- --O-# .**'A%A3O&B + sB C"B==CrtargetcR|}|#||ur td|j}|"yy)Nz%Interpolation to parent node detected)r$rr)rArrr\s r:%_validate_not_dereferencing_to_parentz/Container._validate_not_dereferencing_to_parents=!% 2;'')F  r9 inter_keyc |j|\}} |j |dd|\}}}||td|d|j|| |S#t$r7}td|jt j dd}~wwxYw#t$r7}td|jt j dd}~wwxYw) 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_interpolations 0#'#=#=i#H Iy 0&/&<&<!%,0 '=' #FHe >U]'*=i[ (TU U  6 6D 6 OL+ 0'@FnS\\^A./ 0 0% 02GuMnS\\^A./ 0 0s.AB B!2BB C$2CC inter_type inter_args.inter_args_strcddlm}|j|}||j}||||||St d|)Nr) OmegaConfzUnsupported interpolation type )rr, _get_resolverrr() rAr2rr(r)r*r,resolverr%s r:_evaluate_custom_resolverz#Container._evaluate_custom_resolvers^ (**:6  (I  /1*> r9c t|}|tjk7r|Stt |}|j |||||||St S)N)r\rr2rrr)rr INTERPOLATIONr+rrset)rAr\r2rrr value_kindrs r:rz&Container._maybe_resolve_interpolationsu$E* 00 0L:e,- ::!(C) ;  03u ;  r9cdtdtttdtdffd }dtdtt dfd ttdfdt ffd }t ||| } |j|S#t$rt$rL}tt|jd |jtjd 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>rWc*j||S)N)rr)r')rrrAs r:node_interpolation_callbackzAContainer.resolve_parse_tree..node_interpolation_callbacks33id3S Sr9nameargs.args_strc0j|||S)N)r2rr(r)r*)r/)r7r8r9r2rrAs r:resolver_interpolation_callbackzEContainer.resolve_parse_tree..resolver_interpolation_callbacks,11' 2 r9)r6r;rz' raised while resolving interpolation: r!N)rKr rrrr r,visitr$rrUrFr"r#r$) rArrrr2r6r;visitorr&s ` ` ` r:rzContainer.resolve_parse_trees T T"*3s8"4 T f  T  "38_ 8=c3h   !(C,K   0==, ,+   0.9%%&&McUSnS\\^A./ 0 0s.A??CACCcddlm}ddlm}|jdd|jd<t ||rS|jd}t |t r3|jdjD]}|jyyt ||rE|jd}t |tr%|jdD]}|jyyyy)Nrrrr[r) rrrrr`rTr7rxrmr)rArrrrrs r:rmz!Container._invalidate_flags_caches** == ( 4,0DMM. )$ +-- 3gt,!%z!:!A!A!C557"D-D*--- 3gt, $ j 9446!:-. 5r9)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% s3x L% x $hsmXd^C D L%\5 %5 5  5 +== 5 &* 5 s3x 5  & 5 n)%)) )  ) &* ) & )V*$**QU*$,SX$6 :   #s(O  c3h  >$(  %    &*  s3x   $ 4$(! +0%+0+0s3x +0 c] +0 +0Z7r9rsceZdZdZdZdZy)SCModerr!N)rFrGrHDICT DICT_CONFIG INSTANTIATEr8r9r:rArAs DKKr9rAc neZdZUdZeeed<eede fed< dde de de de d ee d df fd Z deeeefd e fd Zd e d e fdZd e 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). rZNrrr/ is_optionalr2r\r>c J t|sd|ddz}t|t|ttfs t dt ||t|d||ddi|j|y#t$r#}td||t|| Yd}~yd}~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&x08 BUVKL%S))fy(&;<%&NOO G !% $($e,    OOG $ W $CwCGSU V V WsA2A66 B"?BB"c|j}|6|jjyt|jjS|j |jjS)Nr)rrrXr2rKr)rAr2r\s r:rzUnionNode._get_full_keyMsX!!# >~~!!)4>>--..''(:(:; ;r9rc|jd}t|tr|j|}n3t|tr|j|}n|j|}t|tt t fsJ|SNr)r`rTrWrrJrUNotImplemented)rArrrs r:rzUnionNode.__eq__Wsm-- + gt $..'C t $,,w'C..'C#d>&:;<<< r9cF|j|}|turtS| Sr6)rrO)rArxs r:rzUnionNode.__ne__bs% KK   ! !u r9c2t|jdSrN)hashr`r@s r:rzUnionNode.__hash__hDMM*-..r9c^|jd}t|tttfsJ|SrN)r`rTrWrrK)rArs r:rzUnionNode._valueks,-- +'D(C#8999r9rr3c|jd}|jd} |j||y#t$r%}||jd<||jd<|d}~wwxYw)NrrX)r`_set_value_implr)rArr3previous_contentprevious_metadatars r:rzUnionNode._set_valuepsf==4 MM+6    . (8DMM* %):DMM+ &G s3 A! AA!c,ddlm}|jj}|jj}t |}t |rRt|ttfsJ|(|jstdt|d||jd<yt|trtdt||jD]} |||dd||jd<ytd t|d#t$rYDwxYw) 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, 3>>**NN,, 5! u ec8_5 55}((*)I(S\J]I^^_`).DMM* % y )!LXV_M`Lab '/&7&7" 0:#!3$) # 1DMM*-'8&[\den\o[ppqr'sD DDc8|jdjduS)NrXT)r`r1r@s r:rzUnionNode._is_optionals}}[)22d::r9c2t|jdSrN)rr`r@s r:rzUnionNode._is_interpolations z!:;;r9c2t|jdSrN)rKr`r@s r:__str__zUnionNode.__str__s4==,--r9c2t|jdSrN)reprr`r@s r:__repr__zUnionNode.__repr__rTr9rcBtjt|}|jj D].\}}|dvs t j |||j|<0|jd}t|trU|jd} d|jd<t j ||}||jd<||jd<n|}||jd<|jd|jd<|S#||jd<wxYw)N)rrZ)rrrZ) object__new__rUr`rrdrrTrW)rArresr2r src_content old_parent content_copys r: __deepcopy__zUnionNode.__deepcopy__snnT$Z(----/JC11$(MM%d$C S!0mmJ/ k4 ($--i8J =26 $$Y/#}}[tD 36 %%i02< $$Y/'L#/ Z "&-- ": Y 3= $$Y/s 5D D)TNNr6)rFrGrHrr rsrIrrWrKr rJrYrarrrrrrrr rrWrrrbrermrrs@r:rurus i  D$O$$ ! $ WWW W  W  W W><% 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{s ##(OOO$   G!+CT5$67  #!#! #!L  ${23{2|-)$-)`B7B7J T ccr9