Q3j*ddlZddlmZddlmZmZmZmZmZddl m Z m Z ejjdegeeggdZejjddeegdZejjd gd d Zd Zd ZejjdddgejjdddgejjdeegdZdZy)N)clone)FailingCallbackMaxIterEstimatorNotValidCallbackRecordingAutoPropagatedCallbackRecordingCallback)Paralleldelayed callbacksct}|j|}t|dsJ|j|k(sJ||usJy)z,Sanity check for the `set_callbacks` method._skl_callbacksN)r set_callbackshasattrr )r estimatorset_callbacks_returns Y/DATA/.local/lib/python3.12/site-packages/sklearn/callback/tests/test_callback_support.pytest_set_callbacksrsR!"I2922I> 9. // /  # #y 00 0 9 ,, ,callbackct}tjtd5|j |dddy#1swYyxYw)zMCheck the error message when not passing a valid callback to `set_callbacks`.z?callbacks must be instances following the FitCallback protocol.matchN)rpytestraises TypeErrorr)rrs rtest_set_callbacks_errorr$s>!"I O  )    s AA fail_at)setupon_fit_task_beginon_fit_task_endteardownc,t|}tj|}tjt d|5|j ddd|jddk(sJ|jddk(sJy#1swY6xYw)z6Check that a failing callback is properly teared down.rzFailing callback failed at rNrr!)rrrrr ValueErrorfit count_hooks)rrrs rtest_callback_errorr(0s w/H "00:I z+Fwi)P Q  R    (A -- -    +q 00 0 R Qs B  Bctd}t}tj||}t j t d5|jddd|jddk(sJ|jddk(sJ|jddk(sJ|jddk(sJy#1swYbxYw) z?Check that teardown only runs for callbacks that entered setup.rr#z Failing callback failed at setuprNr$r!r) rrrrrrr%r&r') callback_1 callback_2rs r:test_teardown_matches_setup_calls_on_partial_setup_failurer,>s 1J"$J "00ZHI z)K L  M  ! !' *a // /  ! !* - 22 2  ! !' *a // /  ! !* - 22 2 M Ls B>>Cctd}td}tj||}tjt d5}|j dddtjjdk(sJ|jddk(sJ|jddk(sJy#1swYZxYw)zuCheck that all teardown are called even if some fail. All errors are raised together in one ExceptionGroup. r!r#z/The following callback teardown errors occurredrNr$) rrrrrExceptionGroupr&lenvalue exceptionsr')r*r+rexc_infos r)test_multiple_teardown_errors_are_groupedr4Os !4J 4J "00ZHI O     x~~(( )Q .. .  ! !* - 22 2  ! !* - 22 2  s B88Cn_jobsr$r.preferthreads processesCallbackcBdfd}d\}}|}t|j|}||||||jd|k(sJ|jd|d|zzk(sJ|jd|d|zzk(sJ|jd |k(sJy ) aCheck callbacks on estimators within function not supporting callbacks. Since the outer function does not support callbacks, there's no shared root context and the context trees of each sub-estimator are independent. As a result, the callback acts as a regular non-propagated callback: its on_fit_begin and on_fit_end are called once for each fit of the sub-estimator and the number of tasks is the sum of the number of tasks from all the sub-estimators. c6t|jyN)rr&rs r clone_and_fitz8test_function_no_callback_support..clone_and_fitqs ircTt||fdt|Dy)N)r5r6c3@K|]}tywr<)r ).0_r>rs r zBtest_function_no_callback_support..func..us"/ 7D! "GM "9 -}s)r range)rn_fitsr5r6r>s` rfuncz/test_function_no_callback_support..functs%.v./ 7s @r!test_function_no_callback_supportrJbs FHzH (3AA(KIFFF+    (F 22 2    3 4!h,8O OO O    1 2fH 6M MM M    +v 55 5rct}|jtt|dsJ|jt|drJ|jt|drJy)zGCheck that setting no callbacks removes the `_skl_callbacks` attribute.r N)rrrrr=s rtest_set_callback_emptyrLsl "I -/0 9. // / y"233 3y"233 33r)r sklearn.basersklearn.callback.tests._utilsrrrrrsklearn.utils.parallelr r mark parametrizerrr(r,r4rJrLrrrSs35    =?@ - -d,<,>@Q%RS*T* L113"3&Aq6*I{#;<"$CD6=+ 6> 4r