O/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.y)#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)ReadonlyConfigErrorUnsupportedValueTypeValidationErrorc@eZdZUeed<dedzdedeffd ZdefdZdded e e e fdzddfd Z deddfd Z dedefd Zededefd Zde fdZde fdZdede fdZdede fdZdefdZdede eefddfdZde fdZde fdZdeezdzde fdZxZS) ValueNode_valparentNvaluemetadatacddlm}t| ||||5|j |dddy#1swYyxYw)Nr) read_write)rr) omegaconfr!super__init__ _set_value)selfrrrr! __class__s R/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/omegaconf/nodes.pyr$zValueNode.__init__sA( :   # OOE " # # #s <Areturnc|jSN)rr&s r(_valuezValueNode._value&s yyflagsc|jdr tdt|}t|tr5t |dt jt jfvr||_ y|j|}tjdurDt|t|ur.tt|t||jd||_ y)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 >>* %%&QR R&u- eS !n 4'   # #  ' ' ' DI11%8I'++-5$u+TN;*Ki$2D2DT2J"DIr.c|jj}t|trUt||urG|turt|t s.t |jj}td|dyyyNzEValue '$VALUE' of type '$VALUE_TYPE' is incompatible with type hint '') _metadataref_typer5r;r6r r type_hintrr&rrBrCs r(_strict_validate_typezValueNode._strict_validate_type?sw>>** x &U 8+_E?)K !9!9:I!WXaWbbcd *L, 'r.c|B|jryt|jj}t d|d|d|j ddur|j ||S|j|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 =  "#DNN$;$;>) $ -  & &u -L2259 9r.cyr+r&rs r(rIz$ValueNode._validate_and_convert_implas=@r.c,t|jSr+)r6rr,s r(__str__zValueNode.__str__ds499~r.cHt|drt|jSdS)Nr __INVALID__)hasattrreprrr,s r(__repr__zValueNode.__repr__gs")$"7tDIIJ]Jr.othercrt|tr|j|jk(S|j|k(Sr+)r5AnyNoder)r&rUs r(__eq__zValueNode.__eq__js. eW %99 * *99% %r.c>|j|}|tusJ| Sr+)rXNotImplemented)r&rUxs r(__ne__zValueNode.__ne__ps% KK &&&u r.c,t|jSr+hashrr,s r(__hash__zValueNode.__hash__uDIIr.resmemoctj|j||jd<|j|jd<|j |jd<y)N)rcrAr_parent)copydeepcopyrA__dict__rre)r&rbrcs r(_deepcopy_implzValueNode._deepcopy_implxsE$(MM$..t$L [!#yy V#',, Yr.c.|jjSr+)rAoptionalr,s r(rHzValueNode._is_optionals~~&&&r.c4t|jSr+)r r-r,s r(r zValueNode._is_interpolations //r.keyc|j}|6|jjyt|jjS|j |jjS)N) _get_parentrArmr6r<)r&rmrs r(r<zValueNode._get_full_keysX!!# >~~!!)4>>--..''(:(:; ;r.r+)__name__ __module__ __qualname__r__annotations__rrr$r-dictr6boolr%rEr9rrIrOrTrXr\intr`rirHr rr< __classcell__r's@r(rrs< I#sTz####""DdOd,B"d", 3 4 :#:#:,@@@@K#K&C&D& CD #/#/T#s(^//'d'040<s!2T!9>   :5AE"A&!**%HI  r.rcc>t}|j|||Sr+)rWrir&rcrbs r( __deepcopy__zAnyNode.__deepcopy__si C& r.NNNN rqrrrsrrrur6rvr$rIrwrrxrys@r(rWrWs{!(,      d  CI%   c3hIr.rWc zeZdZ d dedededzdeeefdzffd ZdeddfdZ d ee efddfd Z xZ S) NoneNodeNrrmrr/c Vt|||ttt|d|yr|)r#r$rr rs r(r$zNoneNode.__init__s4 !$   r.r)ctd)NzNValue '$VALUE' of type '$VALUE_TYPE' is incompatible with type hint 'NoneType')rrMs r(rIz#NoneNode._validate_and_convert_impls \  r.rcc>t}|j|||Sr+)rrirs r(rzNoneNode.__deepcopy__j C& r.rrrys@r(rrs{!(,    d  CI%  &   c3hJr.rc eZdZ d dedededzdedeeefdzf fd Zdedefd Z d ee efddfd Z xZ S) StringNodeNrrmr is_optionalr/c Vt|||t||tt|yNrmrkrBr~r/r)r#r$rr6r&rrmrrr/r's r(r$zStringNode.__init__4 $   r.r)cddlm}|j|st|st |t r t dt |tr|St|S)Nr) OmegaConfz0Cannot convert '$VALUE_TYPE' to string: '$VALUE') r"r is_configrr5bytesrr r6)r&rrs r(rIz%StringNode._validate_and_convert_implsK'    &%e,%'!"TU U"5/:uJE Jr.rcc>t}|j|||Sr+)rrirs r(rzStringNode.__deepcopy__sl C& r.NNNTN rqrrrsrrrvrur6r$rIrwrrxrys@r(rrs! (,    d    CI%  ( K K Kc3hLr.rc eZdZ d dedededzdedeeefdzf fd Zdeddfd Z dede fd Z d ee efddfd Z xZS)PathNodeNrrmrrr/c Vt|||t||tt|yr)r#r$rrrs r(r$zPathNode.__init__4 $    r.r)c:t|ts tdy)NzIValue '$VALUE' of type '$VALUE_TYPE' is not an instance of 'pathlib.Path')r5rrrMs r(rEzPathNode._strict_validate_types!%&![ 'r.cZt|ttfs tdt|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.rcc>t}|j|||Sr+)rrirs r(rzPathNode.__deepcopy__rr.r)rqrrrsrrrvrur6r$rErrIrwrrxrys@r(rrs! (,    d    CI%  (34 c3hJr.rc eZdZ d dedededzdedeeefdzf fd Zdede fd Z d ee efddfd Z xZ S) IntegerNodeNrrmrrr/c Vt|||t||tt|yr)r#r$rrwrs r(r$zIntegerNode.__init__&rr.r)c t|ttfvr t|}|St#t$r t dwxYw)NzFValue '$VALUE' of type '$VALUE_TYPE' could not be converted to Integer)r;r6rw ValueErrorr)r&rvals r(rIz&IntegerNode._validate_and_convert_impl:sP E{sCj(%j !l" !X  s "0 0Arcc>t}|j|||Sr+)rrirs r(rzIntegerNode.__deepcopy__Fm C& r.r) rqrrrsrrrvrur6r$rwrIrrxrys@r(rr%s! (,    d    CI%  (   c3hMr.rc eZdZ d dedededzdedeeefdzf fd Zdede fd Z d ee efddfd Z xZ S) BytesNodeNrrmrrr/c Vt|||t||tt|yr)r#r$rrrs r(r$zBytesNode.__init__M4 $!   r.r)c<t|ts td|S)Nz;Value '$VALUE' of type '$VALUE_TYPE' is not of type 'bytes')r5rrrMs r(rIz$BytesNode._validate_and_convert_implas#%'!M  r.rcc>t}|j|||Sr+)rrirs r(rzBytesNode.__deepcopy__hk C& r.r)rqrrrsrrrvrur6r$rrIrwrrxrys@r(rrLs! (,    d    CI%  (c3hKr.rc eZdZ ddedededzdedeeefdzf fd Zdede fd Z d edefd Z de fd Z d ee efddfdZxZS) FloatNodeNrrmrrr/c Vt|||t||tt|yr)r#r$rfloatrs r(r$zFloatNode.__init__orr.r)c t|tttfvr t|St #t$r t dwxYw)NzDValue '$VALUE' of type '$VALUE_TYPE' could not be converted to Float)r;rr6rwrrrMs r(rIz$FloatNode._validate_and_convert_implsK E{uc3//U|# l" !V  s &3 3ArUct|tr |j}n|}|j|y|j|y|j|yt|jtrt j |jnd}t|trt j |nd}|j|k(xs|xr|S)NTF)r5rrrmathisnan)r&rU other_valnan1nan2s r(rXzFloatNode.__eq__s eY ' II 99  99 !2 99 U](2499e(Dtzz$))$%(29e(Dtzz)$%yyI%8$-48r.c,t|jSr+r^r,s r(r`zFloatNode.__hash__rar.rcc>t}|j|||Sr+)rrirs r(rzFloatNode.__deepcopy__rr.r)rqrrrsrrrvrur6r$rrIrXrwr`rrxrys@r(rrns! (,    d    CI%  (    9C 9D 9#c3hKr.rc eZdZ d dedededzdedeeefdzf fd Zdedefd Z d ee efddfd Z xZ S) BooleanNodeNrrmrrr/c Vt|||t||tt|yr)r#r$rrvrs r(r$zBooleanNode.__init__rr.r)ct|tr|St|tr|dk7St|tr |j t|Std#t $rc}|j dvrYd}~y|j dvrYd}~ytdjtjd|d}~wwxYw)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 # ?66s5zBB"G  ?;;=$>>[[]&AA )O$nS\\^A%67Q?  ?s#A C 'C>C0CC rcc>t}|j|||Sr+)rrirs r(rzBooleanNode.__deepcopy__rr.rrrys@r(rrs! (,    d    CI%  (,c3hMr.rceZdZdZ ddeedeezdzdededzde de ee fdzf fd Z ded dfd Z ded efd Z edeeded efd Zde 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. N enum_typerrmrrr/c 6t|trt|tst d|di|_||_|jjD]\}}|j|j |< t |-||t|||||y)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**Y2M!@ 1M ') %.'3399; /ND( (DKK  / $"%   r.r)c|jj}t||s.t|jj}t d|dyr?)rArBr5rrCrrDs r(rEzEnumNode._strict_validate_typesP>>**%* !9!9:I!WXaWbbcd +r.c<|j|j|S)N)rr)validate_and_convert_to_enumrrMs r(rIz#EnumNode._validate_and_convert_impls004>>QV0WWr.ct|ttfst||std|t||r|S t|tt frt t|tr||St|tr4|jd}|j|r|t|d} ||SJ#t$r ||cYSwxYw#t tf$rv}dj|jjDcgc]}|ncc}wc}}td|djtj d|d}~wwxYw)Nz4Value $VALUE ($VALUE_TYPE) is not a valid input for ., )Invalid value '$VALUE', expected one of []r)r5r6rwrrrvrrq startswithlenKeyErrorjoinrkeysrrr)rrprefixrr[valids r(rz%EnumNode.validate_and_convert_to_enumsN%#s,Zy5Q!FykR  eY 'L 7%%/  %% ''%%%../q1##F+!#f+-0E,$U++ 5 ,$U++, H% 7II)*?*?*D*D*FGQqGGHE!;E7!DnS\\^A./Q 7 7sO3C4>C3B:8C:C C CCE 'E D AEErccVt|j}|j|||S)N)r)rrrirs r(rzEnumNode.__deepcopy__)s%0 C& r.r)rqrrrs__doc__r;rr6rrrvrur$rErI staticmethodrrwrrxrys@r(rrs$(! (, : czD    d    CI% :34XXX7T 737477Bc3hJr.rceZdZUeed< ddedeezezdzdededzde de ee fdzf fd Z ded effd Z ded efd Z ded dfd Zededed efd Zde eefd dfdZxZS) LiteralNoderBNrrmrrr/c t|std|d||_t|||t |||||y)Nz4LiteralNode can only operate on Literal annotation (rrr)rrrBr#r$r)r&rBrrmrrr/r's r(r$zLiteralNode.__init__2s_%X.!FxjPQR !  $!$   r.r)c|Ctd|jjDr|j|j|St||S)Nc3$K|]}|du ywr+rL).0fields r( z3LiteralNode.validate_and_convert..Ms S5$ SsrBr)anyrB__args__validate_and_convert_to_literalr#r9)r&rr's r(r9z LiteralNode.validate_and_convertLsR =S SDMM|j|j|yrrrMs r(rEz!LiteralNode._strict_validate_typeVs ,,dmm5,Qr.c6t|j}|D]D}t|t|us#t|ts*t|t us<||k(sB|cSdj |Dcgc] }t|c}}td|dcc}w)Nrrr) listrr;r5r r6rrSr)rBrrrr[rs r(rz+LiteralNode.validate_and_convert_to_literalYsh''( EU tE{*uo64;#;M5.    F3q4734 I%PQRSS4s.BrccVt|j}|j|||S)N)rB)rrBrirs r(rzLiteralNode.__deepcopy__fs%4==1 C& r.r)rqrrrsrrtrr6rwrrvrur$r9rIrErrrrxrys@r(rr/sM *.! (,  czC$&   d    CI% 43#3#3YYYR3R4R T# Tc Tc 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|j ddyy)NTr}rr1)r#r$rr _set_flagrs r(r$z InterpolationResultNode.__init__qsQ $C$e   =Je3 NN:t ,4r.r)ch|jdr td|j||_y)Nr1r2)r4rr9r)r&rr/s r(r%z"InterpolationResultNode._set_values, >>* %%&QR R--e4 r.c|Sr+rLrMs r(rIz2InterpolationResultNode._validate_and_convert_impls r.rcctr+)NotImplementedError)r&rcs r(rz$InterpolationResultNode.__deepcopy__s!!r.cy)NFrLr,s r(r z)InterpolationResultNode._is_interpolationsr.)NNNr+)rqrrrsrrrrur6rvr$r%rIrwrr rxrys@r(rrls!(, ---d - CI% -$55DdOd,B5d5 "c3h"4M"4r.r)/rfrrabcrenumrpathlibrtypingromegaconf._conversion_warningsrromegaconf._missingr r omegaconf._utilsr r r rrrrromegaconf.baserrrromegaconf.errorsrrrrrWrrrrrrrrrrrLr.r(rs G   <;WWr<r##L&y&R$)$N D5 5p.).bUyUp:):z&i&r.