i RUdZddlmZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z ddlmZe rddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZeej6dd'dZeej8d d(dd d)dZeej6j:d*dZee_eej>dd+dZeej@dd,dZ eejBdd-dZ!eejDdd.dZ"eejFdd/dZ#ejHdk\r!eejJdd0dZ%edgz ZeejLd1d2dZ&e dd  d3d!Z'e d4d"Z'e dd  d5d#Z'e d6d$Z'e dd  d7d%Z'eejNddd  d8d&Z'y)9z Python polyfills for itertools ) annotationsN)CallableOptionaloverload TYPE_CHECKINGTypeVar) TypeAlias)substitute_in_graph)IterableIterator) accumulatechainchain_from_iterablecompresscycle dropwhile filterfalseislicetee zip_longest_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$rYywxYwwrnext StopIteration)iteratortotalelementfuncr%s r# _accumulatezaccumulate.._accumulate=sY = X  G(EK  !  s< -< 9<9<r+ Iterator[_T]returnr1)iteroperatoradd)r"r.r%r+r/s `` r#rr1s.H~H |||  x  c#2K|D] }|Ed{y7wrr )r"its r#rrMs!  r$c(dt||DS)Nc3,K|] \}}|s |ywrr ).0datumselectors r# zcompress..\s JoeXE Js )zip)data selectorss r#rrZs JT9)= JJr6c6t}dfd }||S)Nc3lKg}D]}||j||r|D]}||ryywr)append)r+savedr-r"s r#_cyclezcycle.._cycledsJ "GM LL ! "    s.44r0r3)r"r+rFs` r#rr`sH~H ( r6c#dKt|}|D]}||r |n|Ed{y7wrrG) predicater"r+xs r#rrrs:H~H |G  s0 0.0cpt|}ttj|Stfd|S)Nc| Srr )rJfunctions r#zfilterfalse..s HQKr6)r3filterr4not_)rMr"r8s` r#rrs1 hBhmmR((/44r6c'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"argssrTrUrVnext_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#pairwiserfs5  Ad A  srfcnt|ddgdfd tfdt|DS)Nc3vK |dt|d<ddg|d<|\}}|$#t$rYywxYww)NrRrr()linkvaluer+s r#_teeztee.._teesX 7?"8nDG#TlDG" t     s9%* 6969c3.K|] }ywrr )r;_rk shared_links r#r>ztee..s5qk"5s)r2r1)r3tuplerY)r"nrkr+rns @@@r#rrs1H~H,K  5E!H5 55r6.) fillvaluecyrr )iter1rqs r#rrs r6cyrr )rsiter2s r#rrs /2r6cyrr )rsrurqs r#rrs+.r6cyrr )rsruiter3r!s r#rrs'*r6cyrr )rsrurxrqr!s r#rrs%(r6c'@Kttt|}t|}|sy g}t |D]"\}} t |}|j|$t|@#t $r'|dz}|sYytj|||<|}YPwxYww)NrR) listmapr3lenrXr)r* itertoolsrepeatrDro)rqr! iterators num_activevaluesr^r+rjs r#rrsSy)*IYJ  $Y/ !KAx "X MM%  !Fm  ! "a !(// : ! !  "s4BBBB)r! Iterable[_T]r2r1r)r"rr.z Optional[Callable[[_T, _T], _T]]r%z Optional[_T]r2r1)r"zIterable[Iterable[_T]]r2r1)r@rrAz Iterable[_U]r2r1)r"rr2r1)rI_Predicate[_T]r"rr2r1)rMrr"rr2r1)r"rr[z int | Noner2r1)r"rr2zIterator[tuple[_T, _T]])r )r"rrpintr2ztuple[Iterator[_T], ...])rs Iterable[_T1]rqrr2zIterator[tuple[_T1]])rsrru Iterable[_T2]r2z'Iterator[tuple[_T1 | None, _T2 | None]])rsrrurrqrr2z#Iterator[tuple[_T1 | _U, _T2 | _U]]) rsrrurrxrr!rr2zIterator[tuple[_T | None, ...]]) rsrrurrxrr!rrqrr2Iterator[tuple[_T | _U, ...]])r!rrqrr2r)(__doc__ __future__rr~r4systypingrrrrrtyping_extensionsr decoratorsr collections.abcr r __all__rrobjectr__annotations__rrrr from_iterablerrrrrr version_inforfrrr r6r#rs# GG',2  T] T] "v. I. en enY__t<= Y))DA.2!! !! *! !  !B!6Y__2234*Y''$?K@K Y__t< = "Y((4@ A Y**TB5C5Y%%=>8w++dCD |GY]]#6$6$        2 2 2- 2 2  . . .  . ) . . * * * *  * % * *  ( ( ( (  (  (#( (Y**TB#Cr6