sh"6SSKrSSKrSSKJr SSKJr SSKJrJ r J r J r SSK J r SSKJrJr /SQr"S S \5rS r\"S \S 9S5r\"S\S 9\4Sj5rSrSr"SS5rSr"SS\5rSrSrg)N) deprecated)warn) ambiguitiesAmbiguityWarningorderingsuper_signature) expand_tuples) isvariadicVariadic) MDNotImplementedErrorambiguity_warn halt_orderingrestart_orderingvariadic_signature_matches_itervariadic_signature_matches DispatchersourceMethodDispatcher str_signature warning_textc\rSrSrSrSrg)r z+A NotImplementedError for multiple dispatchN)__name__ __module__ __qualname____firstlineno____doc____static_attributes__rڑ/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/fx/experimental/unification/multipledispatch/dispatcher.pyr r s5r!r cL[[URU5[5 g)aRaise warning when ambiguity is detected Parameters ---------- dispatcher : Dispatcher The dispatcher on which the ambiguity was detected ambiguities : set Set of type signature pairs that are ambiguous within this dispatcher See Also: Dispatcher.add warning_text N)rrnamer) dispatcherrs r"rrs joo{ 35EFr!z?`halt_ordering` is deprecated, you can safely remove this call.categorycg)z5Deprecated interface to temporarily disable ordering.Nrrr!r"rr.r!z`restart_ordering` is deprecated, if you would like to eagerly order the dispatchers, you should call the `reorder()` method on each dispatcher.cg)z4Deprecated interface to temporarily resume ordering.Nr) on_ambiguitys r"rr6r)r!c## [U5n[U5nUH/n[XC5nUv [U5(aM$[U5nM1 [U5nSv g![a [U5(deSv gf=f7f)a'Check if a set of input types matches a variadic signature. Notes ----- The algorithm is as follows: Initialize the current signature to the first in the sequence For each type in `types`: If the current signature is variadic If the type matches the signature yield True Else Try to get the next signature If no signatures are left we can't possibly have a match so yield False Else yield True if the type matches the current signature Get the next signature FTN)iternext issubclassr StopIteration)typesfull_signaturesigitersigtypmatchess r"rr?s$>"G w-CS& #w-C  w-CK  c?? "?J s.9BB AB B?BBBc<U(de[[X55$N)allr)r1r2s r"rrfs > .uE FFr!c\rSrSrSrSrSSjrSr\S5r \S5r S r \ S 5r \4S jrS rS r\rSrSr\"S\S9S5rSrSr\ S5rSrSrSrSrSrg)rla>Dispatch methods based on type signature Use ``dispatch`` to add implementations Examples -------- >>> # xdoctest: +SKIP("bad import name") >>> from multipledispatch import dispatch >>> @dispatch(int) ... def f(x): ... return x + 1 >>> @dispatch(float) ... def f(x): ... return x - 1 >>> f(3) 4 >>> f(3.0) 2.0 )rr$funcs _ordering_cachedocNcHU=UlUl0UlX l0Ulgr8)r$rr<r?r>)selfr$r?s r"__init__Dispatcher.__init__s#$(( DM  r!c^^^UUU4SjnU$)a]register dispatcher with new implementation >>> # xdoctest: +SKIP >>> f = Dispatcher("f") >>> @f.register(int) ... def inc(x): ... return x + 1 >>> @f.register(float) ... def dec(x): ... return x - 1 >>> @f.register(list) ... @f.register(tuple) ... def reverse(x): ... return x[::-1] >>> f(1) 2 >>> f(1.0) 0.0 >>> f([1, 2, 3]) [3, 2, 1] c0>TR"TU40TD6 U$r8)add)funckwargsrAr1s r"_df Dispatcher.register.._dfs HHUD +F +Kr!r)rAr1rHrIs``` r"registerDispatcher.registers,  r!c[[S5(a0[R"U5nURR 5$g)N signature)hasattrinspectrN parametersvaluesclsrGr4s r"get_func_paramsDispatcher.get_func_paramss7 7K ( (##D)C>>((* * )r!c^URU5nU(aK[RmU4SjU5n[SU55n[ U4SjU55(aU$gg)z1get annotations of function positional parametersc3x># UH/nURTRTR4;dM+Uv M1 g7fr8)kindPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORD).0param Parameters r" 2Dispatcher.get_func_annotations..s;#E::--y/N/NOP#s*: :c38# UHoRv M g7fr8) annotation)r\r]s r"r_r`sEfU 0 0fc3>># UHoTRLv M g7fr8)empty)r\annr^s r"r_r`sE#ioo-sN)rUrPr^tupler9)rTrGparams annotationsr^s @r"get_func_annotationsDispatcher.get_func_annotationssf$$T* ))I#F EfEEKEEEE""F r!c U(dURU5nU(aUn[SU55(a$[U5HnURXB5 M g/n[ USS9Hupg[ U[ [45(d6SRSU55n[SUSUS UR35e[ U[5(aQU[U5:wa [S 5e[U5S:wa [S 5eUR[US 5 MURU5 M X R[U5'UR R#5 U?g![&a gf=f) aAdd new types/method pair to dispatcher >>> # xdoctest: +SKIP >>> D = Dispatcher("add") >>> D.add((int, int), lambda x, y: x + y) >>> D.add((float, float), lambda x, y: x + y) >>> D(1, 2) 3 >>> D(1, 2.0) Traceback (most recent call last): ... NotImplementedError: Could not find signature for add: >>> # When ``add`` detects a warning it calls the ``on_ambiguity`` callback >>> # with a dispatcher/itself, and a set of ambiguous type signature pairs >>> # as inputs. See ``ambiguity_warn`` for an example. c3B# UHn[U[5v M g7fr8) isinstancerg)r\r5s r"r_!Dispatcher.add..s;#z#u%%sNr)start, c3z# UH1n[U[5(a URO [U5v M3 g7fr8)rntyperstr)r\cs r"r_ros,$KTa*Q"5"5AJJ3q6A9s9;zTried to dispatch on non-type: z In signature: In function: z+Variadic signature must be the last elementzVariadic signature must contain exactly one element. To use a variadic union type place the desired types inside of a tuple, e.g., [(int, str)]r)rjanyr rF enumeraternrslistjoin TypeErrorr$lenappendr r<rgr>clearr=AttributeError) rArNrGrityps new_signatureindexr5str_sigs r"rFDispatcher.addsr"33D9K'  ;; ; ;%i0$1  #IQ7JEcD$<00))$KT$ 5cU;&&-Y/$$(II;0#t$$C N*#$QRRs8q=#@ $$Xc!f%56$$S)184,0 5'(     s,E// E<;E<c\UR$![a UR5s$f=fr8)r=r~reorderrAs r"rDispatcher.orderings, ">> ! "<<> ! "s ++c[UR5=Uln[UR5nU(aU"X5 U$r8)rr<r=r)rAr+odambs r"rDispatcher.reorders5&tzz22$**%   # r!c >[UVs/sHn[U5PM sn5nURUnU"U0UD6$s snf![aTnUR"U6nU(d&[ SUR S[U5S35UeXPRU'SnANgSnAff=f![amnUR"U6n[U5 UH"nU"U0UD6s sSnA$![a M f=f [ SUR S[U5S35UeSnAff=f)NCould not find signature for : <>zMatching functions for z(> found, but none completed successfully) rgrsr>KeyErrordispatchNotImplementedErrorr$rr dispatch_iterr.)rAargsrHargr1rGer<s r"__call__Dispatcher.__call__ s9D1DStCyD12 &;;u%D (( (2 &==%(D)3DII;c-PUBVAWWXY"&KK   &% &&.E K000, &)99+Su!5 66^`  s]?AB% B"A BB"% D/DCDD C,(D+C,,+DDc"SURS3$)Nz >> # xdoctest: +SKIP >>> from multipledispatch import dispatch >>> @dispatch(int) ... def inc(x): ... return x + 1 >>> implementation = inc.dispatch(int) >>> implementation(3) 4 >>> print(inc.dispatch(float)) None See Also: ``multipledispatch.conflict`` - module to determine resolution order N)r<r.rr0rAr1s r"rDispatcher.dispatch,sJ$ JJ ::e$ $ **E23 3  s8 AAc'd# [U5nURHn[U5U:Xa3[[[X55(aUR UnUv ME[U5(dMW[ US5(dMl[X5(dM~UR UnUv M g7f)N)r{rr9mapr/r<r r)rAr1nrNresults r"rDispatcher.dispatch_iterFs JI9~"s3z5+L'M'MI. YJy}$=$=-e??!ZZ 2F L'sA+B01B0B0B0z1`resolve()` is deprecated, use `dispatch(*types)`r&c UR"U6$)zxDetermine appropriate implementation for this type signature .. deprecated:: 0.4.4 Use ``dispatch(*types)`` instead )rrs r"resolveDispatcher.resolveQs}}e$$r!c4URURS.$)Nr$r<rrs r" __getstate__Dispatcher.__getstate__[s DJJ77r!cnUSUlUSUl[UR5Ul0Ulg)Nr$r<)r$r<rr=r>)rAds r" __setstate__Dispatcher.__setstate__^s/fI wZ !$**- r!c>SUR3/nUR(aURUR5 /nURSSS2HnURUnUR (aRS[ U5S3nUS[U5-S-- nXTR R5- nURU5 MuUR[ U55 M U(a#URSSRU5-5 S RU5$) NzMultiply dispatched method: rz Inputs: - zOther signatures: z  ) r$r?r|rr<rrr{stripry)rAdocsotherr4rGss r"rDispatcher.__doc__ds.tyyk:; 88 KK !==2&C::c?D|| c 2337S3q6\D((\\'')) A ]3/0'  KK1HMM%4HH I{{4  r!cPUR"[[U56R$r8)rrrsr)rArs r"_helpDispatcher._help{s}}c$o.666r!c4[UR"U65 g)z8Print docstring for the function corresponding to inputsN)printrrArrHs r"helpDispatcher.help~s djj$ r!cvUR"[[U56nU(d [S5e[ U5$)NzNo function found)rrrsrzr)rArrGs r"_sourceDispatcher._sources/}}c$o./0 0d|r!c4[UR"U65 g)z:Print source code for the function corresponding to inputsN)rrrs r"rDispatcher.sources dllD!"r!)rr>r=r?r<r$r8)rrrrr __slots__rBrK classmethodrUrjrFpropertyrrrrr__repr__rrr FutureWarningrrrrrrrr rr!r"rrls$JI8++ ##$>@"" $28+H4 !;m%%8 !!,7! #r!rclS[R"U5S3nU[R"U5-nU$)NzFile: r)rP getsourcefile getsource)rGrs r"rrs6 &&t,-T2A G  d ##A Hr!c8\rSrSrSrSr\S5rSrSr Sr g) rizBDispatch methods based on type signature See Also: Dispatcher objrTc[[S5(aF[R"U5n[R"UR R 5SS5$g)NrNr)rOrPrNitlislicerQrRrSs r"rU MethodDispatcher.get_func_paramssD 7K ( (##D)C::cnn335q$? ? )r!cXlX lU$r8r)rAinstanceowners r"__get__MethodDispatcher.__get__s r!c[UVs/sHn[U5PM sn5nUR"U6nU(d%[SURS[ U5S35eU"UR /UQ70UD6$s snf)Nrrr)rgrsrrr$rr)rArrHrr1rGs r"rMethodDispatcher.__call__szD1DStCyD12}}e$%/ {#mE>R=SSTU DHH.t.v.. 2sA7)rTrN) rrrrrrrrUrrr rr!r"rrs, I@@  /r!rc2SRSU55$)zUString representation of type signature >>> str_signature((int, float)) 'int, float' rqc38# UHoRv M g7fr8)r)r\rTs r"r_ str_signature..s1Sc\\Src)ry)r4s r"rrs 991S1 11r!c SUS3nUS- nUH$nUSSRSU55-S-- nM& US- nUSRUVs/sH!nS [[U55-S US 3-PM# sn5- nU$s snf) zThe text for ambiguity warningsz* Ambiguities exist in dispatched function rz;The following signatures may result in ambiguous behavior:  rqc3D# UHnS[U5-S-v M g7f)[]N)r)r\rs r"r_warning_text..s Lt!}Q'7!7#!=ts rz, Consider making the following additions: z @dispatch(z) def z(...))ryrr )r$rtextpairrs r"rrs 8d CD JJD tyy Lt LLLtSS >>DFKK  =);< <e?T T  D K  s (A> )rP itertoolsrtyping_extensionsrwarningsrconflictrrrr utilsr variadicr r __all__rr rrrrrrrrrrrrr!r"rs(NN * 6/6 G E @ @ A  #1?  ?$NG ^#^#B  /z/:2 r!