O/jwfddlZddlZddlmZmZmZmZmZmZddl m Z ddl m Z m Z e rddlmZddlmZmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(dd l)m*Z*m+Z+m,Z,m-Z-m.Z.dd l/m0Z0dd l1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9dd l:m;Z;mGdde0ee e fZ?y)N) ItemsViewIterableIteratorKeysViewMutableMappingSequence)Enum) TYPE_CHECKINGAny)Self)_conversion_warnings_warn_on_conversion)_DEFAULT_MARKER_ ValueKind _get_value_is_interpolation_is_missing_literal_is_missing_value_is_none_resolve_optional_valid_dict_key_annotation_typeformat_and_raiseget_structured_config_data(get_structured_config_init_field_aliases get_type_ofget_value_kindis_container_annotationis_dictis_intis_primitive_dictis_structured_configis_structured_config_frozentype_hint_contains_none_literaltype_str)Box ContainerContainerMetadata DictKeyTypeNode) BaseContainer)ConfigAttributeErrorConfigKeyErrorConfigTypeErrorInterpolationResolutionErrorKeyValidationErrorMissingMandatoryValueReadonlyConfigErrorValidationError)EnumNode ValueNodekey availablereturnc tjt||Dcgc] }t|c}dd}|syt|dk(r d|ddSd d j d |DdScc}w) Ng333333?)ncutoffr z. Did you mean: r?z. Did you mean one of: , c32K|]}t|ywN)repr).0ks W/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/omegaconf/dictconfig.py z'_make_key_suggestion..As.F1tAw.Fs)difflibget_close_matchesstrlenjoin)r6r7rDcloses rE_make_key_suggestionrM9sx  % % C9-a3q6-3 E  5zQ!%(Q// $TYY.F.F%F$Gq II .sA0 ceZdZUeed<eeefezdzed<dZ dde e e ddfddde de dzd e e e zd e e e zd e e e zd e d eee fdzddffd Zdeee fddfdZdDdZde fdZdEde de ddfdZde de ddfdZde ddfdZdedzde ddfdZde de de ddfdZde defdZd e de defdZedede ddfdZdede ddfdZdeddfd Zdeefd!Zedede ddfd"Z dede fd#Z!dede fd$Z"deddfd%Z#deddfd&Z$dEded'e de fd(Z% dFded'e d)e de fd*Z& dGded+e d)e d,e d-e dedzf d.Z'e(fded/e de fd0Z)de*efd1Z+de,de fd2Z-de.efd3Z/de0ee ffd4Z1dEded/e de fd5Z2 dHd6e d7e3edzde4e5ee ffd8Z6d9e de fd:Z7d9e de fd;Z8dZ:dEde d eee fdzddfd?Z;exZ?S)I DictConfig _metadataN_contentTcontentz)dict[DictKeyType, Any] | DictConfig | Anyr6parentref_typekey_type element_type is_optionalflagsr8c D t|tr||jj}t ||t |||t|||t|std|t|s t|rPtd5|j||dddt|s t|r|jddyt|trQtj |j} || _|| _|| _|| _|| _| |j,d<tdd 5|j||dddyy#1swYxYw#1swYyxYw#t.$r#} t1d|d| t3|  Yd} ~ yd} ~ wwxYw) N)r6optionalrT object_typerUrVrX)rSmetadataUnsupported key type FrXreadonlyTrP) only_if_unset)noder6valuecausemsg) isinstancerOrPrXsuper__init__r(dictrr0r"r _set_valuer# _set_flagcopydeepcopyr6rTrZrVrU__dict__ ExceptionrrI) selfrRr6rSrTrUrVrWrXr\ex __class__s rErgzDictConfig.__init__Is( T':.=#--33E G *(% $%!-  38<(+@ )KLL#G,0DX0N)%0:OOG5O9:.w7;V<NN:t4gz2#}}W->->?H#&HL(0H%(3H%,8H)(0H%19DMM+.)%tD:OOG5O9::<:: :: T $Ct23r7 S S TsUBE3 E0E3A.E3=E'E3E$ E3'E0,E30E33 F<FFmemoctd}tj|jd||jd<tj|jd||jd<|jd}t |t rti}|j D]^\}}|jd} d|jd<tj||}||jd<|||<||jd<`n|}||jd<|jd|jd<|S#||jd<wxYw)NrP)rr _flags_cacherQ_parent)rOrkrlrmrerhitems) rorrres src_content content_copyrDv old_parentvcs rE __deepcopy__zDictConfig.__deepcopy__~s($(MM$-- 2LSW$X [!'+}} MM. )(  ^$mmJ/ k4 (L#))+ 71ZZ 2 7,0AJJy)qt4B-0BKK *&(LO,6AJJy) 7'L#/ Z "&-- ": Y -7AJJy)s 5:D00Ec,tj|SrA)rkros rErkzDictConfig.copysyyc|jjtdfvxr t|jj SrA)rPr[r rrs rE _is_typedzDictConfig._is_typeds=~~))#t< W NN & &F B  rrbc|j}|jddu}||jdvr|r|jddury|s|r|rH|jj t dfvsJd|d|jj jd}nd|d}|t||jz }|j||t| yyy) NstructTrQFzKey 'z ' not in ''z' is not in structr6rbrc) r _get_flagrm_get_node_flagrPr[rh__name__rMkeys_format_and_raiser,)ror6rbis_typed is_structrds rE _validate_getzDictConfig._validate_gets>>#NN8,4 dmmJ/ /&&x0E99>>55dD\III!#j1K1K1T1T0UUVWC!#&89C+C==&&50DS0I'% 0rcddlm}t|}|tjk(ryt |r|j ||y|tjk(s|y||j|n|}t|txr"|jjttfv}|duxs| }|ryt|tsJ|jj}|j!|} t#| r t#|ryt%|r.t%| s#t'dt)| dt)||1| .t+|d|} t-| | s|j/|| |yyyy)Nr OmegaConfzCannot assign z to __origin__) omegaconfrrr INTERPOLATIONr_validate_non_optionalMANDATORY_MISSING _get_nodererOrPrTr rhr*get_typerrr3r%getattr issubclass_raise_invalid_value) ror6rbrvktargettarget_has_ref_typeis_valid_target target_type value_typeorigins rE _validate_setzDictConfig._validate_setsq' E " (( (  E?  ' 'U 3  ,, , (+$T( J ;''T{: !D.C0C,C  &$'''&&// ''. : 7;#7  "; /8O 9 " *!5 6d8K;P:QR   "z'=[, DFj&1))%[I2(> "rcddlm}|}|}|jd||j|}|j r5t |j jr|j j}|j|}|jr:|j jtdfvr|jdt||jry|duxrA|duxr;t |xr.|j xrt| xr t|| }|r(dt|dt|d|}t!|y)Nrrr6rbz Merge error:  is not a subclass of . value: )rrrrrr"rPrT _is_missingr[rhrrrrr%r3) rorbrdestsrc dest_obj_type src_obj_typevalidation_errorrds rE_validate_mergezDictConfig._validate_mergesU' ##D#.!**40 ==?3DNN4K4KL NN33M ))#.    #--";";D$<"O   4z#  ? ??    % <D( <$]3 <LLN" <L))  < |];;   678' 67yG "#& & rct|ddr|Y|j|}|#t|tsJ|j }n3t |j j\}}n|j }||j|n|}||j jn|j j}|s*t|s|j||tdyyyy)NTF)resolvethrow_on_resolution_failurez!field '$FULL_KEY' is not Optionalr) rrrer* _is_optionalrrPrVrTr$rr3)ror6rbchildfield_is_optional_rrs rErz!DictConfig._validate_non_optionals E4U Ks+$%eT222(-(:(:(<%+<33,(%q%)$5$5$7!,/OT^^C(F>++%%..  %-L.&&)*MN'.$' Lrrrcb|J|Jdt|dt|d|}t|)NzInvalid type assigned: rr)r%r3)rorbrrrds rErzDictConfig._raise_invalid_valuesW%%%&&&%hz&:%;<#K015' C c""rcN|j|jj|SrA)_s_validate_and_normalize_keyrPrU)ror6s rE_validate_and_normalize_keyz&DictConfig._validate_and_normalize_key)s11$..2I2I3OOrcV|tur0tjD]}t||s|cSt d|t ur|dvr t |S|t tttt fvr't||st d|jd|St|tr tj||SJd |#t$rZdj!|j"j%Dcgc]}|ncc}wc}}t d|jd|dwxYw) Nz!Incompatible key type '$KEY_TYPE')rr z+Key $KEY ($KEY_TYPE) is incompatible with ()r?z/Key '$KEY' is incompatible with the enum type 'z ', valid: []r])r r)__args__rer0boolrIbytesintfloatrrr r4validate_and_convert_to_enumr3rK __members__r)rorUr6txvalids rErz(DictConfig._s_validate_and_normalize_key,s6 s? )) c1%J %%HI I  #-9  #uc5$7 7c8,(A(BSBSATTUVJ $ ' <*D(c |j||y#t$r#}|j||t|Yd}~yd}~wt$r}|j|||Yd}~yd}~wwxYw)Nr)r6rb type_overridercr)_DictConfig__set_implAttributeErrorrr-rnror6rbes rE __setitem__zDictConfig.__setitem__Isn B OO5O 1   " "uN! #   B  " "s%q " A A Bs A'= A' A""A'cl|j|}|j||j||yrA)r_validate_non_ambiguous_key_set_item_impl)ror6rbs rE __set_implzDictConfig.__set_implTs0..s3 ((- C'rcf|jd}|jjtust |t syd}t |t rgt|r\t|}|j|}t ||k(rt |trt|jturl|}nit|turXt ||vrK|j|}t |trt|jtur t |}|td|d|y)NrQz%Conflicting integer and string keys: z and )rmrPrUr rerhrIr rgetr*type_keyr0)ror6rRconflicting_key numeric_keyras rErz&DictConfig._validate_non_ambiguous_keyYs-- + >> " "# -Z5N .2 c3 F3Kc(K;;{+DK C'tT*%,"- #Y# #c(g"5;;s#DdD)T$))+->#-E"%c(  &$77J%PSwW  'rc|js|jrgS|jdjSNrQ)rrrmrrs rE__dir__zDictConfig.__dir__ss3    I}}Z(--//rcx |j||y#t$r}|j|||Jd}~wwxYw)zu Allow assigning attributes to DictConfig :param key: :param value: :return: rN)rrnrrs rE __setattr__zDictConfig.__setattr__xs?  OOC '   " "s%q " A 5 s 949c|dk(r t |j|tdS#t$r#}|j |d|t Yd}~yd}~wt $r}|j |d|Yd}~yd}~wwxYw)zf Allow accessing dictionary values as attributes :param key: :return: rF)r6 default_value validate_keyNr6rbrcrr)r _get_implrr-rr,rnror6rs rE __getattr__zDictConfig.__getattr__s *  " " A>>'7e"    " "t1>c9I>J J   " "t1N #   A  " "s$a " @ @ As  A*A A* A%%A*c|jdr|j|dtd |jd|=y#t$r"d}|j|dt |YywxYw)ze Allow deleting dictionary values as attributes :param key: :return: r_N6DictConfig in read-only mode does not support deletionrrQzAttribute not found: '$KEY')rrr2rmKeyErrorr,ror6rds rE __delattr__zDictConfig.__delattr__s} >>* %  " ")L #  Y j)#. Y/C  " "s$>RSV>W " X YsA(A,+A,c |j|}|jdr|j|dtd|jdr|j|dt d|j rP|j ddur=|j|dt t|jjd |jd|=y#t$r"d }|j|dt|YywxYw) Nr_rrrz3DictConfig in struct mode does not support deletionFz' (DictConfig) does not support deletionrQzKey not found: '$KEY') rrrr2r.rrr%rPr[rmrr-rs rE __delitem__zDictConfig.__delitem__s..s3 >>* %  " ")L #  >>( #  " "%I #  >>  3 3H =U J  " "% : :;<Q " R SsC (D  D rc~ |j||S#t$r}|j|d|Yd}~yd}~wwxYw)znReturn the value for `key` if `key` is in the dictionary, else `default_value` (defaulting to `None`).rNr)rr0r)ror6rrs rErzDictConfig.getsE A>>c>G G! A  " "s$a " @ @ As <7<rc |j|d|}t |t sJ|j |||S#ttf$r|tur|cYSwxYw)NT)r6throw_on_missing_keyrr6rbr) _get_childr,r-rrer*_resolve_with_default)ror6rrras rErzDictConfig._get_implsz ??d#D$%%%))4}*  %n5 $44$$  s<AAvalidate_accessthrow_on_missing_valuerc` |j|}|r|j||jdj |}|/|r+d|t ||jz}t||S|r|jr td|S#t$r|r|r|rtYywxYw)NrQz Missing key zMissing mandatory value: $KEY) rr0r,rrmrrMrr-rr1)ror6rrrrrbrds rErzDictConfig._get_nodes 2237C    s #!]]:6::3? =#$SG,/CC/UU$S)) $(9(9(;'(GH H '" <'.. sBB-,B-defaultc |jdr td|jdr td|jr>|j ddur+tt |j jd|j|}|j|d}|+t|tsJ|j|||}||=|S|tur|S|j| }||k7rtd |d |d td |d #t $r}|j#|d|Yd}~yd}~wwxYw)Nr_zCannot pop from read-only noderz.DictConfig in struct mode does not support popFz" (DictConfig) does not support pop)r6rr)r6zKey not found: 'z ' (path: 'z')rr)rr2r.rrr%rPr[rrrer*rr _get_full_keyr-rnr)ror6rrarbfullrs rEpopzDictConfig.popsh A~~j))*JKK~~h'%&VWW~~D$7$7$A$N% : :;<<^_2237C??sE?BD!$---224w3I "22"N--#-6Ds{,.sgZvRH-/?wa-HII A  " "s$a " @ @ As$CD D#8D E$D==Ec|js |js|jrijS|jdj}t |t sJ|Sr)rrrrrmrer)rorets rErzDictConfig.keys<sY    !7!7!9T]]_779 mmJ',,.#x((( rc  |j|} |j|}|t|tsJ|y |j||y#t$rYywxYw#t t f$rd}Y:wxYw#t$rYyt$rYywxYw)z A key is contained in a DictConfig if there is an associated value and it is not a mandatory missing value ('???'). :param key: :return: FNrT) rr0rrer*rrrr/r1)ror6ras rE __contains__zDictConfig.__contains__Cs 2237C ??3'D<:dD#9 99 < **s$*?"   .) D 0 (  s9A%A A7 AA A43A47 B B  B c4t|jSrA)iterrrs rE__iter__zDictConfig.__iter__csDIIK  rcVt|jddjS)NT)rr)rhitems_exrvrs rErvzDictConfig.itemsfs#DMM$TM:;AACCrc\||vr|j|}|S|}|j|||SrA)rr)ror6rrs rE setdefaultzDictConfig.setdefaultis< $;""3'C C   S' * rrrclg}|jr|jddtd|jr t d|j D]V}|r||}n2|j d|}t|tr|j}|||vsD|j||fX|S)Nz4Cannot iterate a DictConfig object representing Nonerz#Cannot iterate a missing DictConfigrQ) rr TypeErrorrr1rrmrer5_valueappend)rorrrvr6rbs rErzDictConfig.items_exqs02 ==?  " " VW #     '(MN N99; +CS  j1#6eY/!LLNE|sd{ c5\* + rotherc,||jdduSt|s t|r%t|ddi}tj ||St |trtj ||S|j r t|StS)NrQ allow_objectsTr^) rmr!r"rO _dict_conf_eqrerrNotImplemented)ror s rE__eq__zDictConfig.__eq__s ===,4 4 U #';E'Bu_d,CDE++D%8 8 eZ (++D%8 8    &u- -rcF|j|}|tur| StSrA)rr)ror rs rE__ne__zDictConfig.__ne__s% KK  N "5Lrtype_or_prototypec|yt|std|ddlm}|j |}|j j }d|j _|j|||j _y)a Retypes a node. This should only be used in rare circumstances, where you want to dynamically change the runtime structured-type of a DictConfig. It will change the type and add the additional fields based on the input class or object Nz"Expected structured config class: rr)r" ValueErrorrr structuredrPr[ merge_with)rorrprotor[s rE_promotezDictConfig._promotesr  $ #$56ABSATUV V'%001BCoo11 &*# %0"rc2|jd}|jd}tj|} |j||y#t$rE||jd<|jj |j||jd<wxYw)NrQrP)rmrk_set_value_implrnupdate)rorbrXprevious_contentprevious_metadatametadata_snapshots rErizDictConfig._set_values==4 MM+6 II&78    . (8DMM* %  & & - -.?.H.H I):DMM+ &   s AABcddlm}m}|i}t|trJ|j d|t |dr!d|jd<d|j_ yt|dr!||jd<d|j_ yt|r!||jd<d|j_ yi|jd<t|rtd5d|j_ |jd }t||| }||d d gd5|j!D]\}}|j#|| dddt%||j_ dddyt|t&rt)j*||j_||d d gd5|jdj!D]\}}|j#|| ddd|jj|j_ yt|t.rV||d d gd5|j!D]\}}|j#|| dddt.|j_ yd |} t1| #1swYJxYw#1swYyxYw#1swYxYw#1swYTxYw)Nr)MISSING flag_overriderTrrQ)strict_interpolation_validationFr )r rSrr_zUnsupported value type: )rr!r"rer5rrrmrPr[rrr"rrrrvrrrOrkrlrXrhr3) rorbrXr!r"aodatarDrzrds rErzDictConfig._set_value_impls|4 =EeY/// t51 E4 ((,DMM* %)-DNN & ud K(-DMM* %)-DNN & u %(/DMM* %)-DNN &(*DMM* %#E*)%0 D15DNN.8B5RD'th -CUK3$(JJL3DAq ,,Q2332=U1CDNN. D DE:.'+}}U';$"4(J)?G/ %z : @ @ B/1((A.//.3__-H-H*E4("4(J)?G/ % /1((A.//.2*18%c**)33 D D// //s<=J+J"J38J)6+J5J JJ&)J25J>d1d2c|jddu}|jddu}|r|ry||k7ryt|tsJt|tsJt|t|k7ry|j s|j r |j |j uS|j dD]2\}}||jdvryt j||||r2yy)NrQTFr#)rmrerOrJrrr+_item_eq)r'r(d1_noned2_nonerDrzs rErzDictConfig._dict_conf_eqs++j)T1++j)T1 w g "j)))"j))) r7c"g  >> r~~/>>#r~~'77 7KKK. DAq J// ))"aQ7   rc 6ddlm}|jj}t |sJt |}i}i}|j D]}t|tsJ|j|}t|tsJ |j}|jr"||vr^|j|dtdt|t r|j#|} n|j%} ||vr | |||<| ||< |di|} |j)D]\}} t+ ||  S#t$r}|j|d|Yd}~d}~wwxYw#t&$r+} |jdd| dt| zYd} ~ d} ~ wwxYw) z Instantiate an instance of `self._metadata.object_type`. This requires `self` to be a structured config. Nested subconfigs are converted by calling `OmegaConf.to_object`. rrNrzJStructured config of type `$OBJECT_TYPE` has missing mandatory value: $KEYz-Could not create instance of `$OBJECT_TYPE`: )r6rbrcrd)rrrPr[r"rrrerIrr*_dereference_noder/rrr1r' to_objectr rrvsetattr) rorr[init_field_aliasesinit_field_itemsnon_init_field_itemsrDrarrzresultexcs rE _to_objectzDictConfig._to_objects (nn00 #K000EkR+-/1 ,Aa% %%??1%DdD) )) C--/!..&&/d'$ *''-KKM&&:; !3A!67*+$Q'5 ,8  4#34F)..0 "DAq FAq ! " E0 C&&1D&BB C0   " "Cc#hN #   s0D: E$: E!EE!$ F-!FF)r8r rA)T)TTFF)TN)@r __module__ __qualname__r(__annotations__rhr)r*rI__hash__r r&rrrgrr}rkrrrrrrrrrrrrrrrrrrrrrrrrrrobjectrrrrrvrrlisttuplerrrrrir staticmethodrr7 __classcell__)rqs@rErOrODs  ;$%+d22H !$'$'(+ (,3T<3T3Td 3T S / 3T S / 3TDIo3T3TCI%3T 3Tjc3hL8 4 SD('J'JS'JT'JR 'S 'T 'D+*<ST: # #&) #8; #  #PsP{P=c== =:B{B3B4BB(k(#($( {t40#0  s 3 4  AsAsA*AA A$YsYtY(S{StSBA{A3A#AJN  /2 BF  (!%!',%*    !%  #  >4D A{ AS A ADh{+4@!(;/!Dyc!12DkC3JN*2;*?$*F eK$% &4 C D CD 1$s)d*:1t1,  DdOd,B d /+S/+c4i41G/+SW/+b,LT.6C6rrO)@rkrGcollections.abcrrrrrrenumr typingr r typing_extensionsr rr_utilsrrrrrrrrrrrrrrrrr r!r"r#r$r%baser&r'r(r)r* basecontainerr+errorsr,r-r.r/r0r1r2r3nodesr4r5rIrMrOr.rrErJs %&K0GF(   'JcJhJ3Jz sCx 8z r