a h[ @sLUdZddlmZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z erfddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZe ejdddddddZe ejjdddddZee_e ejddddddddZe ejdddddd d!d"Ze ejdddd#dd$d%d&Zejd'krxe ej dddd(dd)d*Z ed*g7Ze ej!dFdd+d,d-d.d/Z!ed0d1d2d d3d4d5d6Z"ed2d7d8d9d:d6Z"ed0d1d2d7d d;d<d=d6Z"eddddd>d?d@d6Z"ed0d1ddddd dAdBdCd6Z"e ej"dddd1dd dAdDdEd6Z"dS)Gz 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_typez Iterable[_T] Iterator[_T]) iterablesreturncgs|D]}|EdHqdSN)riterablerrO/var/www/auris/lib/python3.9/site-packages/torch/_dynamo/polyfills/itertools.pyr &sr zIterable[Iterable[_T]])rrcCs tj|Sr) itertoolsr )rrrrr ,sr z Iterable[_U])data selectorsrcCsddt||DS)Ncss|]\}}|r|VqdSrr).0Zdatumselectorrrr 7zcompress..)zip)r r!rrrr5srz_Predicate[_T]) predicaterrccs2t|}|D]}||s |Vq$q |EdHdSr)iter)r'riteratorxrrrr;s rz int | None)rargsrc gst|}|jdurdn|j}|j}|jdur0dn|j}|dksV|durN|dksV|dkr^td|dur|}t|D]\}}||krr|V||7}qrnintztuple[Iterator[_T], ...])rnrcs>t|ddgddfdd tfddt|DS)Nr)rc3sRz8|ddur&t|d<ddg|d<|\}}|VqWntyLYdS0dS)Nr,r)next StopIteration)linkvalue)r)rr_tee|s     ztee.._teec3s|]}VqdSrr)r"_)rE shared_linkrrr$r%ztee..)r(tupler3)rr@r)rEr)rGrrws r.) fillvaluez Iterable[_T1]zIterator[tuple[_T1]])iter1rIrcCsdSrr)rJrIrrrrsrz Iterable[_T2]z'Iterator[tuple[_T1 | None, _T2 | None]])rJiter2rcCsdSrr)rJrKrrrrsz#Iterator[tuple[_T1 | _U, _T2 | _U]])rJrKrIrcCsdSrr)rJrKrIrrrrszIterator[tuple[_T | None, ...]])rJrKiter3rrcGsdSrr)rJrKrLrrrrrszIterator[tuple[_T | _U, ...]])rJrKrLrrIrcGsdSrr)rJrKrLrIrrrrrs)rrIrc gsttt|}t|}|sdSg}t|D]X\}}z t|}Wn8tyv|d8}|s`YdSt|||<|}Yn0| |q*t |VqdS)Nr,) listmapr(lenr2rArBrrepeatappendrH)rIr iteratorsZ num_activevaluesr6r)rDrrrrs     )r)#__doc__ __future__rrsystypingrrrrZtyping_extensionsrZ decoratorsr collections.abcr r __all__rrobjectr__annotations__rrr from_iterabler rrr version_infor>rrrrrrs^