9i|UdZddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl mZe rddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZeejddFdZeejd dGdddHdZeejjdIdZee_eejddJd#ZeejddKd$ZeejddLd'Zeej ddMd)Z eej!ddNd,Z!eej"ddOd.Z"eej#dPdQd2Z#e d3d4dRd9Z$e dSd=Z$e d3d4dTd?Z$e dUdBZ$e d3d4dVdDZ$eej$ddd4dWdEZ$dS)Xz Python polyfills for itertools ) annotationsN)Callable)Optionaloverload TYPE_CHECKING TypeAliasTypeVar)substitute_in_graph)IterableIterator) accumulatechainchain_from_iterablecompresscycle dropwhile filterfalseislicetee zip_longestpairwise_T_Ur _Predicate_T1_T2T)is_embedded_type iterables Iterable[_T]return Iterator[_T]c'$K|D] }|Ed{V dSN)riterables T/var/www/icac/venv/lib/python3.11/site-packages/torch/_dynamo/polyfills/itertools.pyrr*s:)initialr&func Optional[Callable[[_T, _T], _T]]r) Optional[_T]cdt|} tjdfd }||S)Niteratorr"r!c3K}|" t|}n#t$rYdSwxYw|V|D]}||}|VdSr$next StopIteration)r.totalelementr*r)s r' _accumulatezaccumulate.._accumulate=s = X        GD((EKKKK  s  ''r.r"r!r")iteroperatoradd)r&r*r)r.r5s `` r'rr1sRH~~H ||        ;x  r(Iterable[Iterable[_T]]c#$K|D] }|Ed{V dSr$r%)r&its r'rrMs4  r(data selectors Iterable[_U]c6dt||DS)Nc3$K|] \}}||V dSr$r%).0datumselectors r' zcompress..\s+ J JoeX JE J J J J J Jr()zip)r=r>s r'rrZs J JT9)=)= J J JJr(cDt}dfd }||S)Nr.r"r!c3jKg}D]}|V|||r |D]}|V| dSdSr$)append)r.savedr4r&s r'_cyclezcycle.._cycledsy " "GMMM LL ! ! ! !          r(r6r7)r&r.rKs` r'rr`s>H~~H 6(  r( predicate_Predicate[_T]c#dKt|}|D]}||s|Vn|Ed{VdSr$rL)rMr&r.xs r'rrrseH~~H y|| GGG E r(functionct|}ttj|Stfd|S)Nc| Sr$r%)rPrQs r'zfilterfalse..sHHQKKr()r7filterr8not_)rQr&r<s` r'rrs@ hBhmR(((////444r(args int | Nonec'Kt|}|jdn|j}|j}|jdn|j}|dks||dks|dkrt d|(|}t |D]\}}||kr |V||z }dSt t||} |}t| |D]\}}||kr |V||z }dS)NrzGIndices for islice() must be None or an integer: 0 <= x <= sys.maxsize.) slicestartstopstep ValueError enumeraterangemaxrF) r&rWsr\r]r^next_iir4indicess r'rrs t AAAagE 6D11AFD qyyT%$((tqyy U    |#H--  JAwF{{ $   E4(())gx00  JAwF{{ $  r(Iterator[tuple[_T, _T]]c#6Kd}d}|D]}|rd}n||fV|}dS)NTFr%)r&afirstbs r'rrsL A E   EEQ$JJJ  r(ninttuple[Iterator[_T], ...]ct|ddgdfd tfdt|DS)Nr!r"c3K |dt|d<ddg|d<|\}}|V+#t$rYdSwxYw)NTrZrr0)linkvaluer.s r'_teeztee.._teesq  7?"8nnDG#TlDG" t       FF s ,1 ??c3.K|]}VdSr$r%)rB_rs shared_links r'rEztee..s-55qk""555555r()r!r")r7tuplera)r&rlrsr.rvs @@@r'rrseH~~H,K       55555E!HH555 5 55r(.) fillvalueiter1 Iterable[_T1]rxIterator[tuple[_T1]]cdSr$r%)ryrxs r'rrs 3r(iter2 Iterable[_T2]'Iterator[tuple[_T1 | None, _T2 | None]]cdSr$r%)ryr}s r'rrs /2cr(#Iterator[tuple[_T1 | _U, _T2 | _U]]cdSr$r%)ryr}rxs r'rrs +.#r(iter3Iterator[tuple[_T | None, ...]]cdSr$r%)ryr}rrs r'rrs '*cr(Iterator[tuple[_T | _U, ...]]cdSr$r%)ryr}rrxrs r'rrs %(Cr(c'xKttt|}t|}|sdS g}t |D]_\}} t |}n4#t $r'|dz}|sYdStj|||<|}YnwxYw| |`t|V)NTrZ) listmapr7lenr`r1r2 itertoolsrepeatrIrw)rxr iterators num_activevaluesrer.rrs r'rrsSy))**IYJ  $Y// ! !KAx "X  " " "a !FFF(/ :: ! !  " MM% Fmm sA  B5BB)rr r!r"r$)r&r r*r+r)r,r!r")r&r:r!r")r=r r>r?r!r")r&r r!r")rMrNr&r r!r")rQrNr&r r!r")r&r rWrXr!r")r&r r!rg)r )r&r rlrmr!rn)ryrzrxrr!r{)ryrzr}r~r!r)ryrzr}r~rxrr!r) ryr r}r rr rr r!r) ryr r}r rr rr rxrr!r)rr rxrr!r)%__doc__ __future__rrr8collections.abcrtypingrrrrr decoratorsr r r __all__rrobjectr__annotations__rrrr from_iterablerrrrrrrrrr%r(r'rsY#"""""$$$$$$HHHHHHHHHHHHHH,,,,,,322222222    WT]] WT]] "v. .... genn gennY_t<<<=< Y)DAAA.2!! !!!!!BA!6Y_23343*Y'$???KKK@?K Y_t<<<   =< "Y(4@@@   A@ Y*TBBB555CB5Y%===>=8Y'$???@?Y]##6666$#6$    222 2  ..... . *** *  ((((( (Y*TBBBCBr(