BjQ&UdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z e rddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZe ejddWdZe ejd dXdddYdZe ejjdZdZee_e ejdd[d#Ze ejdd\d$Ze ejdd]d'Ze ejdd]d(Zed^d-Z ed_d0Z ed`d3Z edad6Z e ej ddbd9Z e ej!ddcd:Z!e ej"dddd=Z"e ej#dded?Z#e ej$dfdgdCZ$edDdEdhdJZ%edidNZ%edDdEdjdPZ%edkdSZ%edDdEdldUZ%e ej%dddEdmdVZ%dS)nz 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_type iterables Iterable[_T]return Iterator[_T]c'$K|D] }|Ed{V dSN)r iterables d/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/torch/_dynamo/polyfills/itertools.pyrr,s:)initialr'funcCallable[[_T, _T], _T] | Noner* _T | Nonecdt|} 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/r6s `` r(r r 3sRH~~H ||        ;x  r)Iterable[Iterable[_T]]c#$K|D] }|Ed{V dSr%r&)r'its r(rrOs4  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(rr\s J JT9)=)= J J JJr)cDt}dfd }||S)Nr/r#r"c3jKg}D]}|V|||r |D]}|V| dSdSr%)append)r/savedr5r's r(_cyclezcycle.._cyclefsy " "GMMM LL ! ! ! !          r)r7)r8)r'r/rLs` r(rrbs>H~~H       6(  r) predicate_Predicate[_T]c#Kt|s%tdt|jdt |}|D]}||s|Vn|Ed{VdSN'z' object is not callable)callable TypeErrortype__name__r8)rMr'r/xs r(rrus I  PNDOO4NNNOOOH~~H y|| GGG E r)c#Kt|s%tdt|jd|D]}||sdS|VdSrPrRrSrTrU)rMr'rVs r(rrsw I  PNDOO4NNNOOO y||  EEr)functionCallable[[], _U]Iterable[tuple[()]]itertools.starmap[_U]cdSr%r&rYr's r(rr  Cr)Callable[[_T], _U]Iterable[tuple[_T]]cdSr%r&r^s r(rrr_r)Callable[[_T, _T1], _U]Iterable[tuple[_T, _T1]]cdSr%r&r^s r(rrr_r)Callable[[_T, _T1, _T2], _U]Iterable[tuple[_T, _T1, _T2]]cdSr%r&r^s r(rrr_r)Callable[..., _T]r c#Kt|s%tdt|jd|D] }||V dSrPrX)rYr'argss r(rrsj H  OMDNN3MMMNNNhor)ct|}ttj|Stfd|S)Nc| Sr%r&)rVrYs r(zfilterfalse..sHHQKKr))r8filterr9not_)rYr'r=s` r(rrs@ hBhmR(((////444r)rk 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 enumeraterangemaxrG) r'rksrurvrwnext_iir5indicess 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)NTrsrr1)linkvaluer/s r(_teeztee.._teesq  7?"8nnDG#TlDG" t       FF s ,1 ??c3.K|]}VdSr%r&)rC_r shared_links r(rFztee..s-55qk""555555r))r"r#)r8tuplerz)r'rrr/rs @@@r(rrseH~~H,K       55555E!HH555 5 55r).) fillvalueiter1 Iterable[_T1]rIterator[tuple[_T1]]cdSr%r&)rrs r(rrs 3r)iter2 Iterable[_T2]'Iterator[tuple[_T1 | None, _T2 | None]]cdSr%r&)rrs r(rrs /2cr)#Iterator[tuple[_T1 | _U, _T2 | _U]]cdSr%r&)rrrs r(rrs +.#r)iter3Iterator[tuple[_T | None, ...]]cdSr%r&)rrrr s r(rr"s '*cr)Iterator[tuple[_T | _U, ...]]cdSr%r&)rrrrr s r(rr-s %(Cr)c'xKttt|}t|}|sdS g}t |D]_\}} t |}n4#t $r'|dz}|sYdStj|||<|}YnwxYw| |`t|V)NTrs) listmapr8lenryr2r3 itertoolsrepeatrJr)rr iterators num_activevaluesr~r/rs r(rr:sSy))**IYJ  $Y// ! !KAx "X  " " "a !FFF(/ :: ! !  " MM% Fmm sA  B5BB)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!r"r#)rMrNr'r!r"r#)rYrZr'r[r"r\)rYr`r'rar"r\)rYrcr'rdr"r\)rYrfr'rgr"r\)rYrir'r r"r!)rYrNr'r!r"r#)r'r!rkrqr"r#)r'r!r"r)r )r'r!rrr"r)rrrrr"r)rrrrr"r)rrrrrrr"r) rr!rr!rr!r r!r"r) rr!rr!rr!r r!rrr"r)r r!rrr"r)&__doc__ __future__rrr9collections.abcrtypingrrrr decoratorsr r r __all__rrobjectr__annotations__rrrr from_iterablerrrrrrrrrrrr&r)r(rs<#"""""$$$$$$>>>>>>>>>>>>,,,,,,322222222   " WT]] WT]] "v. .... genn gennY_t<<<=< Y)DAAA+/! !!!!!BA!6Y_23343*Y'$???KKK@?K Y_t<<<=<$Y(4@@@   A@ Y(4@@@A@                        Y&>>>?>Y*TBBB555CB5Y%===>=8Y'$???@?Y]##6666$#6$    222 2  ..... . *** *  ((((( (Y*TBBBCBr)