U Nv f @s:ddlZddZddZGdddZdd Zd d ZdS) NcCs*ddlm}|f|}||dS)z3Initialize viztracer's profiler in worker processesr) VizTracerN) viztracerr register_exitstart) init_kwargsrtracerrF/tmp/pip-unpacked-wheel-dylwa62s/joblib/externals/loky/initializers.py_viztracer_inits  r c Csz6ddl}|}|dk r4t|ddr4t|jffWSWnDtk rJYn2tk rz}ztd|W5d}~XYnXdS)NrenableFz&Unable to introspect viztracer state: Nr) rZ get_tracergetattrr r ImportError Exceptionwarningswarn)rrerrr (_make_viztracer_initializer_and_initargs s"rc@s eZdZdZddZddZdS)_ChainedInitializerzCompound worker initializer This is meant to be used in conjunction with _chain_initializers to produce the necessary chained_args list to be passed to __call__. cCs ||_dSN) _initializers)selfZ initializersrrr __init__'sz_ChainedInitializer.__init__cGs"t|j|D]\}}||q dSr)zipr)rZ chained_args initializerargsrrr __call__*sz_ChainedInitializer.__call__N)__name__ __module__ __qualname____doc__rrrrrr r srcCsfg}g}|D]$\}}|dk r ||||q |s:dSt|dkrV|d|dfSt||fSdS)zyConvenience helper to combine a sequence of initializers. If some initializers are None, they are filtered out. Nr r)appendlenr)Zinitializer_and_argsZfiltered_initializersZfiltered_initargsrinitargsrrr _chain_initializers/s    r%cCs0|dk rt|std|t||ftgS)Nz%initializer must be a callable, got: )callable TypeErrorr%r)rr$rrr _prepare_initializerCsr()rr rrr%r(rrrr s