jSddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z ddlmZmZddlmZmZmZmZmZmZmZmZdd lmZmZmZmZdd lm Z m!Z!m"Z"Gd d eZ#Gd de#Z$Gdde#Z%Gdde#Z&Gdde#Z'Gdde#Z(Gdde#Z)Gdde#Z*Gdde#Z+Gdde#Z,Gdd e#Z-Gd!d"e#Z.dS)#N)abstractmethod)Enum)Path)Any)_conversion_warning_mode_warn_implicit_conversion)_decode_missing_escape_EscapedMissing)NoneType ValueKind_is_interpolation get_type_ofget_value_kindis_literal_annotationis_primitive_containertype_str)Box DictKeyTypeMetadataNode)ReadonlyConfigErrorUnsupportedValueTypeValidationErrorcReZdZUeed<dedzdedeffd ZdefdZdded e e e fdzddfd Z deddfd Z dedefd Zededefd Zde fdZde fdZdede fdZdede fdZdefdZdede eefddfdZde fdZde fdZdeezdzde fdZxZS) ValueNode_valparentNvaluemetadatacddlm}t||||5||ddddS#1swxYwYdS)Nr) read_write)rr) omegaconfr!super__init__ _set_value)selfrrrr! __class__s r/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/omegaconf/nodes.pyr$zValueNode.__init__s(((((( ::: Z   # # OOE " " " # # # # # # # # # # # # # # # # # #sAA Areturnc|jSN)rr&s r(_valuezValueNode._value&s yflagsc|drtdt|}t|tr2t |dt jt jfvr ||_ dS| |}tj dur\t|t|ur>tt|t||d||_ dS)Nreadonly)Cannot set value of read-only config nodeT)strict_interpolation_validation) _get_flagrr isinstancestrrr INTERPOLATIONMANDATORY_MISSINGrvalidate_and_convertrgettyper _get_full_key)r&rr/ converteds r(r%zValueNode._set_value)s >>* % % S%&QRR R&u-- eS ! ! "n 4' ' '   #  ' ' ' DIII11%88I'+--55$u++TNN;;*KKi$2D2DT2J2J"DIIIr.c|jj}t|tr[t||urL|turt|t s0t |jj}td|ddSdSdSNzEValue '$VALUE' of type '$VALUE_TYPE' is incompatible with type hint '') _metadataref_typer5r;r6r r type_hintrr&rrBrCs r(_strict_validate_typezValueNode._strict_validate_type?s>* x & & U 8++__E?)K)K_ !9::I!dXaddd    ++$_r.c|E|rdSt|jj}t d|d|d|ddur|||S||S)z Validates input and converts to canonical form :param value: input value :return: converted value ("100" may be converted to 100 for example) NzIncompatible value 'z' for field of type 'r@convertF) _is_optionalrrArBrr4rE_validate_and_convert_impl)r&r ref_type_strs r(r9zValueNode.validate_and_convertKs =  "" t#DN$;<>) $ $ - -  & &u - - -L22599 9r.cdSr+r&rs r(rIz$ValueNode._validate_and_convert_implas=@Sr.c*t|jSr+)r6rr,s r(__str__zValueNode.__str__ds49~~r.cNt|drt|jndS)Nr __INVALID__)hasattrreprrr,s r(__repr__zValueNode.__repr__gs#")$"7"7JtDI]Jr.othercbt|tr|j|jkS|j|kSr+)r5AnyNoder)r&rUs r(__eq__zValueNode.__eq__js0 eW % % &9 * *9% %r.cH||}|tusJ| Sr+)rXNotImplemented)r&rUxs r(__ne__zValueNode.__ne__ps+ KK  &&&&u r.c*t|jSr+hashrr,s r(__hash__zValueNode.__hash__uDIr.resmemoctj|j||jd<|j|jd<|j|jd<dS)N)rcrAr_parent)copydeepcopyrA__dict__rre)r&rbrcs r(_deepcopy_implzValueNode._deepcopy_implxsC$(M$.t$L$L$L [!#y V#', Yr.c|jjSr+)rAoptionalr,s r(rHzValueNode._is_optionals ~&&r.cDt|Sr+)r r-r,s r(r zValueNode._is_interpolations ///r.keyc|}|'|jjdSt|jjS||jjS)N) _get_parentrArmr6r<)r&rmrs r(r<zValueNode._get_full_keysT!!## >~!)r4>-...''(:;; ;r.r+)__name__ __module__ __qualname__r__annotations__rrr$r-dictr6boolr%rEr9rrIrOrTrXr\intr`rirHr rr< __classcell__r's@r(rrsJ III#sTz#########"""DdOd,B"d"""", 3 4    :#:#::::,@@@@@^@K#KKKK&C&D&&&& CD #/#/T#s(^/////'d''''040000<s!2T!9>     ::5AA E""A&I!*III  r.rccNt}||||Sr+)rWrir&rcrbs r( __deepcopy__zAnyNode.__deepcopy__s&ii C&&& r.NNNN rqrrrsrrrur6rvr$rIrwrrxrys@r(rWrWs!(,       d  CI%        c3hIr.rWc eZdZ d dedededzdeeefdzffd ZdeddfdZ d ee efddfd Z xZ S) NoneNodeNrrmrr/c t||ttt|d|dSr|)r#r$rr rs r(r$zNoneNode.__init__sW !$   r.r)c td)NzNValue '$VALUE' of type '$VALUE_TYPE' is incompatible with type hint 'NoneType')rrMs r(rIz#NoneNode._validate_and_convert_impls \   r.rccNt}||||Sr+)rrirs r(rzNoneNode.__deepcopy__&jj C&&& r.rrrys@r(rrs!(,     d  CI%       &      c3hJr.rc eZdZ d dedededzdedeeefdzf fd Zded efd Z d ee efd dfd Z xZ S) StringNodeNTrrmr is_optionalr/c t||t||tt|dSNrmrkrBr~r/r)r#r$rr6r&rrmrrr/r's r(r$zStringNode.__init__W $   r.r)cddlm}||s$t|st |t rt dt |tr|nt|S)Nr) OmegaConfz0Cannot convert '$VALUE_TYPE' to string: '$VALUE') r"r is_configrr5bytesrr r6)r&rrs r(rIz%StringNode._validate_and_convert_impls''''''    & & V%e,, V%'' V ""TUU U"5/::JuuE Jr.rccNt}||||Sr+)rrirs r(rzStringNode.__deepcopy__s&ll C&&& r.NNNTN rqrrrsrrrvrur6r$rIrwrrxrys@r(rrs! (,     d    CI%       ( K K K K K Kc3hLr.rc eZdZ ddedededzdedeeefdzf fd Zded dfd Z ded e fd Z d ee efd dfd Z xZS)PathNodeNTrrmrrr/c t||t||tt|dSr)r#r$rrrs r(r$zPathNode.__init__W $    r.r)cNt|tstddS)NzIValue '$VALUE' of type '$VALUE_TYPE' is not an instance of 'pathlib.Path')r5rrrMs r(rEzPathNode._strict_validate_types3%&& ![   r.cvt|ttfstdt|S)NzCValue '$VALUE' of type '$VALUE_TYPE' could not be converted to Path)r5r6rrrMs r(rIz#PathNode._validate_and_convert_impls;%#t-- !U E{{r.rccNt}||||Sr+)rrirs r(rzPathNode.__deepcopy__rr.r)rqrrrsrrrvrur6r$rErrIrwrrxrys@r(rrs! (,     d    CI%       (34 c3hJr.rc eZdZ d dedededzdedeeefdzf fd Zded e fd Z d ee efd dfd Z xZ S) IntegerNodeNTrrmrrr/c t||t||tt|dSr)r#r$rrwrs r(r$zIntegerNode.__init__&rr.r)c t|ttfvrt|}ntn#t$rt dwxYw|S)NzFValue '$VALUE' of type '$VALUE_TYPE' could not be converted to Integer)r;r6rw ValueErrorr)r&rvals r(rIz&IntegerNode._validate_and_convert_impl:sq E{{sCj((%jj ll"   !X   s ;>ArccNt}||||Sr+)rrirs r(rzIntegerNode.__deepcopy__F&mm C&&& r.r) rqrrrsrrrvrur6r$rwrIrrxrys@r(rr%s! (,     d    CI%       (      c3hMr.rc eZdZ d dedededzdedeeefdzf fd Zded e fd Z d ee efd dfd Z xZ S) BytesNodeNTrrmrrr/c t||t||tt|dSr)r#r$rrrs r(r$zBytesNode.__init__MW $!   r.r)cNt|tstd|S)Nz;Value '$VALUE' of type '$VALUE_TYPE' is not of type 'bytes')r5rrrMs r(rIz$BytesNode._validate_and_convert_implas0%'' !M  r.rccNt}||||Sr+)rrirs r(rzBytesNode.__deepcopy__h&kk C&&& r.r)rqrrrsrrrvrur6r$rrIrwrrxrys@r(rrLs! (,     d    CI%       (c3hKr.rc eZdZ ddedededzdedeeefdzf fd Zded e fd Z d ed efd Z d e fd Z dee efd dfdZxZS) FloatNodeNTrrmrrr/c t||t||tt|dSr)r#r$rfloatrs r(r$zFloatNode.__init__orr.r)c t|tttfvrt|St #t$rt dwxYw)NzDValue '$VALUE' of type '$VALUE_TYPE' could not be converted to Float)r;rr6rwrrrMs r(rIz$FloatNode._validate_and_convert_implse E{{uc3///U||# ll"   !V  s1AAArUc`t|tr|j}n|}|j|dS|j|dS|j|dSt|jtrt j|jnd}t|trt j|nd}|j|kp|o|S)NTF)r5rrrmathisnan)r&rU other_valnan1nan2s r(rXzFloatNode.__eq__s eY ' '  III 9 4 9 !25 9 U]5(249e(D(DOtz$)$$$%(29e(D(DOtz)$$$%yI%8$-48r.c*t|jSr+r^r,s r(r`zFloatNode.__hash__rar.rccNt}||||Sr+)rrirs r(rzFloatNode.__deepcopy__rr.r)rqrrrsrrrvrur6r$rrIrXrwr`rrxrys@r(rrns! (,     d    CI%       (       9C 9D 9 9 9 9#c3hKr.rc eZdZ d dedededzdedeeefdzf fd Zded efd Z d ee efd dfd Z xZ S) BooleanNodeNTrrmrrr/c t||t||tt|dSr)r#r$rrvrs r(r$zBooleanNode.__init__rr.r)ct|tr|St|tr|dkSt|tr |t|S#t $rw}|dvrYd}~dS|dvrYd}~dStdtj d|d}~wwxYwtd)Nr)yesyontrueT)nonofffalseFz5Value '$VALUE' is not a valid bool (type $VALUE_TYPE)) r5rvrwr6rIrlowerrwith_tracebacksysexc_info)r&res r(rIz&BooleanNode._validate_and_convert_impls  eT " " L eS ! ! A:  s # #  ?66s5zzBBB ? ? ?;;==$>>>44444[[]]&AAA 55555)O$nS\^^A%677Q?  ?"G s$ !A++ C,5C'C'-:C''C,rccNt}||||Sr+)rrirs r(rzBooleanNode.__deepcopy__rr.rrrys@r(rrs! (,     d    CI%       (,c3hMr.rceZdZdZ ddeedeezdzdededzde d e ee fdzf fd Z ded dfd Z ded efd Z edeeded efdZde eefd dfdZxZS)EnumNodeaQ NOTE: EnumNode is serialized to yaml as a string ("Color.BLUE"), not as a fully qualified yaml type. this means serialization to YAML of a typed config (with EnumNode) will not retain the type of the Enum when loaded. This is intentional, Please open an issue against OmegaConf if you wish to discuss this decision. NT enum_typerrmrrr/c ft|trt|tst d|di|_||_|jD]\}}|j |j|<t ||t|||||dS)Nz.EnumNode can only operate on Enum subclasses ()rr) r5r; issubclassrrfieldsr __members__itemsrr#r$r) r&rrrmrrr/nameconstantr's r(r$zEnumNode.__init__s)T** *Y2M2M !MMMM ') %.'399;; / /ND( (DK   $"%   r.r)c|jj}t||s,t|jj}t d|ddSr?)rArBr5rrCrrDs r(rEzEnumNode._strict_validate_types[>*%**  !9::I!dXaddd   r.c:||j|S)N)rr)validate_and_convert_to_enumrrMs r(rIz#EnumNode._validate_and_convert_impls004>QV0WWWr.c&t|ttfs"t||std|t||r|S t|tt frt t|tr ||St|trZ|jd}||r|t|d} ||S#t$r||cYSwxYwJ#t tf$ry}d d|j D}td|dtjd|d}~wwxYw) Nz4Value $VALUE ($VALUE_TYPE) is not a valid input for .F, cg|]}|SrLrL.0r[s r( z9EnumNode.validate_and_convert_to_enum..$sGGGQqGGGr.)Invalid value '$VALUE', expected one of []r)r5r6rwrrrvrrq startswithlenKeyErrorjoinrkeysrrr)rrprefixrvalids r(rz%EnumNode.validate_and_convert_to_enums%#s,, Zy5Q5Q !RyRR  eY ' ' L 7%%// !  %%% ( y'''%%% ,%.111##F++1!#f++--0E,$U++,,,$9U+++++, LH% 7 7 7IIGG)*?*D*D*F*FGGGHHE!DEDDDnS\^^A.//Q 7 7s?ADA D!C))D>DDDFA4F  Frcc\t|j}||||S)N)r)rrrirs r(rzEnumNode.__deepcopy__)s/000 C&&& r.r)rqrrrs__doc__r;rr6rrrvrur$rErI staticmethodrrwrrxrys@r(rrsc$(! (,  : czD    d    CI%      :34XXXXXX7T 7374777\7Bc3hJr.rceZdZUeed< ddedeezezdzdededzde de ee fdzf fd Z ded effd Z ded efd Z ded dfd Zededed efdZde eefd dfdZxZS) LiteralNoderBNTrrmrrr/c t|std|d||_t||t |||||dS)Nz4LiteralNode can only operate on Literal annotation (rrr)rrrBr#r$r)r&rBrrmrrr/r's r(r$zLiteralNode.__init__2s%X.. !RxRRR !  $!$   r.r)c|?td|jjDr||j|St |S)Nc3K|]}|duV dSr+rL)rfields r( z3LiteralNode.validate_and_convert..Ms& S S5$ S S S S S Sr.rBr)anyrB__args__validate_and_convert_to_literalr#r9)r&rr's r(r9z LiteralNode.validate_and_convertLsf =S S SDM||j|dSrrrMs r(rEz!LiteralNode._strict_validate_typeVs# ,,dm5,QQQQQr.c>t|j}|D]U}t|t|us+t|tr t|t ur ||kr|cSVdd|D}td|d)Nrc,g|]}t|SrL)rSrs r(rz?LiteralNode.validate_and_convert_to_literal..cs333q477333r.rr)listrr;r5r r6rr)rBrrrrs r(rz+LiteralNode.validate_and_convert_to_literalYsh'((  EU tE{{**uo66+;?;;#;M;M5..  33F33344R%RRRSSSr.rcc\t|j}||||S)N)rB)rrBrirs r(rzLiteralNode.__deepcopy__fs/4=111 C&&& r.r)rqrrrsrrtrr6rwrrvrur$r9rIrErrrrxrys@r(rr/sMMM *.! (,   czC$&   d    CI%      43#3#333333YYYYYYR3R4RRRR T# Tc Tc T T T\ Tc3hMr.rc eZdZdZ ddedededzdeeefdzffd Z ddedeeefdzddfd Z dedefd Z d ee efddfd Z defd ZxZS)InterpolationResultNodez@ Special node type, used to wrap interpolation results. Nrrmrr/c t||ttd|d||d|vr|dddSdS)NTr}rr1)r#r$rr _set_flagrs r(r$z InterpolationResultNode.__init__qs| $C$e     =Je33 NN:t , , , , ,43r.r)c|drtd|||_dS)Nr1r2)r4rr9r)r&rr/s r(r%z"InterpolationResultNode._set_values? >>* % % S%&QRR R--e44 r.c|Sr+rLrMs r(rIz2InterpolationResultNode._validate_and_convert_impls r.rcctr+)NotImplementedError)r&rcs r(rz$InterpolationResultNode.__deepcopy__s!!r.cdS)NFrLr,s r(r z)InterpolationResultNode._is_interpolationsur.)NNNr+)rqrrrsrrrrur6rvr$r%rIrwrr rxrys@r(r r ls1!(, ----d - CI% ------$555DdOd,B5d5555 "c3h"4M""""4r.r )/rfrrabcrenumrpathlibrtypingromegaconf._conversion_warningsrromegaconf._missingr r omegaconf._utilsr r r rrrrromegaconf.baserrrromegaconf.errorsrrrrrWrrrrrrrrrr rLr.r(rso GFFFFFFF                    <;;;;;;;;;;;WWWWWWWWWWr<r<r<r<r<r<r<r########L&&&&&y&&&R$$$$$)$$$N D55555 555p.....)...bUUUUUyUUUp:::::):::z&&&&&i&&&&&r.