idZddlZddlZddlmZddlZddlmZddlmZm Z da dZ GddejZ d Z Gd d Zd Zdd Zy)zWCustomizations of :mod:`joblib` and :mod:`threadpoolctl` tools for scikit-learn usage. N)update_wrapper)ThreadpoolController)config_context get_configcvt|dr|j||Stjdt|S)zFHelper function that intends to attach a config to a delayed function.with_config_and_warning_filtersz`sklearn.utils.parallel.Parallel` needs to be used in conjunction with `sklearn.utils.parallel.delayed` instead of `joblib.delayed` to correctly propagate the scikit-learn configuration to the joblib workers.)hasattrrwarningswarn UserWarning) delayed_funcconfigwarning_filterss k/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/sklearn/utils/parallel.py _with_config_and_warning_filtersrs>|>?;;FOTT 7  c"eZdZdZfdZxZS)ParallelaTweak of :class:`joblib.Parallel` that propagates the scikit-learn configuration. This subclass of :class:`joblib.Parallel` ensures that the active configuration (thread-local) of scikit-learn is propagated to the parallel workers for the duration of the execution of the parallel tasks. The API does not change and you can refer to :class:`joblib.Parallel` documentation for more details. .. versionadded:: 1.3 ctttdd}||ntjfd|D}t||S)aHDispatch the tasks and return the results. Parameters ---------- iterable : iterable Iterable containing tuples of (delayed_function, args, kwargs) that should be consumed. Returns ------- results : list List of results of the tasks. _get_filtersNc3FK|]\}}}t|||fywN)r).0r argskwargsrrs r z$Parallel.__call__..Ss84 /7* dF 1vW  /7s!)rgetattrr filterssuper__call__)selfiterable filters_func(iterable_with_config_and_warning_filtersrr __class__s @@rr zParallel.__call__6sX$ x> *6LNH.delayed_functionysH%tV33r functoolswraps)r.r/s` rdelayedr3_s(4__X44 rc"eZdZdZdZdZdZy)r-z:Load the global configuration before calling the function.c>||_t||jyr)r.r)r!r.s r__init__z_FuncWrapper.__init__s  tT]]+rc"||_||_|Sr)rr)r!rrs rrz,_FuncWrapper.with_config_and_warning_filterss . rc t|di}t|dg}|r|stjdtt d i|5tj 5tj gd}|D]}t||Dcic] \}}||| } }}d| vrd| vrtjd i| ddiFd D]6} | j| } | t| tr(| j| | <8tjd i| ddi|j|i|cdddcdddScc}}w#1swYnxYwdddy#1swYyxYw) Nrrz`sklearn.utils.parallel.delayed` should be used with `sklearn.utils.parallel.Parallel` to make it possible to propagate the scikit-learn configuration of the current thread to the joblib workers.)actionmessagecategorymodulelinenor:r<appendT)r:r<)rr r r rcatch_warnings resetwarningszip simplefilterget isinstancestrpatternfilterwarningsr.) r!rrrrwarning_filter_keys filter_argskvthis_warning_filter_dict special_key this_values rr z_FuncWrapper.__call__scx,!$(92>_ MM+   %f %x'>'>'@  " " $"W . !$$7 E, E1}qD E),%== (@@))R,DRTR(= %=%A%A+%N %1*ZQT:UDNDVDV4[A(= ++T.FTtT1 /4!4==$1&1=(A'@ % % , (A'@ % % %sBE-D:D4 =D:D:$=D:! E4D::E ?EEN)r&r'r(r)r6rr r?rrr-r-sD, ,2rr-c.t tatS)z1Return the global threadpool controller instance.)_threadpool_controllerrr?rr_get_threadpool_controllerrRs%!5!7 !!rcfd}|S)aDecorator to limit the number of threads used at the function level. It should be preferred over `threadpoolctl.ThreadpoolController.wrap` because this one only loads the shared libraries when the function is called while the latter loads them at import time. cHtjfd}|S)Nczt}|j5|i|cdddS#1swYyxYw)N)limitsuser_api)rRlimit)rr controllerfuncrVrWs rwrapperzD_threadpool_controller_decorator..decorator..wrappers835J!!(!CT,V,DCCs1:r0)rZr[rVrWs` r decoratorz3_threadpool_controller_decorator..decorators%   -  - rr?)rVrWr\s`` r _threadpool_controller_decoratorr]s r)blas)r)r1r rjoblib threadpoolctlrsklearn._configrrrQrrr3r-rRr]r?rrrcsY$ .6 "2Jv2JlB8282v"r