o I h[@sUdZddlmZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z er3ddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZe ejdddFddZe ejjdGddZee_e ejdddHdd Ze ejdddId#d$Ze ejdddJd'd(Zejd)kre ej dddKd+d,Z ed,g7Ze ej!dLdMd0d1Z!ed2d3dNd8d9Z"edOd=d9Z"ed2d3dPd?d9Z"edQdBd9Z"ed2d3dRdDd9Z"e ej"dddd3dSdEd9Z"dS)Tz Python polyfills for itertools ) annotationsN)Callableoverload TYPE_CHECKINGTypeVar) TypeAlias)substitute_in_graph)IterableIterator)chainchain_from_iterablecompress dropwhileislicetee zip_longest_T_Ur _Predicate_T1_T2T)is_embedded_type iterables Iterable[_T]return Iterator[_T]cgs|D]}|EdHqdSN)riterablerrU/var/www/vscode/kcb/lib/python3.10/site-packages/torch/_dynamo/polyfills/itertools.pyr &s r rIterable[Iterable[_T]]cCs tj|Sr) itertoolsr )rrrr r ,s r data selectors Iterable[_U]cCsddt||DS)Ncss|] \}}|r|VqdSrr).0datumselectorrrr 7szcompress..)zip)r#r$rrr r5sr predicate_Predicate[_T]ccs4t|}|D] }||s|Vnq|EdHdSr)iter)r+riteratorxrrr r;srargs int | Nonec gst|}|jdur dn|j}|j}|jdurdn|j}|dks,|dur(|dks,|dkr0td|durL|}t|D]\}}||krI|V||7}q:dStt||} |}t| |D]\}}||kri|V||7}qZdS)NrzGIndices for islice() must be None or an integer: 0 <= x <= sys.maxsize.) slicestartstopstep ValueError enumeraterangemaxr*) rr0sr4r5r6next_iielementindicesrrr rIs2 r) Iterator[tuple[_T, _T]]ccs0d}d}|D]}|rd}n||fV|}qdS)NTFr)rafirstbrrr pairwisehs rFninttuple[Iterator[_T], ...]cs:t|ddgdfdd tfddt|DS)Nrrc3sNz |ddurt|d<ddg|d<|\}}|Vqty&YdSw)NTr2r)next StopIteration)linkvalue)r.rr _tee|s    ztee.._teec3s|]}VqdSrr)r&_)rN shared_linkrr r)sztee..)rr)r-tupler9)rrGr)rNr.rPr rws r.) fillvalueiter1 Iterable[_T1]rRIterator[tuple[_T1]]cCdSrr)rSrRrrr rsriter2 Iterable[_T2]'Iterator[tuple[_T1 | None, _T2 | None]]cCrVrr)rSrWrrr rs#Iterator[tuple[_T1 | _U, _T2 | _U]]cCrVrr)rSrWrRrrr riter3Iterator[tuple[_T | None, ...]]cGrVrr)rSrWr\rrrr rr[Iterator[tuple[_T | _U, ...]]cGrVrr)rSrWr\rRrrrr rsc gsttt|}t|}|sdS g}t|D],\}}zt|}Wnty=|d8}|s2YdSt|||<|}Ynw| |qt |Vq)NTr2) listmapr-lenr8rJrKr"repeatappendrQ)rRr iterators num_activevaluesr=r.rMrrr rs(    )rrrr)rr!rr)r#rr$r%rr)r+r,rrrr)rrr0r1rr)rrrrB)r)rrrGrHrrI)rSrTrRrrrU)rSrTrWrXrrY)rSrTrWrXrRrrrZ) rSrrWrr\rrrrr]) rSrrWrr\rrrrRrrr^)rrrRrrr^)#__doc__ __future__rr"systypingrrrrtyping_extensionsr decoratorsr collections.abcr r __all__rrobjectr__annotations__rrr from_iterabler rrr version_inforFrrrrrr s`