%VjiNdZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZmZmZddlmZmZdd lmZmZdd lmZmZdd lmZdd lmZmZm Z m!Z!m"Z"m#Z#m$Z$d dl%m&Z&m'Z'd dl(m)Z)e r ddl*m+Z+ddl,m-Z-dZ.d+dZ/dZ0Gdde'Z1Gdde'Z2Gdde1Z3Gdde1Z4Gdde1Z5Gdd e5Z6Gd!d"e5Z7Gd#d$e'Z8Gd%d&e8Z9Gd'd(e8Z:Gd)d*e8Z;dS),a Dictionary-related variable tracking classes for PyTorch Dynamo. This module implements variable tracking for different types of dictionary-like objects: - Regular Python dictionaries (dict) - Ordered dictionaries (collections.OrderedDict) - Default dictionaries (collections.defaultdict) - Dictionary views (keys and values) - Sets and frozensets (implemented internally using dictionaries) These classes are responsible for tracking dictionary operations during graph compilation, maintaining proper guards for dictionary mutations and key existence checks. They handle dictionary creation, modification, key/value access, and view operations while ensuring correct behavior in the compiled code through appropriate guard installation. The implementation uses a special _HashableTracker wrapper to handle dictionary keys while preserving proper aliasing semantics. Sets are implemented as dictionaries with None values for efficiency and code reuse. N)Hashable)Optional TYPE_CHECKING)is_fake)graph_break_hints polyfills variables)create_call_functioncreate_instruction)raise_observed_exceptionunimplemented_v2) GuardBuilder install_guard)is_from_local_source)cmp_name_to_op_mapping dict_items dict_keys dict_valuesistyperaise_args_mismatchspecialize_symnode)ValueMutationNewVariableTrackerConstantVariable) PyCodegenInstructionTranslatorcRt|jddS)N__instancecheck__F)type__dict__get)objs W/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/_dynamo/variables/dicts.pywas_instancecheck_overrider(=s" 99  ! !"5u = ==c |ddlm}|j}tt|t dt |gdS)Nrrzunhashable type: args)torch._dynamo.symbolic_convertr current_txr TypeErrorrr#)argtxr s r'raise_unhashabler2Asm zHHHHHH - " - / /2-.M$s)).M.MNNOr)ct|tjr*|s|rdSt|tjr3|jj dduSt|tj rtd|j DSt|tj r0td|jDSt|tjrnt#|jsZt'j|jdt*jur4t|jt*rt|jt.St|tjtjtjtjtjtjtjtjtj j!tj"tj#tj$tj%tj&tj'tj(tj)fS)NT example_valuec34K|]}t|VdSN is_hashable.0es r' zis_hashable..]s(33a;q>>333333r)c34K|]}t|VdSr6r7r9s r'r<zis_hashable.._s(==a;q>>======r)__hash__)* isinstancer LazyVariableTracker is_realizedr8TensorVariableas_proxynodemetar% TupleVariableallitemsFrozenDataClassVariablefieldsvaluesUserDefinedObjectVariabler(valueinspectgetattr_staticintr> py_HashableBuiltinVariableSymNodeVariabler EnumVariableFrozensetVariableUserDefinedClassVariableUserFunctionVariableSkipFunctionVariablemisc NumpyVariableNNModuleVariableUnspecializedNNModuleVariableMethodWrapperVariableTorchInGraphFunctionVariableTypingVariableFunctoolsPartialVariableWeakRefVariable TorchHigherOrderOperatorVariable)xs r'r8r8Ks 1i344  MMOO t!Y-..& zz|| %))/::$FF Ay. / /! 3317333333 Ay8 9 9 ==18??+<+<======1i9:: *1733   "17J 7 73< G G qw $ $ H!';/// ))*&+2..,*7/6(2):#    r)cFeZdZejZdhejZGddZe fde eefddffd Z dZ d Z d Z d Zd Zd ZdefdZdefdZdefdZdZdZd(dZdddefdZdddefdZdefdZdefdZdZdZ d)fd$ Zd%Z d&Z!fd'Z"xZ#S)*ConstDictVariableuser_clsc\eZdZdZd dZedZdZedZ ddde fd Z dS) "ConstDictVariable._HashableTrackeraB Auxiliary opaque internal class that wraps a VariableTracker and makes it hashable This should not be seen or touched by anything outside of ConstDictVariable and its children Note that it's also fine to put VTs into dictionaries and sets, but doing so does not take into account aliasing returnNcnt|}t|st|||_dSr6)rr8r2vt)selfrks r'__init__z+ConstDictVariable._HashableTracker.__init__s8#B''Br?? % $$$DGGGr)cZt|jtjrK|js2|jr|jSt|jtjr+|jj j d}nt|jtj r3tj tfd|jjD}nt|jtjr |jjSt|jtjr |jjSt|jtjr|jSt|jtjr&tj |jjjSt|jtjrmtj fd|jjD}tj|j|St|jtjr |jjS|j}|S)Nr4c38K|]}|jVdSr6underlying_value)r:r;rs r'r<zFConstDictVariable._HashableTracker.underlying_value..s.NN1((1++6NNNNNNr)c8i|]\}}||jSrp)r:kvrs r' zGConstDictVariable._HashableTracker.underlying_value..s7!!!8<1Axx{{3!!!r))r?rkr r@rAr8original_valuerBrCrDrErFre_HashableTrackertuplerHr[rMr\rW get_functionra referent_vtrqrIrJ HashWrapper python_typerLas_python_constant)rlrc fields_valuesrs @r'rqz3ConstDictVariable._HashableTracker.underlying_valuesF47I$ABB 0++-- 0G'')) 0 w--///$'9#;<< 1G$$&&+0ADGY%<== 1,=NNNN NNNNNDGY%?@@ 1w}$DGY%LMM 1w}$DGY%CDD 1w++---DGY%>?? 1,=x 344EEDGY%FGG 1,=!!!!@D@T@T@V@V!!! !8DDG''))=DGY%HII 1w}$G..00Hr)c*t|jSr6)hashrqrls r'r>z+ConstDictVariable._HashableTracker.__hash__s-.. .r)cNt|t|krdSt|trUtjt |t |ko(t fdt||DSt|r||uS||kS)NFc3JK|]\}}||VdSr6)_eq_impl)r:ururs r'r<z>ConstDictVariable._HashableTracker._eq_impl..sH00041H%%a++000000r)) r#r?ryrerxlenrGzipr)abrs @r'rz+ConstDictVariable._HashableTracker._eq_implsAww$q''!!uAu%% ,=1vvQ'C00008;Aq 000-- Av Av r)otherc&tj}t||s+tj|sJt |t||r ||j|jS||j|Sr6)rerxr?r is_literalr#rrq)rlrrs r'__eq__z)ConstDictVariable._HashableTracker.__eq__s(9HeX.. 2B2Me2T2T  U   T%** X(()>@VWWW$$T%:EBB Br)riN) __name__ __module__ __qualname____doc__rmpropertyrqr> staticmethodrboolrrsr)r'rxrhs       % %  % N / / /      C D C C C C C C Cr)rxrHriNc d|vr|dd|vr|dtjdi|tjt fd|DsJfd||}|fd|D|_t|j  |_ | |_ ||_ dS)Noriginal_itemsshould_reconstruct_allc3vK|]3\}}t|tfot|tV4dSr6)r?r)r:rcrurs r'r<z-ConstDictVariable.__init__..s^  1 q?H5 6 6 /1o..      r)c>t|r|n |Sr6)r?)keyrs r' make_hashablez1ConstDictVariable.__init__..make_hashables$$S(33F33# Fr)c.i|]\}}||Srsrs)r:rcrurs r'rvz.ConstDictVariable.__init__..s)MMMtq!}}Q//MMMr)rs)popsuperrmrerxrGrH_get_dict_cls_from_user_clsrsourcercopyrrf)rlrHrfkwargsdict_clsrr __class__s @@r'rmzConstDictVariable.__init__sI v % % JJ' ( ( ( #v - - JJ/ 0 0 0""6"""$5              G G G G G33H==XMMMMu{{}}MMMNN +?t{*K*K&K##jjll  r)cttjtjf|vr|}n t fd|jD}|vs J||tjurt}|S)Nc3$K|] }|v|V dSr6rs)r:baseaccepted_dict_typess r'r<z@ConstDictVariable._get_dict_cls_from_user_cls..s7T=P5P5P5P5P5P5Pr))dict collections OrderedDict defaultdictnext__mro__)rlrfrrs @r'rz-ConstDictVariable._get_dict_cls_from_user_clss#[%UV * * *HH!)!1H....... {. . .Hr)cHd|jDS)Ncli|]1\}}|j|2SrsrkrCr:rtrus r'rvz.ConstDictVariable.as_proxy..s0MMM$!Q MMMr)rHrs r'rCzConstDictVariable.as_proxys$MM$*:J:J:L:LMMMMr)czddd|jDzdzS)N{, c3|K|]7\}}|jd|V8dS)z: Nrk debug_reprrs r'r<z/ConstDictVariable.debug_repr..sY=AQ14??$$88 88r)})joinrHrs r'rzConstDictVariable.debug_reprsT iiEIZEUEUEWEW    r)cHd|jDS)Ncli|]1\}}|j|2Srsrkr~rs r'rvz8ConstDictVariable.as_python_constant.. sH   1 D # # % %q';';'='=   r)rrs r'r~z$ConstDictVariable.as_python_constants1   ((**    r)cp|d|jDS)NcHi|]\}}|j| Srsrrs r'rvz=ConstDictVariable.keys_as_python_constant..'s,LLLA''))1LLLr))install_dict_keys_match_guardrHrs r'keys_as_python_constantz)ConstDictVariable.keys_as_python_constant%s6 **,,,LL9I9I9K9KLLLLr)c|jSr6)rfrs r'r}zConstDictVariable.python_type)s }r)ct|tsJtj}t |o@|||jvo.t|j||t j Sr6)r?rrerxr8rHr DeletedVariable)rlrkrs r' __contains__zConstDictVariable.__contains__,sn"o.....$5 OO T  * Ttz((2,,79RSSS r)cbtd|jDS)Nc3LK|]}t|tj V dSr6)r?r rr:rcs r'r<z(ConstDictVariable.len..6sC  => 1i788 8      r))sumrHrKrs r'rzConstDictVariable.len5s>  BF*BSBSBUBU      r)cvjp1tfdjDS)Nc3K|]:\}}j|j|V;dSr6) is_new_itemrr%rk)r:rrMrls r'r<z2ConstDictVariable.has_new_items..;s^2 2 U   T044SV<U&&(( >U->->-@-@ >emmoo&&"U]]__*=*== =%yyBuII%%r)cHd}|jD]a\}}|j|j}|||s|jr ||j|||dz }b|td|dS)Nrr BUILD_MAPr0)rHrr%rkrr append_outputr )rlcodegennum_argsrrMitems r'reconstruct_kvs_into_new_dictz/ConstDictVariable.reconstruct_kvs_into_new_dictFs***,,  JC&**3622De,, 0K A 0(KKKLLLLLr)rrc|jtjurRfd|t dddS|dS)NctdgS)Nr extend_outputcreate_load_python_modulercreate_load_attrrsr'z/ConstDictVariable.reconstruct..Vs>--99+FF00??r)rF)rfrr add_push_nullrrr rlrs `r' reconstructzConstDictVariable.reconstructRs =K3 3 3  ! !     . .w 7 7 7  ! !"6q%"@"@ A A A A A  . .w 7 7 7 7 7r)r1r r0ct|}||jvrtt||j|Sr6)rerxrHr KeyError)rlr1r0rs r''getitem_const_raise_exception_if_absentz9ConstDictVariable.getitem_const_raise_exception_if_absentbs@ 0055 dj $Xr 2 2 2z#r)ct|}||jvr3d|jd}t dd|j|dgt j|j|S)NzDictionary key z not found during tracingzkey not found in dictzKey z>Check if the key exists in the dictionary before accessing it.gb_typecontext explanationhints)rerxrHrMrr USER_ERROR)rlr1r0rmsgs r' getitem_constzConstDictVariable.getitem_constjs0055 dj HCIHHHC /*sy**T&1     z#r)cft|}||jvrdS|j|Sr6)rerxrH)rlr0rs r'maybe_getitem_constz%ConstDictVariable.maybe_getitem_constys30055 dj 4z#r)cp||vsJt|}t|j|}t|j|}t|tj r| dSdSr6) rerxryrHkeysindexrr?r r@r)rlr0rroriginal_key_vts r'realize_key_vtz ConstDictVariable.realize_key_vtsd{{{{0055djoo''((..s33 3 8 8 : :;;EB oy'D E E &  # # % % % % % & &r)cp|jr.t|tjdSdSr6)rr make_guardrDICT_KEYS_MATCHrs r'rz/ConstDictVariable.install_dict_keys_match_guards< ; I $//,*FGG H H H H H I Ir)c ddlm}|jsdS|jj|rdS|d|v}|djmt |d|rWt|tj t|j |dj | dS|djr5|r||ddS|dSdS)Nrrr)rinvert)rroutput side_effects is_modifiedr?rr functoolspartialr#CONTAINS_GUARDrMrr)rlr1r,rcontainss r'install_dict_contains_guardz-ConstDictVariable.install_dict_contains_guards& '&&&&&{  F 9 ! - -d 3 3  F7d? 7> !ja:J&K&K ! %T 1 GM#+|     !W^ 5 5##DG,,,,,2244444  5 5r)r,list[VariableTracker]rdict[str, VariableTracker]rcPddlm}mtj|ot |d}|dkrvt jtj|g|Ri|}|j j ||j |j jdS|dkr?t|dkrt!|||||dS|dkrb|s|rt!||||jr$|j j|jt-|S|dkrmt|rt!||||jr$|j j|jt/|S|dkr{|s|rt!||||jr$|j j|j|s|rt1t2|t5|S|d krc||s|rt!||||j t;d S|d krJ|s|rt!|||jt|j S|d kr|r|st?|d||st|d ksJ|j j ||d|j |d<jdS|dkr|r|rs|d|_ |j j ||j !|djdS|dkrt|dvrt!|||st?|d|d|vrA|"||t|dkrjdS|dS|#||dS|dkr|rt|dvrt!|||st?|d|d|vrF|"||t|dkrt1tH||dSd|_ |j j ||j %|dS|dkr|rtM|j'tr>tM|j'tPj)st|rt!|||j s(jd}t1tH||g|j'tPj)urt|dksd|vrt|dkr$tU|dr|dj+} n?|,dx} rtU| r| j+} nt!|||j -| \} } n|j -\} } d|_ |j j |t j.| j/| gS|dkrc|s|rt!||d|_ |j j ||j 0jdS|dkrd|rO|t|dk} t|dk} | s| r|j j || rtU|dtr#|d|d}n"|1|t|d}|j |j | r;fd| D}|j |jdSte3||||S|dkrft|st!|||st?|d|"|||d|v}j|S|dkr|rt|dvrt!|||st?|d||rJt|d ksJ|4|d}||St|dkrjd}n|d}|j j |||j |d<|S|dkr||j j ||d|vrt1tH|d} t|d kr#tU|dr |dj+} |r4d|vr0tU|dr|,dj+} |d}|j 5|| jdS|dkrstm|tr^t|dkrt!||t j7tpj9:|||dgiS|dkr,j|3|d||j+ S|d krXt|dksJ|d}tm|tt j;fsOd!|<j=d"|<j=d#}t1t2||g|j'|j'h}t}d$|Dr tPj)nt}|||j t;d|%}|d|j |dj |S|d&kr|3|d|||Ste3||||S)'Nr)rRrrrm __getitem__rHrrKr)rH mutation_typer__len__ __setitem__r __delitem__Tr%)rrrpopitemzpopitem(): dictionary is emptyr+last)rclearupdatecJi|]\}}j|| Srs)create)r:rtrurrs r'rvz1ConstDictVariable.call_method..]sH Aq!!8!1!8!;!;<.s/JJ!z![%<==JJJJJJr))rHr rrf__ior__)?rrRrrerxr8r r call_dictrrmutationrHrrrrrrrguard_on_key_orderaddDictItemsVariableDictKeysVariabler r/DictValuesVariableclonerr is_mutabler2rrrrrrrrfrrr?rMr%rrFrkrcall_custom_dictr call_methodrrrrWr dict___eq__ call_functionUserDefinedDictVariabler}rr)rlr1namer,rrR arg_hashable temp_dict_vtrrrurthas_arg has_kwargsdict_vtrrMrcrrtsrf new_dict_vtrrrs @@r'r*zConstDictVariable.call_methods 87777777$54 DG 4 4 :  D94T::D#L I " + +D 1 1 1 J  l0 1 1 1*#*400 0 ] " "4yyA~~#B---??DGLL L W__ .v .#B---  . . 0 0 0{ > ,00===$T** * V^^4yy .#B---  . . 0 0 0{ > ,00===#D)) ) X   .v .#B---  . . 0 0 0{ > ,00=== 8v 8(B777%d++ + V^^  . . 0 0 0 .v .#B---::joo''7G7I7IRV Y   .v .#B---  . . 0 0 0*#*3tz??;; ; ] " "t'8'8 " * a)))  . . 0 0 0 0#d))q...0 I " + +D 1 1 1,0GDJxxQ(( )*#*400 0 ] " "| "8I8I "  . . 0 0 0*.D ' I " + +D 1 1 1 J " "88DG#4#4 5 5 5*#*400 0 U]]4yy&&#B--- * a)))Awd""00T:::t99>>2+24888Aw%%b$q'22 2 U]]t00]4yy&&#B--- * a)))Awd""00T:::t99>>,Xr:::Aw*.D ' I " + +D 1 1 1:>>((47"3"344 4 Y  4??#4#4 4=$// ."4=+2IJJ .II . $B---: C-&-.NOO(2SEBBBB} 777D Q&F"2"2t99>>ja:J&K&K>7=DD!::f---a2:aAQ3R3R27DD'D111z))t)4411z))++1*.D ' I " + +D 1 1 1*AD!955 5 W__ .v .#B---*.D ' I " + +D 1 1 1 J     *#*400 0 X  $//"3"3   . . 0 0 0$ii1nGVqJ C* C &//5555!$q'+<==VQ==???"&q'"1"B"B2tTRSW"U"UJ%%gm444.$*LLNNFJ%%f---.'.t444ww**2tT6BBB ^ # #t99 .#B--- * a)))  , ,R 6 6 6Aw$H*#*844 4 \ ! !doo&7&7 !4yy&&#B--- * a)))  . . 0 0 0  :t99>>>>,,T!W55E  t99>>/(/55AAQA &//55501 88DG,,- ] " "  . . 0 0 0 I " + +D 1 1 1Awd""(2666D4yyA~~*T!W6F"G"G~Aw} 0f$$vf~/?@@%zz&))/(47##C J " "3T " 2 2 2*#*400 0 X  & ## # 4yyA~~#B---1)2GHHVVT47OR X  *#*$$R4@@FF X  t99>>>>GE&)9+LM D<4;K;K;M;M;V<<!--//8<<<)BcUCCCC-0BJJrJJJJJ ''   . . 0 0 0**joo''.00! %K G 1 1 3 3 3   $ $T!W] 3 3 3  Y     R4 8 8 8K77&&r4v>> >r)cp|d|jDS)Ncg|] }|j Srsrkrs r' z9ConstDictVariable.unpack_var_sequence..s000000r))rrHrrlr1s r'unpack_var_sequencez%ConstDictVariable.unpack_var_sequences4 **,,,00djoo//0000r)c|jtur6||jjvrtjdStjdSd|jd}t dd|j|dgt jdS) NTFz hasattr on z is not supportedzunsupported hasattr operationzClass z+Consider using a regular dictionary insteadr)rfrr$rrrr SUPPORTABLE)rlr1r.rs r'call_obj_hasattrz"ConstDictVariable.call_obj_hasattrs =D t}---'.t444#*511 1r' __classcell__rs@r'reres!/N   'N SCSCSCSCSCSCSCSCp!!!!O_45!!  !!!!!!!!!!!!F&NNN      MMM $     S     t    &&& M M M8888 )0?  7 o     &/&&&&III%5%5%5Ni?& i? - i?  i?i?i?i?i?i?V 111   &'''''''''r)recleZdZdeddffd ZdZdZdd Zd ed d e e d fdd fd Z xZ S)MappingProxyVariabledv_dictriNc vtjdi|t|tsJ||_dSr@)rrmr?rerIrlrIrrs r'rmzMappingProxyVariable.__init__sB""6"""'#455555 r)ctjSr6)typesMappingProxyTypers r'r}z MappingProxyVariable.python_types %%r)c6|j|Sr6)rIr;r:s r'r;z(MappingProxyVariable.unpack_var_sequences|//333r)rrc|jr3d|jd}tdd|j|dgtjfd|jtdd dS) Nz*Preexisting MappingProxyVariable (source: zS) cannot be reconstructed because the connection to the original dict will be lost.z%mapping proxy cannot be reconstructedSource: zCUse a mapping proxy constructed in the same `torch.compile` region.rctdgS)NrN)rrrMrrsr'rz2MappingProxyVariable.reconstruct..s?G))55e<<,,-?@@r)rF)rrrr=rrIrr )rlrrs ` r'rz MappingProxyVariable.reconstructs ; LT[LLL  ?04;00Y&2               21e<<=====r)r,rrc|jr>|jjr d}t dd|jd|ddg|j||||S)NaA dict has been modified while we have an existing mappingproxy object. A mapping proxy object, as the name suggest, proxies a mapping object (usually a dict). If the original dict object mutates, it is reflected in the proxy object as well. For an existing proxy object, we do not know the original dict it points to. Therefore, for correctness we graph break when there is dict mutation and we are trying to access a proxy object.z-mapping proxy affected by dictionary mutationrQz, Dict mutation detectedzNAvoid modifying dictionaries that might be referenced by mapping proxy objectszROr avoid using the mapping proxy objects after modifying its underlying dictionaryr)rrrhas_existing_dict_mutationrrIr*)rlr1r.r,rrs r'r*z MappingProxyVariable.call_method"s ; 291LLNN 7  GH4;HHHdh     |''D$???r)rA) rrrrermr}r;rlistrstrr*rErFs@r'rHrHs 1 &&&444>>>>6@$% @ S++, @  @@@@@@@@r)rHceZdZdZdZdS)NNModuleHooksDictVariablecdSr6rsrs r'rz7NNModuleHooksDictVariable.install_dict_keys_match_guardB r)cdSr6rsrlr1r,s r'rz5NNModuleHooksDictVariable.install_dict_contains_guardErZr)N)rrrrrrsr)r'rXrX@s2        r)rXcjeZdZddfd ZfdZfdZedZ dfd Zd Z xZ S)DefaultDictVariableNric ltj||fi||tjusJ||_dSr6)rrmrrdefault_factory)rlrHrfr`rrs r'rmzDefaultDictVariable.__init__JsF33F333;22222.r)c|jtttfvr |jsdSt S)NF)r`rUryrrHris_python_constantrlrs r'rbz&DefaultDictVariable.is_python_constantOs<  eT': : :4: :5ww))+++r)cd|jdtdS)Nz defaultdict(r))r`rrrcs r'rzDefaultDictVariable.debug_reprVs= W4/::<< W W@R@R@T@T W W W r)ct|tjr!|jtt t tfvSt|tjj Sr6) r?r rRfnrUryrset functionsBaseUserFunctionVariablers r'is_supported_argz$DefaultDictVariable.is_supported_arg[sD c94 5 5 Q6dE455 5c9#6#OPP Pr)r,r rr rc|dkrt|dksJ|d|vr|||dS|jt|d|j|gi}t |d|d|f||St ||||S)Nr rrr)rrr`rr,rr*)rlr1r.r,r default_varrs r'r*zDefaultDictVariable.call_methodbs = t99>>>>Aw$))"d1g666'/"d1g<000"&"6"D"DRR"P"PKGG''MDG[+A6'&77&&r4v>> >r)cfd|j|t dddS)NctdgS)Nrrrsr'rz1DefaultDictVariable.reconstruct..}s>G))55kBB,,];;r)rF)rr`rrr rs `r'rzDefaultDictVariable.reconstructzs}        $%%% **733321e<<=====r)r6rrB) rrrrmrbrrrkr*rrErFs@r'r^r^Is/////// ,,,,,     QQ\Q ?& ? - ?  ??????0 > > > > > > >r)r^ceZdZdZejZdeeddffd Z dZ e dZ e dZd Zd Zd ZddZdZdeedeeefddffd ZdddefdZdZfdZxZS) SetVariablez.We model a sets as dictionary with None valuesrHriNc t|t}t j|fi|dSr6)rfromkeysrq_default_valuerrmrlrHrrs r'rmzSetVariable.__init__sI  e[%?%?%A%ABB))&)))))r)c|jsdSddd|jDzdzS)Nzset()r,c3HK|]}|jVdSr6rr:rts r'r<z)SetVariable.debug_repr...!O!O!$//"3"3!O!O!O!O!O!Or)rrHrrrs r'rzSetVariable.debug_reprsJz V7!O!OTZ__=N=N!O!O!OOOORUU Ur)cNt|jSr6)rhrHrrs r' set_itemszSetVariable.set_itemss4:??$$%%%r)c*tjdSr6)rrrsr)r'rtzSetVariable._default_values &t,,,r)c$d|jDS)Nc@h|]}|jSrsrrys r' z'SetVariable.as_proxy..s"888A 888r)r}rs r'rCzSetVariable.as_proxys888888r)ctSr6)rhrs r'r}zSetVariable.python_types r)c$d|jDS)Nc@h|]}|jSrsrrys r'rz1SetVariable.as_python_constant..s&BBBa''))BBBr)rrs r'r~zSetVariable.as_python_constantsBB4>BBBBr)rrc|d|jD|tdt |jdS)Ncg|] }|j Srsr8rs r'r9z+SetVariable.reconstruct..666!666r) BUILD_SETr)foreachr}rr rrs r'rzSetVariable.reconstructsW66t~6667770#dnBUBUVVVWWWWWr)c R |d|g|Did|D}n`#t$rS}tt||t t t j|jYd}~nd}~wwxYwtj ||S)Nc6g|]}|Srsr~rs r'r9z0SetVariable._fast_set_method..s$@@@Q!&&((@@@r)c>i|]\}}||Srsrrs r'rvz0SetVariable._fast_set_method..s*HHHA1a**,,HHHr)r+) rH Exceptionr r#rUmaprrr,rbuild)rlr1rgr,rresexcs r'_fast_set_methodzSetVariable._fast_set_methods "@@4-$-@@@HHHHHCC    $S 2D-=-Dch)O)O$P$P          $R---s14 BA B  Br,rrc ddlm}|dvra|||rU|tur:|}||t ||||S|dkrt jtj|g||R}|j j ||j |j |j tjdS|dkrL|rJt#|dkrt%||d}|dt&f}n|d kr|rJ|rJ |jj}nX#t0$rK} t3t0|t5t7tj| j Yd} ~ nd} ~ wwxYwt;|||f||S|d krbt#|dkrt%|||rJt jt@j!"|||dgiS|d kr:|rJt jt@j#"||g|iS|d kr:|rJt jt@j$"||g|iS|dkr:|rJt jt@j%"||g|iS|dkr:|rJt jt@j&"||g|iS|dkr:|rJt jt@j'"||g|iS|dkr]t#|dkrt%|||rJt jt@j("||g|iS|dkr]t#|dkrt%|||rJt jt@j)"||g|iS|dkrN|*r:|rJt jt@j+"||g|iS|dkr^|rJt#|dksJ|d|vrt3t0|| t;|d ||S|dkr\|rJt#|dksJ|d|vr$t;|d ||StjdS|dvrt#|dkrt%||tXj-tXj.d} |d/} ta| t&s/t jt"|| gi} t j| 1|"||| giS|dvrd dddd1|} te|dt&t j3fs`tjd|d|4d|d4d} t3tj|| g ||| ||S|dvrte|dt&t j3fs`tjd|d|4d|d4d} t3tj|| g d dddd1|} ||| |||S|dkr}te|dt&t j3fstjdS||d||}tjt#|jdkS|tlvrxte|dt&t j3fstjtnStjtm||j|djSt;||||S) Nr)check_constant_args) isdisjointunion intersection differencesymmetric_differencermr#rrrrr+rrintersection_updaterrdifference_updatersymmetric_difference_updaterremovediscard)rM)issubset issuperset)__and__r__xor____sub__z unsupported operand type(s) for z: 'z' and 'r)__iand__r__ixor____isub__rF)8utilsrr}rhrgetattrr rRcall_setrrr!rHrrrrrrrqrtr}rrkrr rUrr,rr*rWr set_isdisjointr,set_intersectionset_intersection_update set_unionset_differenceset_difference_updateset_symmetric_differenceset_symmetric_difference_updater( set_updateoperatorlegerrr%r?UserDefinedSetVariablepython_type_namer/rNotImplemented)rlr1r.r,rrpy_type temp_set_vtresultr;oprmrrrs r'r*zSetVariable.call_methods  0/////   $#D&11   ""c))&&((G((WWd-C-CT6RR R :  A)3C88A"UtUfUUUK I " + +D 1 1 1 J      J  k/ 0 0 0#*400 0 U]]  :4yyA~~#B--- DG[7799:DD U]]  :OO8 ++--0   (btC0@0G,P,P'Q'Q  GG  D6)V < < <M \ ! !4yyA~~#B---  :1(mBtAw44 5^ # #  :1*mB  r22 3* * *  :11mB  r22 3W__  :1)2EFFTTTMDM2 \ ! !  :1(mB  r22 3( ( (  :1/mB  r22 3+ + +4yyA~~#B---  :12mB  r22 32 2 24yyA~~#B---  :19mB  r22 3X  $//"3"3   :1)2FGGUUTMDM2 X    :t99>>>>Awd""(2DAAAA77&&r5$?? ? Y    :t99>>>>Aw$ww**2udFCCC'.T:::: / / /4yyA~~#B---%K&kBGOO%%E%-- V!1#66DDR%RTUU,RVVD\\::HHT5M2 @ @ @)!1'  c$ii  d1g Y5U'VWW D&-}t}}@U@U@W@W}}`def`g`x`x`z`z}}})BcUCCCC##B488 8 D D Dd1g Y5U'VWW D&-}t}}@U@U@W@W}}`def`g`x`x`z`z}}})BcUCCCC1#9/  c$ii    RD& 1 1 1K X  d1g Y5U'VWW 6'.u555  %;T6JJA#*3q{+;+;q+@AA A + + +d1g Y5U'VWW ?'.~>>>#*&t,T^T!W=NOO ww""2tT6:::s&F GAGGr1r r0c td)NzIllegal to getitem on a set) RuntimeError)rlr1r0s r'rzSetVariable.getitem_consths8999r)cdSr6rsrs r'rz)SetVariable.install_dict_keys_match_guardk r)cLt||dSr6)rr)rlr1r,rs r'rz'SetVariable.install_dict_contains_guardos# ++B55555r)rA)rrrrr SET_CONTAINSrrUrrmrrr}rrtrCr}r~rrrrVr*rrrrErFs@r'rqrqs88!.N*O$*  ******VVV &&X&--\-999CCCXXXX . . .e;?# e; S/)* e;  e;e;e;e;e;e;N: 7:o::::   666666666r)rqceZdZdeeddffd ZdZedZdZ dZ dd Z d eed e e efddffd ZxZS)rUrHriNc <tj|fi|dSr6rrmrus r'rmzFrozensetVariable.__init__t* ))&)))))r)c|jsdSddd|jDzdzS)Nz frozenset()rrwc3HK|]}|jVdSr6rrys r'r<z/FrozensetVariable.debug_repr..rzr)rr{rs r'rzFrozensetVariable.debug_repr{sJz V =!O!OTZ__=N=N!O!O!OOOORUU Ur)c4|jSr6)rHrrs r'r}zFrozensetVariable.set_itemssz   r)ctSr6) frozensetrs r'r}zFrozensetVariable.python_typer)c>td|jDS)Nc@h|]}|jSrsrrys r'rz7FrozensetVariable.as_python_constant..s&LLL!$1133LLLr))rr}rs r'r~z$FrozensetVariable.as_python_constants!LLT^LLLMMMr)rrcd|jDfdt dddS)Ncg|] }|j Srsr8rs r'r9z1FrozensetVariable.reconstruct..rr)cVdgS)Nr)rcreate_load_globalrsr'rz/FrozensetVariable.reconstruct..s-G))..{;;r)rF)rr}rrr rs `r'rzFrozensetVariable.reconstructsy66t~666777        21e<<=====r)r,rrc&|dvrtd|d|dkrtjdS|dvr8t||||}t |jSt||||S)Nr#rrrrrIllegal call_method z on a frozensetrm)rrrr)rrrrr*rUrH)rlr1r.r,rrrs r'r*zFrozensetVariable.call_methods I I IKdKKKLL L Z  $*400 0    ##BdF;;A$QW-- -ww""2tT6:::r)rA)rrrrUrrmrrr}r}r~rrrVr*rErFs@r'rUrUss*O$*  ******VVV !!X!NNN > > > >;?# ; S/)* ;  ;;;;;;;;;;r)rUceZdZdeeddffd ZdZdZdZe dZ d Z d Z d eed e eefdd ffd ZxZS)DictKeySetVariablerHriNc <tj|fi|dSr6rrus r'rmzDictKeySetVariable.__init__rr)c|jsdSddd|jDzdzS)Nz dict_keys([])z dict_keys([rwc3HK|]}|jVdSr6rrys r'r<z0DictKeySetVariable.debug_repr..s.HH14??,,HHHHHHr)z])r{rs r'rzDictKeySetVariable.debug_reprsSz "?((HHdjoo6G6GHHHHHI r)cdSr6rsrs r'rz0DictKeySetVariable.install_dict_keys_match_guardrr)cdSr6rsr\s r'rz.DictKeySetVariable.install_dict_contains_guardrr)c|jSr6rrs r'r}zDictKeySetVariable.set_itemss zr)ctSr6rrs r'r}zDictKeySetVariable.python_typerr)cztd|jDdS)Nc@h|]}|jSrsrrys r'rz8DictKeySetVariable.as_python_constant..s& ? ? ?1QT $ $ & & ? ? ?r))rrsr}rrs r'r~z%DictKeySetVariable.as_python_constants6}} ? ? ? ? ?  $&& r)r,rrcz|dvrtd|dt||||S)Nrrz on a dict_keys)rrr*rlr1r.r,rrs r'r*zDictKeySetVariable.call_methodsI I I IKdKKKLL Lww""2tT6:::r))rrrrUrrmrrrrr}r}r~rrVr*rErFs@r'rrs*O$*  ******      X ;?# ; S/)* ;  ; ; ; ; ; ; ; ; ; ;r)rceZdZUdZdZeeed<deddffd Z e dZ e dZ d Z dd Zd Zdeddeedfddffd ZxZS)DictViewVariablezu Models _PyDictViewObject This is an "abstract" class. Subclasses will override kv and the items method NkvrIric tjdi||jdvsJt|tsJ||_dS)N)rrKrHrs)rrmrr?rerIrKs r'rmzDictViewVariable.__init__sV""6"""w55555'#455555 r)cPt|jj|jSr6)rrIrHrrs r' view_itemszDictViewVariable.view_itemss!3wt|)4733555r)ctr6)NotImplementedErrorrs r' view_items_vtzDictViewVariable.view_items_vts "!r)c|jSr6)rr:s r'r;z$DictViewVariable.unpack_var_sequences !!r)rrc||j||j|ddS)Nr)rI load_methodrr*rs r'rzDictViewVariable.reconstructsD DG$$$Ar)c||jvrtjdStjdS)NTF)r}r$rr)rlr1r.s r'r>z!DictViewVariable.call_obj_hasattrs> 4##%%. . .#*400 0&u---r)r,rrc|dkr|j||||St||||S)Nr)rIr*rrs r'r*zDictViewVariable.call_method sJ 9  <++BdFCC Cww""2tT6:::r)rA)rrrrrrrV__annotations__rermrrrr;rr>rUrr*rErFs@r'rrs# B  1 66X6""X" """ ... ;$% ; S++, ;  ; ; ; ; ; ; ; ; ; ;r)rc~eZdZdZedZedZdZdedde e dfddffd Z xZ S) r%rc*t|jSr6)rhrrs r'r}zDictKeysVariable.set_itemss4?###r)c$d|jDS)Ncg|] }|j Srsr8rs r'r9z2DictKeysVariable.view_items_vt..!s......r)rrs r'rzDictKeysVariable.view_items_vts/.do....r)ctSr6rrs r'r}zDictKeysVariable.python_type#rr)r,rrric|dkr|j||||S|dvr:t|j|}||dj}t |S|t vrst |dttfstj tStj t ||j|djSt||||S)Nr)rrrrrrrrr) rIr*rr}rqrr?r%rrrr)rlr1r.r,rrrrs r'r*zDictKeysVariable.call_method&s > ! !<++BdFCC C     --A$q'#$$Aq>> ! ) ) )d1g 5E'FGG ?'.~>>>#*&t,T^T!W=NOO ww""2tT6:::r)) rrrrrr}rr}rUrrVr*rErFs@r'r%r%s B $$X$//X/;$% ; S++, ;  ;;;;;;;;;;r)r%c.eZdZdZedZdZdS)r&rKc*t|jSr6)rUrrs r'rz DictValuesVariable.view_items_vtJsDO$$$r)ctSr6)rrs r'r}zDictValuesVariable.python_typeNsr)Nrrrrrrr}rsr)r'r&r&FsA B %%X%r)r&c.eZdZdZedZdZdS)r$rHc$d|jDS)NcJg|] \}}tj|j|g!Srs)r rFrkrs r'r9z3DictItemsVariable.view_items_vt..Xs-OOOtq! 'q 22OOOr)rrs r'rzDictItemsVariable.view_items_vtUsPOtOOOOr)ctSr6)rrs r'r}zDictItemsVariable.python_typeZsr)Nrrsr)r'r$r$RsD B PPXPr)r$r6)r s( 333333********1111116666666666NNNNNNNN<<<<<<<<00000000))))))43333333&&&&&&E//////DDDDDD>>>2 2 2 jw 'w 'w 'w 'w 'w 'w 'w 'tC@C@C@C@C@?C@C@C@L     1   =>=>=>=>=>+=>=>=>Fd6d6d6d6d6#d6d6d6N=;=;=;=;=; =;=;=;@/;/;/;/;/;/;/;/;d/;/;/;/;/;/;/;/;d,;,;,;,;,;',;,;,;^     )        (     r)