i| UdZddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl mZe rddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZeej2dd%dZeej4d d&dd d'dZeej2j6d(dZee_eej:dd)dZeej<dd*dZeej>dd+dZeej@dd,dZ eejBdd-dZ!eejDdd.dZ"eejFd/d0dZ#e dd d1dZ$e d2d Z$e dd d3d!Z$e d4d"Z$e dd d5d#Z$eejHddd d6d$Z$y)7z 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_typec'2K|D] }|Ed{y7wN) iterablesiterables [/var/www/html/engine/venv/lib/python3.12/site-packages/torch/_dynamo/polyfills/itertools.pyrr*s! )initialc^t|}tjdfd }||S)Nc3|K}| t|}||D]}||}|y#t$rYywxYwwr next StopIteration)iteratortotalelementfuncr&s r$ _accumulatezaccumulate.._accumulate=sY = X  G(EK  !  s< -< 9<9<r, Iterator[_T]returnr2)iteroperatoradd)r#r/r&r,r0s `` r$rr1s.H~H |||  x  c#2K|D] }|Ed{y7wr r!)r#its r$rrMs!  r%c(dt||DS)Nc3,K|] \}}|s |ywr r!).0datumselectors r$ zcompress..\s JoeXE Js )zip)data selectorss r$rrZs JT9)= JJr7c6t}dfd }||S)Nc3lKg}D]}||j||r|D]}||ryywr )append)r,savedr.r#s r$_cyclezcycle.._cycledsJ "GM LL ! "    s.44r1r4)r#r,rGs` r$rr`sH~H ( r7c#dKt|}|D]}||r |n|Ed{y7wr rH) predicater#r,xs r$rrrs:H~H |G  s0 0.0cpt|}ttj|Stfd|S)Nc| Sr r!)rKfunctions r$zfilterfalse..s HQKr7)r4filterr5not_)rNr#r9s` r$rrs1 hBhmmR((/44r7c'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@) r#argssrUrVrWnext_iir.indicess r$rrs t AAaggE 66D1AFFD qyT%$(tqy U   |#H- JAwF{ $  E4()gx0 JAwF{ $ sA?C :C = C c#<Kd}d}|D]}|rd}n||f|}yw)NTFr!)r#afirstbs r$rrs5 A E  EQ$J  scnt|ddgdfd tfdt|DS)Nc3vK |dt|d<ddg|d<|\}}|$#t$rYywxYww)NrSrr))linkvaluer,s r$_teeztee.._teesX 7?"8nDG#TlDG" t     s9%* 6969c3.K|] }ywr r!)r<_ri shared_links r$r?ztee..s5qk"5s)r3r2)r4tuplerZ)r#nrir,rls @@@r$rrs1H~H,K  5E!H5 55r7.) fillvaluecyr r!)iter1ros r$rrsr7cyr r!)rqiter2s r$rrs /2r7cyr r!)rqrsros r$rrs+.r7cyr r!)rqrsiter3r"s r$rrs'*r7cyr r!)rqrsrvror"s r$rrs%(r7c'@Kttt|}t|}|sy g}t |D]"\}} t |}|j|$t|@#t $r'|dz}|sYytj|||<|}YPwxYww)NrS) listmapr4lenrYr*r+ itertoolsrepeatrErm)ror" iterators num_activevaluesr_r,rhs r$rrsSy)*IYJ  $Y/ !KAx "X MM%  !Fm  ! "a !(// : ! !  "s4BBBB)r" Iterable[_T]r3r2r )r#rr/z Optional[Callable[[_T, _T], _T]]r&z Optional[_T]r3r2)r#zIterable[Iterable[_T]]r3r2)rArrBz Iterable[_U]r3r2)r#rr3r2)rJ_Predicate[_T]r#rr3r2)rNrr#rr3r2)r#rr\z int | Noner3r2)r#rr3zIterator[tuple[_T, _T]])r )r#rrnintr3ztuple[Iterator[_T], ...])rq Iterable[_T1]rorr3zIterator[tuple[_T1]])rqrrs Iterable[_T2]r3z'Iterator[tuple[_T1 | None, _T2 | None]])rqrrsrrorr3z#Iterator[tuple[_T1 | _U, _T2 | _U]]) rqrrsrrvrr"rr3zIterator[tuple[_T | None, ...]]) rqrrsrrvrr"rrorr3Iterator[tuple[_T | _U, ...]])r"rrorr3r)%__doc__ __future__rr|r5collections.abcrtypingrrrrr decoratorsr r r __all__rrobjectr__annotations__rrrr from_iterablerrrrrrrrrr!r7r$rsg#$HH,2  T] T] "v. I. en enY__t<= Y))DA.2!! !! *! !  !B!6Y__2234*Y''$?K@K Y__t< = "Y((4@ A Y**TB5C5Y%%=>8Y''$?@Y]]#6$6$        2 2 2- 2 2  . . .  . ) . . * * * *  * % * *  ( ( ( (  (  (#( (Y**TB#Cr7