9jQ&UdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z e rddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZe ej0dd+dZe ej2d d,dd d-dZe ej0j4d.dZee_e ej8dd/dZe ej:dd0dZe ej<dd1dZe ej>dd1dZe d2dZ e d3dZ e d4dZ e d5dZ e ej@dd6dZ e ejBdd7dZ!e ejDdd8d Z"e ejFdd9d!Z#e ejHd:d;d"Z$ed#d$ dd'Z%e d?d(Z%ed#d$ d@d)Z%e ejJddd$ dAd*Z%y)Bz Python polyfills for itertools ) annotationsN)Callable)overload TYPE_CHECKING TypeAliasTypeVar)substitute_in_graph)IterableIterator) accumulatechainchain_from_iterablecompresscycle dropwhile filterfalseislicepairwisestarmap takewhiletee zip_longest_T_Ur _Predicate_T1_T2T)is_embedded_typec'2K|D] }|Ed{y7wN) iterablesiterables a/media/conek/DATA/Code/OCR/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]returnr3)iteroperatoradd)r$r0r'r-r1s `` r%r r 3s.H~H |||  x  c#2K|D] }|Ed{y7wr!r")r$its r%rrOs!  r&c(dt||DS)Nc3,K|] \}}|s |ywr!r").0datumselectors r% zcompress..^s JoeXE Js )zip)data selectorss r%rr\s JT9)= JJr8c6t}dfd }||S)Nc3lKg}D]}||j||r|D]}||ryywr!)append)r-savedr/r$s r%_cyclezcycle.._cyclefsJ "GM LL ! "    s.44r2)r5)r$r-rHs` r%rrbsH~H  ( r8c#Kt|s"tdt|jdt |}|D]}||r |n|Ed{y7wN'z' object is not callable)callable TypeErrortype__name__r5) predicater$r-xs r%rrusb I !DO4455MNOOH~H |G  sAA AAAc#Kt|s"tdt|jd|D]}||sy|ywrJrLrMrNrO)rPr$rQs r%rrsN I !DO4455MNOO | sAAcyr!r"functionr$s r%rr  r8cyr!r"rUs r%rrrWr8cyr!r"rUs r%rrrWr8cyr!r"rUs r%rrrWr8c#Kt|s"tdt|jd|D] }|| ywrJrS)rVr$argss r%rrsI H !DN3344LMNNos=?cpt|}ttj|Stfd|S)Nc| Sr!r")rQrVs r%zfilterfalse..s HQKr8)r5filterr6not_)rVr$r:s` r%rrs1 hBhmmR((/44r8c'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 enumeraterangemaxrA) r$r\srerfrgnext_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)Nrcrr*)linkvaluer-s r%_teeztee.._teesX 7?"8nDG#TlDG" t     s9%* 6969c3.K|] }ywr!r")r=_rx shared_links r%r@ztee..s5qk"5s)r4r3)r5tuplerj)r$nrxr-r{s @@@r%rrs1H~H,K  5E!H5 55r8.) fillvaluecyr!r")iter1r~s r%rrsr8cyr!r")riter2s r%rrs /2r8cyr!r")rrr~s r%rrs+.r8cyr!r")rriter3r#s r%rr"s'*r8cyr!r")rrrr~r#s r%rr-s%(r8c'@Kttt|}t|}|sy g}t |D]"\}} t |}|j|$t|@#t $r'|dz}|sYytj|||<|}YPwxYww)Nrc) listmapr5lenrir+r, itertoolsrepeatrFr|)r~r# iterators num_activevaluesrnr-rws r%rr:sSy)*IYJ  $Y/ !KAx "X MM%  !Fm  ! "a !(// : ! !  "s4BBBB)r# Iterable[_T]r4r3r!)r$rr0zCallable[[_T, _T], _T] | Noner'z _T | Noner4r3)r$zIterable[Iterable[_T]]r4r3)rBrrCz Iterable[_U]r4r3)r$rr4r3)rP_Predicate[_T]r$rr4r3)rVzCallable[[], _U]r$zIterable[tuple[()]]r4itertools.starmap[_U])rVzCallable[[_T], _U]r$zIterable[tuple[_T]]r4r)rVzCallable[[_T, _T1], _U]r$zIterable[tuple[_T, _T1]]r4r)rVzCallable[[_T, _T1, _T2], _U]r$zIterable[tuple[_T, _T1, _T2]]r4r)rVzCallable[..., _T]r$r r4r)rVrr$rr4r3)r$rr\z int | Noner4r3)r$rr4zIterator[tuple[_T, _T]])r )r$rr}intr4ztuple[Iterator[_T], ...])r Iterable[_T1]r~rr4zIterator[tuple[_T1]])rrr Iterable[_T2]r4z'Iterator[tuple[_T1 | None, _T2 | None]])rrrrr~rr4z#Iterator[tuple[_T1 | _U, _T2 | _U]]) rrrrrrr#rr4zIterator[tuple[_T | None, ...]]) rrrrrrr#rr~rr4Iterator[tuple[_T | _U, ...]])r#rr~rr4r)&__doc__ __future__rrr6collections.abcrtypingrrrr decoratorsr r r __all__rrobjectr__annotations__rrrr from_iterablerrrrrrrrrrrr"r8r%rsb#$>>,2 " T] T] "v. I. en enY__t<= Y))DA+/! !! '! !  !B!6Y__2234*Y''$?K@K Y__t<=$Y((4@ A Y((4@A   !         !       % &       * +     Y&&>?Y**TB5C5Y%%=>8Y''$?@Y]]#6$6$        2 2 2- 2 2  . . .  . ) . . * * * *  * % * *  ( ( ( (  (  (#( (Y**TB#Cr8