ցjy& SrSSKrSSKJr SSKJr SSKJr SSK J r SS /0r /S Qr "S S 5r \SSj5rSrSrSSSS \SSSSS. SjrSrSrSS.Sjrg)z, Thin wrappers around `concurrent.futures`. N)contextmanager length_hint)tqdm) TqdmWarningz github.com/ casperdcl) thread_map process_mapinterpreter_mapcR\rSrSrSrSSKJr SSKJr Sr S Sjr Sr S r S rS rg )_InterpreterLockz3Reentrant lock backed by a cross-interpreter queue.r) get_ident) monotoniccPSSKJn XlU"5UlSUlSUlg)Nr)RLock) threadingr_queue_lock_owner_depth)selfqueuers Q/mnt/workspace/venv-train/lib/python3.13/site-packages/tqdm/contrib/concurrent.py__init___InterpreterLock.__init__s!# W   cSSKJn UR5nUS:XaURR U5nOURR X5nU(dgUR (aU=R S- slgU(dUR R5 OZUS:XaUR R5 O9[SX R5U- - 5nUR RUS9 UR5Ul SUlg!Ua URR5 gf=f)Nr)EmptyFT)timeout) rr _timeracquirerr get_nowaitgetmaxreleaserr)rblockingr#r startacquired remainings rr%_InterpreterLock.acquires  b=zz))(3Hzz))(get (create if necessary) and then restore `tqdm_class`'s lockrN)getattrget_lockset_lockr) tqdm_class lock_namelockold_locks r ensure_lockrMGsfz7D1H |0:..0 4D )D J  H%sA#A%c S[R<SUR<SURR S5<SU<S3 n[ U44$)zGReturn an initializer which bootstraps the parent import path and lock.zimport sys sys.path[:] = z from concurrent import interpreters from importlib import import_module from tqdm.contrib.concurrent import _InterpreterLock tqdm_class = import_module(z) for name in .z: tqdm_class = getattr(tqdm_class, name) tqdm_class.monitor_interval = 0 tqdm_class.set_lock(_InterpreterLock(interpreters.Queue(z))))syspathr=r>splitexec)rI lock_queue_idcodes r_get_interpreter_initrVVsk  %&'1&;&;%>?!..44S9<=CDQBSSV X  $=rc.^[U4SjU55$)z min(map(length_hint, iterables))c3P># UHn[US5=mS:dMTv M g7f)r!rNr).0itns r _min_map_len..hs&H9Rk"b.A)Aa(Gqq9s& &)min) iterablesr[s @r _min_map_lenr`fs H9H HHrr"g) max_workersr# chunksizerJrI smoothingr _initializer _initargsc R^^U R5n SU ;a[U 5U S'0nSU ;aU RS5US'0nSHnUU ;dM U RU5UU'M! SnU S(aCSU ;a=SSKJn U=(d [S U"5=(d S S -5nU SU:aUU S'S n[XeUS 9nU cURn U4n U"SX)U S.UD6nU"SSU0U D6mUbS Tl URmUU4SjnUUl [UR"U/U Q7X4S.UD65sSSS5 sSSS5 sSSS5 $![ a SSKJn Nf=f!,(df  O=fSSS5 O!,(df  O=fSSS5 g!,(df  g=f)z Implementation of `thread_map`, `process_map` and `interpreter_map`. Parameters ---------- max_workers : int timeout : int buffersize : int Requires Python>=3.14. thread_name_prefix : str max_tasks_per_child : int mp_context : str total buffersize)thread_name_prefixmax_tasks_per_child mp_contextNminitersr)process_cpu_count) cpu_count r"T)rJrK)ra initializerinitargsrccB>T"U0UD6nURU4Sj5 U$)Nc$>TR5$r5)update)_pbars r4_executor_map..patchsubmit..s DKKMr)add_done_callback)argskwargsfut orisubmitrws r patchsubmit"_executor_map..patchsubmits&#T4V4C))*ABJr)r#rbrC)copyr`poposrm ImportErrorrnr^rMrHdynamic_miniterssubmitlistmap) PoolExecutorfnrar#rbrJrIrcrrdrer_ tqdm_kwargsr| map_kwargs pool_kwargskrrn rough_maxlkexrr~rws @@r _executor_maprks$   Ff&y1wJv#)::l#; < K H ;#ZZ]KNI g:V3 % 9 B3rIK,<1+A#B '?Y &!*F: #  Z5 AR  %..LI )kV_)'),.:i:6:d#/,0D)II ( BFFX"X,3XLVXY;:)) B A % $ %;::))) B A AsU=E F! E>,AE#1 E>: F E E # E1 -E>5 F> F F F&c,SSKJn [X0/UQ70UD6$)a` Equivalent of `list(map(fn, *iterables))` driven by `concurrent.futures.ThreadPoolExecutor`. Parameters ---------- max_workers : int, optional Maximum number of workers to spawn; passed to `concurrent.futures.ThreadPoolExecutor`. thread_name_prefix : str, optional Passed to `concurrent.futures.ThreadPoolExecutor` [default: '']. timeout : int or float, optional Seconds to wait before raising `TimeoutError` if `__next__` is called and the result isn't available. [default: None]. buffersize : int, optional Requires Python>=3.14 [default: None]. tqdm_class : optional `tqdm` class to use for bars [default: tqdm.auto.tqdm]. smoothing : float, optional Passed to `tqdm_class`; the [default: 0] is average (due to erratic update frequency). lock_name : str, optional Member of `tqdm_class.get_lock()` to use [default: '']. r)ThreadPoolExecutor)concurrent.futuresrr)rr_rrs rr r s.6 + K) K{ KKrcSSKJn SSKJn UR 5nUR S5 UR S[5n[XeR5upx[X@/UQ7[U5XxS.UD6$)aB Equivalent of `list(map(fn, *iterables))` driven by `concurrent.futures.InterpreterPoolExecutor` (Python 3.14+). Parameters ---------- Same as `thread_map`. Notes ----- `fn`, its arguments, and its return values must be pickleable. Worker progress bars using the same `tqdm_class` share a cross-interpreter write lock. r) interpreters)InterpreterPoolExecutorNrI)rrdre) concurrentrrr create_queuer1r' tqdm_autorVidrr) rr_rrr lock_queuerIrqrrs rr r sz(:**,JNN4y9J1*mmLK  E&/ E7G 7S  E8C EErmp_lock)rJcSSKJn U(a,SU;a&[U5nUS:aSSKJn U"SU-[ SS9 [ X@/UQ7S U0UD6$) a Equivalent of `list(map(fn, *iterables))` driven by `concurrent.futures.ProcessPoolExecutor`. Parameters ---------- max_workers : int, optional Maximum number of workers to spawn; passed to `concurrent.futures.ProcessPoolExecutor`. timeout : int or float, optional Seconds to wait before raising `TimeoutError` if `__next__` is called and the result isn't available. [default: None]. chunksize : int, optional Approximate size of chunks sent to worker processes; passed to `concurrent.futures.ProcessPoolExecutor.map`. [default: 1]. buffersize : int, optional Requires Python>=3.14 [default: None]. max_tasks_per_child : int, optional Maximum number of tasks a worker process can complete before being replaced with a new process; passed to `concurrent.futures.ProcessPoolExecutor`. mp_context : multiprocessing.BaseContext, optional Multiprocessing context to use, e.g. `multiprocessing.get_context('fork')`. lock_name : str, optional Member of `tqdm_class.get_lock()` to use [default: mp_lock]. tqdm_class : optional `tqdm` class to use for bars [default: tqdm.auto.tqdm]. smoothing : float, optional Passed to `tqdm_class`; the [default: 0] is average (due to erratic update frequency). r)ProcessPoolExecutorrbi)warnzIterable length %d > 1000 but `chunksize` is not set. This may seriously degrade multiprocess performance. Set `chunksize=1` or more.r) stacklevelrJ)rrr`warningsrrr)rrJr_rrshortest_iterable_lenrs rr r sd:7[ 3!-Y 7 4 ' % /1FG , , a9 a aU` aar)rDN)r@rP contextlibroperatorrautorrstdr __author____all__rrMrVr`rr r r rCrrrs % $k] + :55p & &  I /3DAY[Ct$RV9YxL6E2+4(br