U Kv fe@s"dZddlmZmZdddZdS)aParallel utility function using joblib copied from https://github.com/mne-tools/mne-python Author: Alexandre Gramfort License: Simplified BSD changes for statsmodels (Josef Perktold) - try import from joblib directly, (does not import all of sklearn) )ModuleUnavailableWarningmodule_unavailable_docc Cszzddlm}m}Wn$tk r:ddlm}m}YnX|||d}||}|dkrzddl}|}Wn6ttfk rddl}| t dt d}YnXWn:tk rddl}| t dt d}|}t }YnX|||fS) aReturn parallel instance with delayed function Util function to use joblib only if available Parameters ---------- func : callable A function n_jobs : int Number of jobs to run in parallel verbose : int Verbosity level Returns ------- parallel : instance of joblib.Parallel or list The parallel object my_func : callable func if not parallel or delayed(func) n_jobs : int Number of jobs >= 0 Examples -------- >>> from math import sqrt >>> from statsmodels.tools.parallel import parallel_func >>> parallel, p_func, n_jobs = parallel_func(sqrt, n_jobs=-1, verbose=0) >>> print(n_jobs) >>> parallel(p_func(i**2) for i in range(10)) r)Paralleldelayed)verboseNmultiprocessingjoblib)r rr ImportErrorZsklearn.externals.joblibr cpu_countNotImplementedErrorwarningswarnrformatrlist) funcZn_jobsrrrparallelZmy_funcr rr>/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/tools/parallel.py parallel_funcs4     rN)r)__doc__Zstatsmodels.tools.sm_exceptionsrrrrrrrs