o i@sddlZddlmZddlZddlZddlmZmZmZddlm Z ddl m Z m Z ddZ dd Zd d Zd d ZejdgdddZejje ddddZddZdS)N)ThreadPoolExecutor)config_context get_config set_config)_IS_WASM)Paralleldelayedc Cshtddddddddddd ksJtddtd dus Jtddtddddddddddd ks8JWdn1sBwYtd dusPJtddtddtd duseJWdn1sowYtd dus}JtddFtd dusJtddtd dusJtddtd dusJWdn1swYtd dusJWdn1swYtd dusJWdn1swYtddddddddddd ksJtt tdWdn 1swYtttdd WddS1s-wYdS) NFiTdiagramdefault) assume_finiteZworking_memoryZprint_changed_onlydisplayarray_api_dispatchZpairwise_dist_chunk_sizeZenable_cython_pairwise_distZtransform_outputZenable_metadata_routingZskip_parameter_validationr r Zdo_something_else)rrrpytestraises TypeError __enter__rrh/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/sklearn/tests/test_config.pytest_config_context s~           $rcCsttddus Jztdd tddusJt1swYWn ty.Ynwtddus8JdS)Nr FTr)rr ValueErrorrrrrtest_config_context_exceptionXs  rcCstddus JtddtddusJtddtddus%Jtddtddus3JtddtddusAJtttddWddS1sWwYdS)Nr FrTr)rrrrrrrrrtest_set_configcs      "rcCsBt|dt|tdWdS1swYdS)zAReturn the value of assume_finite after waiting `sleep_duration`.rr N)rtimesleepr)r Zsleep_durationrrrset_assume_finitess  $rbackend)Zlokymultiprocessing threadingcCsDgd}gd}t|ddddt||D}|gdks JdS)aoTest that the global config is threadsafe with all joblib backends. Two jobs are spawned and sets assume_finite to two different values. When the job with a duration 0.1s completes, the assume_finite value should be the same as the value passed to the function. In other words, it is not influenced by the other job setting assume_finite to True. FTFT皙?皙?r#r$)rZn_jobscss"|] \}}tt||VqdS)N)rr).0r Z sleep_durrrr s  z0test_config_threadsafe_joblib..N)rzip)rassume_finitessleep_durationsitemsrrrtest_config_threadsafe_joblibzs r,zcannot start threads)reasoncCsfgd}gd}tdd}dd|t||D}Wdn1s$wY|gdks1JdS)zUses threads directly to test that the global config does not change between threads. Same test as `test_config_threadsafe_joblib` but with `ThreadPoolExecutor`.r!r"r%) max_workerscSsg|]}|qSrr)r&outputrrr sz*test_config_threadsafe..N)rmapr)r)r*er+rrrtest_config_threadsafes  r3c Cs|tjjjddtjtddtddWdn1s"wYWdn1s1wYtjtddt ddWddS1sNwYdS)z!Check error when SciPy is too old __version__z1.13.0zSciPy must be 1.14.0 or newer)matchT)rN) setattrsklearnutilsZ _array_apiZscipyrr ImportErrorrr)Z monkeypatchrrr*test_config_array_api_dispatch_error_scipys  "r:)rconcurrent.futuresrrr7rrrZsklearn.utils.fixesrZsklearn.utils.parallelrrrrrrmarkZ parametrizer,Zxfailr3r:rrrrs   L