Q3j*,@ddlmZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZmZiZiZGdd Z dd lmZmZmZmZdd lmZGd d eZGddZGddeeZGddeeZGddeZ GddZ!y#e$rYwxYw))Integral)Queue)Thread)get_context_path)close_listener open_listenersend)check_rich_support)Intervalvalidate_paramsc feZdZdZedeeddddgiddd Zd Zd Z d Z d Z y) ProgressBarauCallback that displays progress bars for each iterative step of an estimator. Parameters ---------- max_propagation_depth : int, default=1 The maximum depth of nested levels of estimators to display progress bars for. 0 means that the progress of only the outermost estimator is displayed. If set to None, all levels are displayed. max_propagation_depthrNleft)closedT)prefer_skip_nested_validationc6td||_i|_y)N Progressbar)r r_listener_handles)selfrs J/DATA/.local/lib/python3.12/site-packages/sklearn/callback/_progressbar.py__init__zProgressBar.__init__s =)%:""$ct}t|j|j|j<t |}|j |t|j<|t|j<y)N)queue) rrputr root_uuidRichProgressMonitorstart _run_queues _run_monitors)r estimatorcontextrprogress_monitors rsetupzProgressBar.setup+sd5B%))4Lw001.U; ). G%%&+; g''(rc r|jdk7s|j |jt|Dcgc]}|j}}t |j |jd||j|j|j|j|j|jdyycc}w)Nrbegin)eventpath task_nametask_id max_subtasksestimator_namesource_estimator_namesource_task_name) r,parentr.rr+r rrr*r-r/)rr"r#ctxr)s ron_fit_task_beginzProgressBar.on_fit_task_begin>s  A %~~%,,8,rMr@rrrGrGms MrrGc"eZdZdZfdZxZS)_StyledColumnMixinz8Apply finished/in-progress color style to rendered text.c^t||}|jr d|_|Sd|_|S)N#29ABE2#F7931E)superrenderfinishedstyle)rtasktext __class__s rrTz_StyledColumnMixin.renderws27>$'D&*mmDJK:CDJKr)r<r=r>r?rT __classcell__rYs@rrOrOtsF  rrOceZdZdZy)_StyledTimeRemainingColumnzTime column with color styling.Nr<r=r>r?r@rrr]r]|s-rr]ceZdZdZy)_StyledPercentageColumnz%Percentage column with color styling.Nr^r@rrr`r`s3rr`c4eZdZdZfdZdZdZdZxZS)raWThread monitoring the progress of an estimator with rich based display. The display is a list of nested rich tasks using `rich.Progress`. There is one for each non-leaf node in the task tree of the estimator. Parameters ---------- queue : `queue.Queue` instance This thread will run until the queue is empty. c0t|||_yr7)rSrr)rrrYs rrzRichProgressMonitor.__init__s  rc ttdttdtdtdt dt d|_d|_|j 5|jjx}rT|jd d k(r|j|n|j||jjx}rT|j jdddy#1swYyxYw) Nz([progress.description]{task.description}rR)colorrQ)complete_stylefinished_style pulse_stylez{task.percentage:>3.0f}%T)elapsed_when_finishedr(r')rGrCrArEr`r] progress_ctxroot_rich_taskrgetr8_on_task_begin _on_task_endrefresh)r task_infos rrunzRichProgressMonitor.runs% A B $95$95! 2  $$> ? &T B  #   #zz~~//)/==)W4'' 2%%i0 $zz~~//)/    % % '  s-A0DDD c|jd}t|j|t|dz }|jdk(r||_n0|j j |d}||j|d<|j Dcgc]%}|jj|j'c}|j_ ycc}w)zHCreate a progress bar for the task and update the list of ordered tasks.r)r;)rirodepthrN) r8RichTaskrilenrrrjget_descendantschildrentasksidrI)rror) rich_taskr0rWs rrlz"RichProgressMonitor._on_task_begins}}V$**is4yST}  ??a "+D ((88>rBF(1FOODH %:>9L9L, 9LD   # #DGG ,9L, (, s *Cc|jd}|jj|^}}|7t|j|t |}||dj |d<n9|j|_|jj|jddt|D]O}|j|j|jz}|jj|j|Qy)z>Update the progress of the task and its ancestors recursively.r)Nrrrtr;) completedtotal)r~) r8rjrwrurirvrxrr~updaterzreversedprogress)rror) ancestorsrWancestorr~s rrmz RichProgressMonitor._on_task_ends}}V$..>>tDD <D--yIOD/3IbM " "48 ,"ZZDN    $ $TWW $ C +H~~% ))HNN:I    $ $X[[I $ F ,r) r<r=r>r?rrprlrmrZr[s@rrrs (2 "Grrc8eZdZdZdZedZdZdZdZ y)ruaA task, i.e. progressbar, in the tree of rich tasks. There is a rich task for each non-leaf task in the context tree of the estimator. Parameters ---------- progress_ctx : `rich.Progress` instance The progress context to which this task belongs. task_info : dict Information about the task for which this rich task is created. depth : int The depth of the task in the tree of rich tasks. Attributes ---------- completed : int The number of completed subtasks. total : int or None The total number of subtasks. None if the total number of subtasks is not known. progress : float The fraction, between 0 and 1, of the task that is completed. id : int The ID of the task in the Progress context. children : dict A mapping from the index of a child to the child node `{idx: RichTask}`. For a leaf, it's an empty dictionary. ci|_||_d|_|jdd|_|r4|j |}|j ||j|_yy)Nrr,)r)rxrrr~r8r_format_task_descriptionadd_taskrz)rrirorr descriptions rrzRichTask.__init__s\  ]]>15 77 BK"++Ktzz+JDG rc|j|jk(ry|jytd|jj D|jz S)z2Return the fraction of the task that is completed.g?gc34K|]}|jywr7)r).0childs r z$RichTask.progress.. sF/Ee5>>/Es)r~rsumrxvaluesrLs rrzRichTask.progresssK >>TZZ ' :: Ft}}/C/C/EFFSSrcd|jz}|dd|d}|jdkDrd|dnd}|d |d d|d d nd}||||S) z,Return a formatted description for the task.z r-z - r*rz #r+r.r/z | r})rroindent task_descid_marksource_task_descs rrz!RichTask._format_task_description s4::%& !1233y7M6NO 15aBy+,-R01=012#i@R6S5TTW X  *+I;wi@@rct|dk(r|gS|d|jvr|dgS|g|j|dj|ddzS)z;Return the descendants from this task along the given path.r;N)rvrxrw)rr)s rrwzRichTask.get_descendantssY t9>6M 7$-- '$< v d1g.>>tABxHHHrc#Kt|dr.||jjD] }|Ed{yy7w)zJPre-order depth-first traversal, only including tasks with a progress bar.rzN)hasattrrxr)rrs r__iter__zRichTask.__iter__!s? 4 J--/   0 !s3A> AN) r<r=r>r?rpropertyrrrwrr@rrrurus5 DKTT AI!rruN)"numbersrrr threadingr"sklearn.callback._callback_contextrsklearn.callback._transportrrr $sklearn.utils._optional_dependenciesr sklearn.utils._param_validationr r r r!r rich.progressrArBrCrD rich.stylerErGrOr]r` ImportErrorrrur@rrrs?KKCE   RFRFj RR MHM.%79L.4"4j4NG&NGbO!O!k  s?BBB