o Zh[@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)Zis_embedded_type iterables Iterable[_T]return Iterator[_T]cgs|D]}|EdHqdSN)riterablerrP/var/www/auris/lib/python3.10/site-packages/torch/_dynamo/polyfills/itertools.pyr &s r rIterable[Iterable[_T]]cCs tj|Sr) itertoolsr )rrrrr ,s r data selectors Iterable[_U]cCsddt||DS)Ncss|] \}}|r|VqdSrr).0Zdatumselectorrrr 7szcompress..)zip)r"r#rrrr5sr predicate_Predicate[_T]ccs4t|}|D] }||s|Vnq|EdHdSr)iter)r)riteratorxrrrr;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() rr.sr2r3r4Znext_iielementindicesrrrrIs2 r) Iterator[tuple[_T, _T]]ccs0d}d}|D]}|rd}n||fV|}qdS)NTFr)rafirstbrrrpairwisehs rCninttuple[Iterator[_T], ...]cs:t|ddgdfdd tfddt|DS)Nrrc3sNz |ddurt|d<ddg|d<|\}}|Vqty&YdSw)NTr0r)next StopIteration)linkvalue)r,rr_tee|s    ztee.._teec3s|]}VqdSrr)r%_)rK shared_linkrrr'sztee..)rr)r+tupler7)rrDr)rKr,rMrrws r.) fillvalueiter1 Iterable[_T1]rOIterator[tuple[_T1]]cCdSrr)rPrOrrrrsriter2 Iterable[_T2]'Iterator[tuple[_T1 | None, _T2 | None]]cCrSrr)rPrTrrrrs#Iterator[tuple[_T1 | _U, _T2 | _U]]cCrSrr)rPrTrOrrrriter3Iterator[tuple[_T | None, ...]]cGrSrr)rPrTrYrrrrrrXIterator[tuple[_T | _U, ...]]cGrSrr)rPrTrYrOrrrrrsc gsttt|}t|}|sdS g}t|D],\}}zt|}Wnty=|d8}|s2YdSt|||<|}Ynw| |qt |Vq)NTr0) listmapr+lenr6rGrHr!repeatappendrN)rOr iteratorsZ num_activevaluesr:r,rJrrrrs(    )rrrr)rr rr)r"rr#r$rr)r)r*rrrr)rrr.r/rr)rrrr?)r)rrrDrErrF)rPrQrOrrrR)rPrQrTrUrrV)rPrQrTrUrOrrrW) rPrrTrrYrrrrrZ) rPrrTrrYrrrrOrrr[)rrrOrrr[)#__doc__ __future__rr!systypingrrrrZtyping_extensionsrZ decoratorsr collections.abcr r __all__rrobjectr__annotations__rrr from_iterabler rrr version_inforCrrrrrrs`