-iSrSSKrSSKrSSKJr SSKrSSKJr SSKJrJ r Sq Sr "SS \R5r S r "S S 5rS rSSjrg)zWCustomizations of :mod:`joblib` and :mod:`threadpoolctl` tools for scikit-learn usage. N)update_wrapper)ThreadpoolController)config_context get_configc[US5(aURX5$[R"S[5 U$)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.)hasattrr warningswarn UserWarning) delayed_funcconfigwarning_filterss I/var/www/html/venv/lib/python3.13/site-packages/sklearn/utils/parallel.py _with_config_and_warning_filtersrs?|>??;;FTT 7  c,^\rSrSrSrU4SjrSrU=r$)Parallel)aTweak 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 ct>^^[5m[RmUU4SjU5n[TU]U5$)zDispatch 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. c3J># UHupn[UTT5UU4v M g7fN)r).0rargskwargsrrs r $Parallel.__call__..Js54 /7* F 1vW  /7s #)rr filterssuper__call__)selfiterable(iterable_with_config_and_warning_filtersrr __class__s @@rr!Parallel.__call__6s<$"**4 /7 4 0w HIIr)__name__ __module__ __qualname____firstlineno____doc__r!__static_attributes__ __classcell__)r%s@rrr)s JJrrcF^[R"T5U4Sj5nU$)aDecorator used to capture the arguments of a function. This alternative to `joblib.delayed` is meant to be used in conjunction with `sklearn.utils.parallel.Parallel`. The latter captures the scikit- learn configuration by calling `sklearn.get_config()` in the current thread, prior to dispatching the first task. The captured configuration is then propagated and enabled for the duration of the execution of the delayed function in the joblib workers. .. versionchanged:: 1.3 `delayed` was moved from `sklearn.utils.fixes` to `sklearn.utils.parallel` in scikit-learn 1.3. Parameters ---------- function : callable The function to be delayed. Returns ------- output: tuple Tuple containing the delayed function, the positional arguments, and the keyword arguments. c>[T5X4$r) _FuncWrapper)rrfunctions rdelayed_function!delayed..delayed_functionpsH%t33r functoolswraps)r2r3s` rdelayedr8Vs(4__X44 rc*\rSrSrSrSrSrSrSrg)r1wz:Load the global configuration before calling the function.c:Xl[XR5 gr)r2r)r"r2s r__init___FuncWrapper.__init__zs t]]+rcXlX lU$r)rr)r"rrs rr ,_FuncWrapper.with_config_and_warning_filters~s . rc[US05n[US/5nU(aU(d[R"S[5 [ S0UD6 [R "5 U[lUR"U0UD6sSSS5 sSSS5 $!,(df  O=fSSS5 g!,(df  g=f)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.r')getattrr r r rcatch_warningsrr2)r"rrrrs rr!_FuncWrapper.__call__sx,!$(92>_ MM+   %f %x'>'>'@.H ==$1&1(A'@ % %'@'@ % % %s$B/%B B/ B# B// B=)rr2rN) r(r)r*r+r,r<r r!r-r'rrr1r1wsD, 2rr1c0[c [5q[$)z1Return the global threadpool controller instance.)_threadpool_controllerrr'rr_get_threadpool_controllerrFs%!5!7 !!rc^^UU4SjnU$)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. cL>^[R"T5UUU4Sj5nU$)Nc~>[5nURTTS9 T"U0UD6sSSS5 $!,(df  g=f)N)limitsuser_api)rFlimit)rr controllerfuncrJrKs rwrapperD_threadpool_controller_decorator..decorator..wrappers835J!!(!CT,V,DCCs. <r5)rNrOrJrKs` r decorator3_threadpool_controller_decorator..decorators%   -  - rr')rJrKrQs`` r _threadpool_controller_decoratorrSs r)blas)r,r6r rjoblib threadpoolctlr_configrrrErrr8r1rFrSr'rrrYsX$ .0 ")Jv)JZB22>"r