U md @sdZddlmZddlmZddlmZmZddlm Z ddgiZ d d d gZ Gd d d e Z ddZ ddefdd Zdd Zdd ZdS)z^ Thin wrappers around common functions. Subpackages contain potentially unstable extensions. )warn)tqdm)TqdmDeprecationWarningr) ObjectWrapperz github.com/Z casperdcl tenumeratetziptmapcs2eZdZdZfddZd ddZddZZS) DummyTqdmFilez'Dummy file-like that will write to tqdmcstt||g|_dS)N)superr __init___buf)selfwrapped __class__N/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/tqdm/contrib/__init__.pyr szDummyTqdmFile.__init__FcCsnt|trdnd}||\}}}|r^t|}tj||j||g||j|d|g|_n |j |dS)N  )endfilenolock) isinstancebytes rpartitiontyperwritejoinr _wrappedappend)rxrnlpreseppostblankrrrrs  zDummyTqdmFile.writec CsR|jrNt|jd}ztj||j||jdWnttfk rLYnXdS)Nr)rr)r rrrrrOSError ValueError)rr&rrr__del__"s zDummyTqdmFile.__del__)F)__name__ __module__ __qualname____doc__r rr) __classcell__rrrrr s  r cCstdtdd|S)zReturns `func`z?This function has no effect, and will be removed in tqdm==5.0.0r) stacklevel)rr)funcrrrbuiltin_iterable+s r1NcKshz ddl}Wntk r Yn.Xt||jrN|||fd|pD|ji|St||fd|i||S)z Equivalent of `numpy.ndenumerate` or builtin `enumerate`. Parameters ---------- tqdm_class : [default: tqdm.auto.tqdm]. rNtotal)numpy ImportErrorrZndarrayZ ndenumeratesize enumerate)iterablestartr2 tqdm_class tqdm_kwargsnprrrr2s  cos:|}|dt}t||f|f|D] }|Vq*dS)zr Equivalent of builtin `zip`. Parameters ---------- tqdm_class : [default: tqdm.auto.tqdm]. r9N)copypop tqdm_autozip)Ziter1Z iter2plusr:kwargsr9irrrrEs cost||D]}||Vq dS)zr Equivalent of builtin `map`. Parameters ---------- tqdm_class : [default: tqdm.auto.tqdm]. N)r)function sequencesr:rArrrr Ss)r-warningsrautorr>Zstdrutilsr __author____all__r r1rrr rrrrs