o j9:jQ&@sUdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z e r3ddlmZmZgd Ze d Ze d ZeegefZd ed <e dZe dZe ejddddddZe ejdd dedddfddZe ejjdgd!d"Zee_e ejdddhd&d'Ze ejdddid(d)Ze ejdddjd,d-Ze ejdddjd.d/Zedkd4d5Z edld8d5Z edmd;d5Z ednd>d5Z e ej dddodAd5Z e ej!dddpdBdCZ!e ej"dddqdFdGZ"e ej#dddrdIdJZ#e ej$dsdtdNdOZ$edPdQdudVdWZ%edvd[dWZ%edPdQdwd]dWZ%edxd`dWZ%edPdQdydbdWZ%e ej%ddddQdzdcdWZ%dS){z 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]cg|D]}|EdHqdSN)riterabler%r%h/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/torch/_dynamo/polyfills/itertools.pyr ,s r )initialr&funcCallable[[_T, _T], _T] | Noner( _T | Nonecs.t|}dur tjdfdd }||S)Niteratorr"r!c3sT}|durzt|}Wn tyYdSw|V|D] }||}|VqdSr$next StopIteration)r,totalelementr)r(r%r' _accumulate?s   zaccumulate.._accumulater,r"r!r")iteroperatoradd)r&r)r(r,r3r%r2r'r 3s  r Iterable[Iterable[_T]]ccr#r$r%)r&itr%r%r'rOs rdata selectors Iterable[_U]cCsddt||DS)Ncss|] \}}|r|VqdSr$r%).0datumselectorr%r%r' ^szcompress..)zip)r:r;r%r%r'r\srcst}dfdd }||S)Nr,r"r!c3s@g}D] }|V||q|r|D]}|Vq|sdSdSr$)append)r,savedr1r&r%r'_cyclefs  zcycle.._cycler4)r5)r&r,rEr%rDr'rbs r predicate_Predicate[_T]ccsRt|stdt|jdt|}|D] }||s!|Vnq|EdHdSN'z' object is not callable)callable TypeErrortype__name__r5)rFr&r,xr%r%r'rusrccsBt|stdt|jd|D] }||sdS|VqdSrHrJrKrLrM)rFr&rNr%r%r'rsrfunctionCallable[[], _U]Iterable[tuple[()]]itertools.starmap[_U]cCdSr$r%rPr&r%r%r'rrCallable[[_T], _U]Iterable[tuple[_T]]cCrTr$r%rUr%r%r'rrVCallable[[_T, _T1], _U]Iterable[tuple[_T, _T1]]cCrTr$r%rUr%r%r'rrVCallable[[_T, _T1, _T2], _U]Iterable[tuple[_T, _T1, _T2]]cCrTr$r%rUr%r%r'rrVCallable[..., _T]r ccs8t|stdt|jd|D]}||VqdSrHrO)rPr&argsr%r%r'rs  cs.t|}durttj|Stfdd|S)Ncs | Sr$r%)rNrPr%r's zfilterfalse..)r5filterr6not_)rPr&r9r%r_r'rs rr^ 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 enumeraterangemaxrA) r&r^srfrgrhnext_iir1indicesr%r%r'rs2 rIterator[tuple[_T, _T]]ccs0d}d}|D]}|rd}n||fV|}qdS)NTFr%)r&afirstbr%r%r'rs rninttuple[Iterator[_T], ...]cs:t|ddgdfdd tfddt|DS)Nr!r"c3sNz |ddurt|d<ddg|d<|\}}|Vqty&YdSw)NTrdrr-)linkvalue)r,r%r'_tees    ztee.._teec3s|]}VqdSr$r%)r=_)rz shared_linkr%r'r@sztee..)r!r")r5tuplerk)r&rur%)rzr,r|r'rs r.) fillvalueiter1 Iterable[_T1]r~Iterator[tuple[_T1]]cCrTr$r%)rr~r%r%r'rsriter2 Iterable[_T2]'Iterator[tuple[_T1 | None, _T2 | None]]cCrTr$r%)rrr%r%r'rs#Iterator[tuple[_T1 | _U, _T2 | _U]]cCrTr$r%)rrr~r%r%r'riter3Iterator[tuple[_T | None, ...]]cGrTr$r%)rrrrr%r%r'r"rIterator[tuple[_T | _U, ...]]cGrTr$r%)rrrr~rr%r%r'r-s c gsttt|}t|}|sdS g}t|D],\}}zt|}Wnty=|d8}|s2YdSt|||<|}Ynw| |qt |Vq)NTrd) listmapr5lenrjr.r/ itertoolsrepeatrBr})r~r iterators num_activevaluesror,ryr%r%r'r:s(    )rr r!r"r$)r&r r)r*r(r+r!r")r&r8r!r")r:r r;r<r!r")r&r r!r")rFrGr&r r!r")rPrQr&rRr!rS)rPrWr&rXr!rS)rPrYr&rZr!rS)rPr[r&r\r!rS)rPr]r&r r!r )rPrGr&r r!r")r&r r^rcr!r")r&r r!rq)r)r&r rurvr!rw)rrr~rr!r)rrrrr!r)rrrrr~rr!r) rr rr rr rr r!r) rr rr rr rr r~rr!r)rr r~rr!r)&__doc__ __future__rrr6collections.abcrtypingrrrr decoratorsr r r __all__rrobjectr__annotations__rrr r from_iterablerrrrrrrrrrrr%r%r%r's