Vh[UdZddlmZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z erddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZe ej2dd#dZe ej2j4d$dZee_e ej8dd%dZe ej:dd&dZe ej<dd'dZej>dk\r!e ej@dd(dZ edgz Ze ejBd)d*dZ!edd d+dZ"e d,dZ"edd d-dZ"e d.d Z"edd d/d!Z"e ejDddd d0d"Z"y)1z Python polyfills for itertools ) annotationsN)Callableoverload TYPE_CHECKINGTypeVar) TypeAlias)substitute_in_graph)IterableIterator)chainchain_from_iterablecompress dropwhileislicetee zip_longest_T_Ur _Predicate_T1_T2T)is_embedded_typec'2K|D] }|Ed{y7wN) iterablesiterables Q/home/dcms/DCMS/lib/python3.12/site-packages/torch/_dynamo/polyfills/itertools.pyr r &s!s c&tj|Sr) itertoolsr )rs rrr,s ??H %%c(dt||DS)Nc3,K|] \}}|s |ywrr).0datumselectors r zcompress..7s JoeXE Js )zip)data selectorss rrr5s JT9)= JJr"c#dKt|}|D]}||r |n|Ed{y7wr)iter) predicateriteratorxs rrr;s:H~H |G  s0 0.0c'Kt|}|jdn |j}|j}|jdn |j}|dks ||dks|dkr t d|%|}t |D]\}}||k(s |||z }yt t||} |}t| |D]\}}||k(s |||z }yw)NrzGIndices for islice() must be None or an integer: 0 <= x <= sys.maxsize.) slicestartstopstep ValueError enumeraterangemaxr)) rargssr4r5r6next_iielementindicess rrrIs t AAaggE 66D1AFFD qyT%$(tqy U   |#H- JAwF{ $  E4()gx0 JAwF{ $ sA?C :C = C ) c#<Kd}d}|D]}|rd}n||f|}yw)NTFr)rafirstbs rpairwiserGhs5  Ad A  srGcnt|ddgdfd tfdt|DS)Nc3vK |dt|d<ddg|d<|\}}|$#t$rYywxYww)Nr2r)next StopIteration)linkvaluer/s r_teeztee.._tee|sX 7?"8nDG#TlDG" t     s9%* 6969c3.K|] }ywrr)r%_rN shared_links rr(ztee..s5qk"5s)return Iterator[_T])r-tupler9)rnrNr/rQs @@@rrrws1H~H,K  5E!H5 55r".) fillvaluecyrr)iter1rVs rrrs r"cyrr)rXiter2s rrrs /2r"cyrr)rXrZrVs rrrs+.r"cyrr)rXrZiter3rs rrrs'*r"cyrr)rXrZr]rVrs rrrs%(r"c'@Kttt|}t|}|sy g}t |D]"\}} t |}|j|$t|@#t $r'|dz}|sYytj|||<|}YPwxYww)Nr2) listmapr-lenr8rJrKr!repeatappendrT)rVr iterators num_activevaluesr>r/rMs rrrsSy)*IYJ  $Y/ !KAx "X MM%  !Fm  ! "a !(// : ! !  "s4BBBB)r Iterable[_T]rRrS)rzIterable[Iterable[_T]]rRrS)r*rhr+z Iterable[_U]rRrS)r.z_Predicate[_T]rrhrRrS)rrhr;z int | NonerRrS)rrhrRzIterator[tuple[_T, _T]])r )rrhrUintrRztuple[Iterator[_T], ...])rX Iterable[_T1]rVrrRzIterator[tuple[_T1]])rXrjrZ Iterable[_T2]rRz'Iterator[tuple[_T1 | None, _T2 | None]])rXrjrZrkrVrrRz#Iterator[tuple[_T1 | _U, _T2 | _U]]) rXrhrZrhr]rhrrhrRzIterator[tuple[_T | None, ...]]) rXrhrZrhr]rhrrhrVrrRIterator[tuple[_T | _U, ...]])rrhrVrrRrl)#__doc__ __future__rr!systypingrrrrtyping_extensionsr decoratorsr collections.abcr r __all__rrobjectr__annotations__rrr from_iterablerrrr version_inforGrrrr"rrys# ==',2  T] T] "v. I. en enY__t<= Y__223&4&*Y''$?K@K Y((4@ A Y%%=>8w++dCD |GY]]#6$6$        2 2 2- 2 2  . . .  . ) . . * * * *  * % * *  ( ( ( (  (  (#( (Y**TB#Cr"