sg6ddlmZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z mZddlZgdZGd d eZd Zed e dZede efdZdZdZGddZdZGddeZdZdZy))warnN) deprecated)ordering ambiguitiessuper_signatureAmbiguityWarning) expand_tuples)Variadic isvariadic) MDNotImplementedErrorambiguity_warn halt_orderingrestart_orderingvariadic_signature_matches_itervariadic_signature_matches DispatchersourceMethodDispatcher str_signature warning_textceZdZdZy)r z- A NotImplementedError for multiple dispatch N)__name__ __module__ __qualname____doc__p/var/www/html/venv/lib/python3.12/site-packages/torch/fx/experimental/unification/multipledispatch/dispatcher.pyr r s7rr cLtt|j|ty)aA Raise 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 rrrs joo{ 35EFrz?`halt_ordering` is deprecated, you can safely remove this call.categorycy)z5Deprecated interface to temporarily disable ordering.Nrrrrrr rz`restart_ordering` is deprecated, if you would like to eagerly order the dispatchers, you should call the `reorder()` method on each dispatcher.cy)z4Deprecated interface to temporarily resume ordering.Nr) on_ambiguitys rrr(r&rc#Kt|}t|}|D])}t||}|t|rt|}+ t|}dy#t$rt|sJdYywxYww)agCheck 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 rrr1s$>"G w-CS#& #w-C  w-CK  c? "?J s.7A8 A8 AA8A52A84A55A8c4|sJtt||SN)allr)r.r/s rrrXs > .unE FFrceZdZdZdZddZdZedZedZ dZ e d Z e fd Zd Zd ZeZd ZdZededZdZdZe dZdZdZdZdZy)ra 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_cachedocNcJ|x|_|_i|_||_i|_yr5)r!rr8r;r:)selfr!r;s r__init__zDispatcher.__init__rs%$(( DM  rcfd}|S)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] c0j|fi|Sr5)add)funckwargsr=r.s r_dfz Dispatcher.register.._dfs DHHUD +F +Krr)r=r.rCrDs``` rregisterzDispatcher.registerys*  rcttdr/tj|}|jj Sy)N signature)hasattrinspectrG parametersvaluesclsrBr1s rget_func_paramszDispatcher.get_func_paramss4 7K (##D)C>>((* * )rc|j|}|rDtjfd|D}td|D}t fd|Dr|Syy)z; get annotations of function positional parameters c3jK|]*}|jjjfvr|,ywr5)kindPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORD).0param Parameters r z2Dispatcher.get_func_annotations..s;9 00 66889s03c34K|]}|jywr5) annotation)rTrUs rrWz2Dispatcher.get_func_annotations..s %   %c3:K|]}|juywr5)empty)rTannrVs rrWz2Dispatcher.get_func_annotations..sE#3ioo-EsN)rNrIrVtupler6)rMrBparams annotationsrVs @rget_func_annotationszDispatcher.get_func_annotationssk$$T* ))I99F   %# %%KEEE""F rc |s|j|}|r|}td|Dr#t|D]}|j||yg}t |dD]\}}t |t tfs6djd|D}td|d|d |jt |trN|t|k7r td t|dk7r td |jt|d |j|||jt|<|j j# |`y#t&$rYywxYw) ay Add 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. c3<K|]}t|tywr5) isinstancer^)rTr2s rrWz!Dispatcher.add..s;#z#u%;sNr)start, c3jK|]+}t|tr |jn t|-ywr5)rdtyperstr)rTcs rrWz!Dispatcher.add..s3$D453=Q2EAJJ),Q%0$Ds13zTried 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)raanyr rA enumeraterdrhlistjoin TypeErrorr!lenappendr r8r^r:clearr9AttributeError) r=rGrBr`typs new_signatureindexr2str_sigs rrAzDispatcher.adds"33D9K'  ;; ;%i0 %t$ %  #IQ7 *JE3cD$<0))$D9B$DD"A#G229;004 {!<== #t$C N*#Es8q=#@ $$Xc!f%56$$S)/ *2,0 5'(     sE E! E!cZ |jS#t$r|jcYSwxYwr5)r9rsreorderr=s rrzDispatcher.orderings, ">> ! "<<> ! "s **c~t|jx|_}t|j}|r ||||Sr5)rr8r9r)r=r(odambs rryzDispatcher.reorders7&tzz22$**%  s # rc &t|Dcgc] }t|c}} |j|} ||i|Scc}w#t$rP}|j|}|s&t d|j dt|d|||j|<Yd}~bd}~wwxYw#t$rj}|j|}t||D] } ||i|ccYd}~S#t$rYwxYwt d|j dt|d|d}~wwxYw)NCould not find signature for : <>zMatching functions for z(> found, but none completed successfully) r^rhr:KeyErrordispatchNotImplementedErrorr!rr dispatch_iterr+)r=argsrCargr.rBer8s r__call__zDispatcher.__call__sCD1StCy12 &;;u%D i(( (2 & 4==%(D)3DII;c-PUBVAWWXY[`ab!%DKK   &% i&D&&.E K 000,  &)99+Su!5 66^`bgh i is]<AB B ABB D&D C D D C!D  C!!*D  Dc"d|jdS)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)r8r+rr-r=r.s rrzDispatcher.dispatchsO$ DJJ ::e$ $ ***E23 3  s7 AAc'2Kt|}|jD]y}t||k(r.ttt||r|j |}|?t|sKt |dsZt||sg|j |}|{yw)N)rprr6mapr,r8r r)r=r.nrGresults rrzDispatcher.dispatch_iter2s J !I9~"s3z5)+L'MI. YJy}$=-eY?!ZZ 2F L !sA"B%B4 BBz1`resolve()` is deprecated, use `dispatch(*types)`r#c |j|S)z Determine appropriate implementation for this type signature .. deprecated:: 0.4.4 Use ``dispatch(*types)`` instead )rrs rresolvezDispatcher.resolve>s t}}e$$rc4|j|jdS)Nr!r8rrzs r __getstate__zDispatcher.__getstate__Fs % %rcn|d|_|d|_t|j|_i|_y)Nr!r8)r!r8rr9r:)r=ds r __setstate__zDispatcher.__setstate__Js/fI wZ !$**- rcd|jg}|jr|j|jg}|jdddD]}|j|}|j rRdt |d}|dt|zdzz }||j jz }|j|p|jt ||r#|jddj|zd j|S) NzMultiply dispatched method: rz Inputs: - zOther signatures: z  ) r!r;rqrr8rrrpstriprn)r=docsotherr1rBss rrzDispatcher.__doc__Ps.tyyk:; 88 KK !==2& 1C::c?D|| c 2337S3q6\D((T\\'')) A ]3/0 1  KK1HMM%4HH I{{4  rcP|jtt|jSr5)rrrhr)r=rs r_helpzDispatcher._helpgst}}c$o.666rc4t|j|y)z: Print docstring for the function corresponding to inputs N)printrr=rrCs rhelpzDispatcher.helpjs jdjj$ rcl|jtt|}|s tdt |S)NzNo function found)rrrhror)r=rrBs r_sourcezDispatcher._sourcens1t}}c$o./0 0d|rc4t|j|y)z< Print source code for the function corresponding to inputs N)rrrs rrzDispatcher.sourcets ldllD!"rr5)rrrr __slots__r>rE classmethodrNrarApropertyrrryrr__repr__rrr FutureWarningrrrrrrrrrrrr^s"JI4++ ##&=~"" $2i4+H4 !Cm\%]%% !!,7! #rrchdtj|d}|tj|z}|S)NzFile: r)rI getsourcefile getsource)rBrs rrrys6 &&t,-T2A G  d ##A Hrc0eZdZdZdZedZdZdZy)rzO Dispatch methods based on type signature See Also: Dispatcher objrMcttdrDtj|}tj|j j ddSy)NrGr)rHrIrGitlislicerJrKrLs rrNz MethodDispatcher.get_func_paramssA 7K (##D)C::cnn335q$? ? )rc"||_||_|Sr5r)r=instanceowners r__get__zMethodDispatcher.__get__s rct|Dcgc] }t|c}}|j|}|s%td|jdt |d||j g|i|Scc}w)Nrrr)r^rhrrr!rr)r=rrCrr.rBs rrzMethodDispatcher.__call__suD1StCy12t}}e$%(Edii[PSTabgThSiij&kl lDHH.t.v.. 2sA/N) rrrrrrrNrrrrrrrs,I@@  /rrc2djd|DS)zb String representation of type signature >>> str_signature((int, float)) 'int, float' rfc34K|]}|jywr5)r)rTrMs rrWz str_signature..s1cS\\1rZ)rn)r1s rrrs 991S1 11rc d|d}|dz }|D]#}|ddjd|Dzdzz }%|dz }|dj|Dcgc] }d tt|zd |d z"c}z }|Scc}w) z! The text for ambiguity warnings z* Ambiguities exist in dispatched function rz;The following signatures may result in ambiguous behavior:  rfc3>K|]}dt|zdzyw)[]N)r)rTrs rrWzwarning_text..sE1# a 0036Esrz, Consider making the following additions: z @dispatch(z) def z(...))rnrr)r!r}textpairrs rrrs 8d CD JJDM  EEEFHLM MM >>DFKK:=?56& oa6H(II$TF%01? @@D K?s %A9 )warningsrrItyping_extensionsrconflictrrrr utilsr variadicr r itertoolsr__all__rr rrrrrrrrrrrrrrrs(NN * v8/8 G E @ @ A  #1?  ?$NG X#X#v /z/42 r