%Vji .UdZddlmZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z ddlmZe rddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZeejddHdZeejd dIdddJdZeejjdKdZee_eejddLd#Zeej ddMd$Z eej!ddNd'Z!eej"ddOd)Z"eej#ddPd,Z#ej$d-kr"eej%ddQd/Z%ed0gz Zeej&dRdSd4Z&e d5d6dTd;Z'e dUd?Z'e d5d6dVdAZ'e dWdDZ'e d5d6dXdFZ'eej'ddd6dYdGZ'dS)Zz 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_type iterables Iterable[_T]return Iterator[_T]c'$K|D] }|Ed{V dSN)riterables [/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/_dynamo/polyfills/itertools.pyrr*s:)initialr%func Optional[Callable[[_T, _T], _T]]r( Optional[_T]cdt|} 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-r4s `` r&rr1sRH~~H ||        ;x  r'Iterable[Iterable[_T]]c#$K|D] }|Ed{V dSr#r$)r%its r&rrMs4  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&rrZs J JT9)=)= J J JJr'cDt}dfd }||S)Nr-r!r c3jKg}D]}|V|||r |D]}|V| dSdSr#)append)r-savedr3r%s r&_cyclezcycle.._cycledsy " "GMMM LL ! ! ! !          r'r5r6)r%r-rJs` r&rr`s>H~~H 6(  r' predicate_Predicate[_T]c#dKt|}|D]}||s|Vn|Ed{VdSr#rK)rLr%r-xs r&rrrseH~~H y|| GGG E r'functionct|}ttj|Stfd|S)Nc| Sr#r$)rOrPs r&zfilterfalse..sHHQKKr')r6filterr7not_)rPr%r;s` r&rrs@ hBhmR(((////444r'args 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 enumeraterangemaxrE) r%rVsr[r\r]next_iir3indicess 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&pairwisermsL   A d AA   r'rmninttuple[Iterator[_T], ...]ct|ddgdfd tfdt|DS)Nr r!c3K |dt|d<ddg|d<|\}}|V+#t$rYdSwxYw)NTrYrr/)linkvaluer-s r&_teeztee.._teesq  7?"8nnDG#TlDG" t       FF s ,1 ??c3.K|]}VdSr#r$)rA_ru shared_links r&rDztee..s-55qk""555555r')r r!)r6tupler`)r%rnrur-rxs @@@r&rrseH~~H,K       55555E!HH555 5 55r'.) fillvalueiter1 Iterable[_T1]rzIterator[tuple[_T1]]cdSr#r$)r{rzs r&rrs 3r'iter2 Iterable[_T2]'Iterator[tuple[_T1 | None, _T2 | None]]cdSr#r$)r{rs r&rrs /2cr'#Iterator[tuple[_T1 | _U, _T2 | _U]]cdSr#r$)r{rrzs r&rrs +.#r'iter3Iterator[tuple[_T | None, ...]]cdSr#r$)r{rrrs r&rrs '*cr'Iterator[tuple[_T | _U, ...]]cdSr#r$)r{rrrzrs r&rrs %(Cr'c'xKttt|}t|}|sdS g}t |D]_\}} t |}n4#t $r'|dz}|sYdStj|||<|}YnwxYw| |`t|V)NTrY) listmapr6lenr_r0r1 itertoolsrepeatrHry)rzr iterators num_activevaluesrdr-rts r&rrsSy))**IYJ  $Y// ! !KAx "X  " " "a !FFF(/ :: ! !  " MM% Fmm sA  B5BB)rrr r!r#)r%rr)r*r(r+r r!)r%r9r r!)r<rr=r>r r!)r%rr r!)rLrMr%rr r!)rPrMr%rr r!)r%rrVrWr r!)r%rr rh)r )r%rrnror rp)r{r|rzrr r})r{r|rrr r)r{r|rrrzrr r) r{rrrrrrrr r) r{rrrrrrrrzrr r)rrrzrr r)(__doc__ __future__rrr7systypingrrrrrtyping_extensionsr decoratorsr collections.abcr r __all__rrobjectr__annotations__rrrr from_iterablerrrrrr version_informrrr$r'r&rs~#""""" GGGGGGGGGGGGGG'''''',,,,,,322222222    WT]] WT]] "v. .... genn gennY_t<<<=< Y)DAAA.2!! !!!!!BA!6Y_23343*Y'$???KKK@?K Y_t<<<   =< "Y(4@@@   A@ Y*TBBB555CB5Y%===>=8w+dCCCDC |GY]##6666$#6$    222 2  ..... . *** *  ((((( (Y*TBBBCBr'