o ei@sUdZddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl mZe r5ddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZeejdddQddZeejdd dRdddSddZeejjdTd!d"Zee_eejdddUd&d'ZeejdddVd(d)ZeejdddWd,d-Zeej dddXd/d0Z eej!dddYd3d4Z!eej"dddZd6d7Z"eej#d[d\d;d<Z#e d=d>d]dCdDZ$e d^dHdDZ$e d=d>d_dJdDZ$e d`dMdDZ$e d=d>dadOdDZ$eej$dddd>dbdPdDZ$dS)cz 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]cg|D]}|EdHqdSN)riterabler$r$k/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/_dynamo/polyfills/itertools.pyr*s r)initialr%func Optional[Callable[[_T, _T], _T]]r' Optional[_T]cs.t|}dur tjdfdd }||S)Niteratorr!r c3sT}|durzt|}Wn tyYdSw|V|D] }||}|VqdSr#next StopIteration)r+totalelementr(r'r$r& _accumulate=s   zaccumulate.._accumulater+r!r r!)iteroperatoradd)r%r(r'r+r2r$r1r&r 1s  r Iterable[Iterable[_T]]ccr"r#r$)r%itr$r$r&rMs rdata selectors Iterable[_U]cCsddt||DS)Ncss|] \}}|r|VqdSr#r$).0datumselectorr$r$r& \szcompress..)zip)r9r:r$r$r&rZsrcst}dfdd }||S)Nr+r!r c3s@g}D] }|V||q|r|D]}|Vq|sdSdSr#)append)r+savedr0r%r$r&_cycleds  zcycle.._cycler3r4)r%r+rDr$rCr&r`s r predicate_Predicate[_T]ccs4t|}|D] }||s|Vnq|EdHdSr#rE)rFr%r+xr$r$r&rssrfunctioncs.t|}durttj|Stfdd|S)Ncs | Sr#r$)rHrIr$r&s zfilterfalse..)r4filterr5not_)rIr%r8r$rJr&rs rargs int | Nonec gst|}|jdur dn|j}|j}|jdurdn|j}|dks,|dur(|dks,|dkr0td|durL|}t|D]\}}||krI|V||7}q:dStt||} |}t| |D]\}}||kri|V||7}qZdS)NrzGIndices for islice() must be None or an integer: 0 <= x <= sys.maxsize.) slicestartstopstep ValueError enumeraterangemaxr@) r%rNsrRrSrTnext_iir0indicesr$r$r&rs2 rIterator[tuple[_T, _T]]ccs0d}d}|D]}|rd}n||fV|}qdS)NTFr$)r%afirstbr$r$r&rs rninttuple[Iterator[_T], ...]cs:t|ddgdfdd tfddt|DS)Nr r!c3sNz |ddurt|d<ddg|d<|\}}|Vqty&YdSw)NTrPrr,)linkvalue)r+r$r&_tees    ztee.._teec3s|]}VqdSr#r$)r<_)rf shared_linkr$r&r?sztee..)r r!)r4tuplerW)r%rar$)rfr+rhr&rs r.) fillvalueiter1 Iterable[_T1]rjIterator[tuple[_T1]]cCdSr#r$)rkrjr$r$r&rsriter2 Iterable[_T2]'Iterator[tuple[_T1 | None, _T2 | None]]cCrnr#r$)rkror$r$r&rs#Iterator[tuple[_T1 | _U, _T2 | _U]]cCrnr#r$)rkrorjr$r$r&riter3Iterator[tuple[_T | None, ...]]cGrnr#r$)rkrortrr$r$r&rrsIterator[tuple[_T | _U, ...]]cGrnr#r$)rkrortrjrr$r$r&rs c gsttt|}t|}|sdS g}t|D],\}}zt|}Wnty=|d8}|s2YdSt|||<|}Ynw| |qt |Vq)NTrP) listmapr4lenrVr-r. itertoolsrepeatrAri)rjr iterators num_activevaluesr[r+rer$r$r&rs(    )rrr r!r#)r%rr(r)r'r*r r!)r%r7r r!)r9rr:r;r r!)r%rr r!)rFrGr%rr r!)rIrGr%rr r!)r%rrNrOr r!)r%rr r])r )r%rrarbr rc)rkrlrjrr rm)rkrlrorpr rq)rkrlrorprjrr rr) rkrrorrtrrrr ru) rkrrorrtrrrrjrr rv)rrrjrr rv)%__doc__ __future__rrzr5collections.abcrtypingrrrrr decoratorsr r r __all__rrobjectr__annotations__rrrr from_iterablerrrrrrrrrr$r$r$r&sn