a -=ic$ @sdZddlmZddlmZddlmZddlm Z zddl m Z Wne y`ddd Z Yn0zdd l mZWn>e yzdd lmZWne yd d ZYn0Yn0dd lZddgiZddgZedddZddZddZddZd S)z, Thin wrappers around `concurrent.futures`. )absolute_import)contextmanager)tqdm) TqdmWarning) length_hintcCs&z t|WSty |YS0dS)z,Returns `len(it)`, falling back to `default`N)len TypeError)itdefaultr h/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tqdm/contrib/concurrent.pyrs  r) cpu_countcCsdS)Nr r r r r rsrNz github.com/Z casperdcl thread_map process_mapccsPt|dd}|p|}t|||}|||V|durB|`n ||dS)z>get (create if necessary) and then restore `tqdm_class`'s lock_lockN)getattrZget_lockset_lockr) tqdm_class lock_nameZold_locklockr r r ensure_lock"s    rc OsJ|}d|vr t|d|d<|dt}|dtdtd}|dd}|d d }t||d } d|i} tjd d } | dkr| j |j | fdi} d| krdksn| j |d|fi| F} t || j |g|Ri| fi|Wd Wd S1s0YWd n1s<0Yd S)a Implementation of `thread_map` and `process_map`. Parameters ---------- tqdm_class : [default: tqdm.auto.tqdm]. max_workers : [default: min(32, cpu_count() + 4)]. chunksize : [default: 1]. lock_name : [default: "":str]. totalrr max_workers r chunksizerr)rNr))Z initializerZinitargs)rr)r)r) copyrpop tqdm_autominrrsys version_infoupdaterlistmap)Z PoolExecutorfn iterables tqdm_kwargskwargsrrrrlk pool_kwargs sys_versionZmap_argsexr r r _executor_map0s"     r3cOs$ddlm}t||g|Ri|S)a Equivalent of `list(map(fn, *iterables))` driven by `concurrent.futures.ThreadPoolExecutor`. Parameters ---------- tqdm_class : optional `tqdm` class to use for bars [default: tqdm.auto.tqdm]. max_workers : int, optional Maximum number of workers to spawn; passed to `concurrent.futures.ThreadPoolExecutor.__init__`. [default: max(32, cpu_count() + 4)]. r)ThreadPoolExecutor)concurrent.futuresr4r3)r+r,r-r4r r r rOs cOs|ddlm}|rLd|vrLttt|}|dkrLddlm}|d|tddd |vrd|}d |d <t ||g|Ri|S) a Equivalent of `list(map(fn, *iterables))` driven by `concurrent.futures.ProcessPoolExecutor`. Parameters ---------- tqdm_class : optional `tqdm` class to use for bars [default: tqdm.auto.tqdm]. max_workers : int, optional Maximum number of workers to spawn; passed to `concurrent.futures.ProcessPoolExecutor.__init__`. [default: min(32, cpu_count() + 4)]. chunksize : int, optional Size of chunks sent to worker processes; passed to `concurrent.futures.ProcessPoolExecutor.map`. [default: 1]. lock_name : str, optional Member of `tqdm_class.get_lock()` to use [default: mp_lock]. r)ProcessPoolExecutorri)warnzIterable length %d > 1000 but `chunksize` is not set. This may seriously degrade multiprocess performance. Set `chunksize=1` or more.r) stacklevelrZmp_lock) r5r6maxr*rwarningsr7rr"r3)r+r,r-r6Zlongest_iterable_lenr7r r r ras   )r)r)__doc__ __future__r contextlibrautorr$Zstdroperatorr ImportErrorosrmultiprocessingr& __author____all__rr3rrr r r r s.