qtj*,nddlmZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZmZiZiZGdd Z dd lmZmZmZmZdd lmZGd d eZGddZGddeeZGddeeZn #e$rYnwxYwGddeZ GddZ!dS))Integral)Queue)Thread)get_context_path)close_listener open_listenersend)check_rich_support)Intervalvalidate_paramsc zeZdZdZedeeddddgiddd Zd Zd Z d Z dZ dS) 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_validationc@td||_i|_dS)N Progressbar)r r_listener_handles)selfrs [/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/sklearn/callback/_progressbar.py__init__zProgressBar.__init__s* =)))%:""$ct}t|j|j|j<t |}||t|j<|t|j<dS)N)queue) rrputr root_uuidRichProgressMonitorstart _run_queues _run_monitors)r estimatorcontextrprogress_monitors rsetupzProgressBar.setup+sm5B%)4L4Lw01.U;;;   ). G%&+; g'(((rc |jdks|j|jbdt|D}t |j|jd||j|j|j|j |j|j ddSdS)Nrcg|] }|j Stask_id.0ctxs r z1ProgressBar.on_fit_task_begin..HsEEECCKEEErbegin)eventpath task_namer+ max_subtasksestimator_namesource_estimator_namesource_task_name) r4parentr6rr rrr3r+r5r7)rr#r$r2s ron_fit_task_beginzProgressBar.on_fit_task_begin>s  A % %~%,8FE+;G+D+DEEED &w'89$ !(!2&$+$8&-&<-4-J(/(@        98rczt|j|jddt|DddS)Nendcg|] }|j Sr)r*r,s rr/z/ProgressBar.on_fit_task_end..]sJJJJJJr)r1r2)r rrrrr#r$s ron_fit_task_endzProgressBar.on_fit_task_endWsS  "7#4 5JJ0@0I0IJJJ       rc$t|jdt|jt |j|jdSN)r!poprrr"joinrrr=s rteardownzProgressBar.teardownasr )**..t444'+,,11333t-11'2CDDEEEEEr)r) __name__ __module__ __qualname____doc__r r rrr&r9r>rCr)rrrrs_ 88Haf#M#M#Mt"TU&*$$$ $<<<&2   FFFFFrr) BarColumnProgress TextColumnTimeRemainingColumn)StyleceZdZdZdS) _Progressc#VK|t|dgVdS)N_ordered_tasks)make_tasks_tablegetattrrs rget_renderablesz_Progress.get_renderablesqs3''6F(K(KLL L L L L LrN)rDrErFrTr)rrrNrNms( M M M M MrrNc"eZdZdZfdZxZS)_StyledColumnMixinz8Apply finished/in-progress color style to rendered text.cjt|}|jrdnd|_|S)N#29ABE2#F7931E)superrenderfinishedstyle)rtasktext __class__s rr[z_StyledColumnMixin.renderws/77>>$''D&*mBDJKr)rDrErFrGr[ __classcell__r`s@rrVrVts>FF         rrVceZdZdZdS)_StyledTimeRemainingColumnzTime column with color styling.NrDrErFrGr)rrrdrd|s----rrdceZdZdZdS)_StyledPercentageColumnz%Percentage column with color styling.Nrer)rrrgrgs3333rrgc4eZdZdZfdZdZdZdZxZS)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. cVt||_dSr@)rZrr)rrr`s rrzRichProgressMonitor.__init__s$  rc lttdttdtdtdt dt d|_d|_|j5|j x}r_| d d kr| |n| ||j x}_|j ddddS#1swxYwYdS) Nz([progress.description]{task.description}rY)colorrX)complete_stylefinished_style pulse_stylez{task.percentage:>3.0f}%T)elapsed_when_finishedr1r0)rNrJrHrLrgrd progress_ctxroot_rich_taskrgetrA_on_task_begin _on_task_endrefresh)r task_infos rrunzRichProgressMonitor.runs|% A B B $9555$9555! 222    $$> ? ? &T B B B   #   ( (#z~~///) 1==))W44'' 2222%%i000 $z~~///) 1   % % ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sBD))D-0D-c@|d}tj|t|dz }|jdkr|_n0j|d}||j|d<fdjDj_dS) zHCreate a progress bar for the task and update the list of ordered tasks.r2r)rprvdepthrc>g|]}jj|jSr))rptasksid)r-r^rs rr/z6RichProgressMonitor._on_task_begin..s3, , , 15D  #DG ,, , , rN) rARichTaskrplenryrqget_descendantschildrenrP)rrvr2 rich_taskr8s` rrsz"RichProgressMonitor._on_task_begins}}V$$*is4yyST}    ?a  "+D  (88>>rBF(1FODH %, , , , 9=9L, , , (((rc|d}|j|^}}|;t|j|t |}||dj|d<n.|j|_|j |j ddt|D]:}|j |j |jz}|j |j |;dS)z>Update the progress of the task and its ancestors recursively.r2Nryr{r) completedtotal)r) rArqrrrprrrrupdater~reversedprogress)rrvr2 ancestorsr^ancestorrs rrtz RichProgressMonitor._on_task_ends}}V$$.>>tDDD <D-yIOOOD/3IbM "48 , ,"ZDN   $ $TW $ C C C ++ G GH~% )HN:I   $ $X[I $ F F F F  G Gr) rDrErFrGrrwrsrtrarbs@rrrsv  (((2   "GGGGGGGrrc@eZdZdZdZedZdZdZdZ dS)raA 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|_|dd|_|r8||}|||j|_dSdS)Nrr4)r)rryrrAr_format_task_descriptionadd_taskr~)rrprvry descriptions rrzRichTask.__init__st  ]]>155  K77 BBK"++Ktz+JJDGGG K Krc|j|jkrdS|jdStd|jD|jz S)z2Return the fraction of the task that is completed.g?Ngc3$K|] }|jV dSr@)r)r-childs r z$RichTask.progress.. s$FFe5>FFFFFFr)rrsumrvaluesrSs rrzRichTask.progresssW >TZ ' '3 : 3FFt}/C/C/E/EFFFFFSSrcd|jz}|dd|d}|jdkr d|dnd}|d |d d|d d nd}||||S) z,Return a formatted description for the task.z r5z - r3rz #r+r6Nr7z | r)rrvindent task_descid_marksource_task_descs rrz!RichTask._format_task_description s4:%' !12OOy7MOO 15a-y+---R01=01 X Xi@R6S X X X X  @*@I@w@@@rct|dkr|gS|d|jvr|dgS|g|j|d|ddzS)z;Return the descendants from this task along the given path.rN)rrr)rr2s rrzRichTask.get_descendantssa t99>>6M 7$- ' '$< v d1g.>>tABBxHHHHrc#~Kt|dr(|V|jD] }|Ed{V dSdS)zJPre-order depth-first traversal, only including tasks with a progress bar.r~N)hasattrrr)rrs r__iter__zRichTask.__iter__!sm 4   !JJJ--// ! !          ! ! ! !rN) rDrErFrGrpropertyrrrrr)rrrrs  DKKKTTXT A A AIII!!!!!rrN)"numbersrrr threadingr"sklearn.callback._callback_contextrsklearn.callback._transportrrr $sklearn.utils._optional_dependenciesr sklearn.utils._param_validationr r r!r"r rich.progressrHrIrJrK rich.stylerLrNrVrdrg ImportErrorrrr)rrrsj??????KKKKKKKKKKCCCCCCEEEEEEEE   RFRFRFRFRFRFRFRFj RRRRRRRRRRRR      MMMMMHMMM.....%79L...44444"4j4444   D NGNGNGNGNG&NGNGNGbO!O!O!O!O!O!O!O!O!O!sA BBB