qtjUddlZddlZddlZddlZddlZddlmZddlmZmZddl m Z gdZ ej ej ZGddZdZd ZdS) N)contextmanager)datetimetimezone)AutoPropagatedCallback)Xymetadatafitted_estimatorceZdZdZedZedZdZdZdZ dZ dd Z d Z d d d d ddZ d d d d ddZedZd S)CallbackContextaTask level context for the callbacks. This class is responsible for managing the callbacks and holding the tree structure of an estimator's tasks. Each instance corresponds to a task of the estimator. This class should not be instantiated directly, but through the `_init_callback_context` method of the estimator to create the root context or using the `subcontext` method of this class to create sub-contexts. These contexts are passed to the callback hooks to be able to keep track of the position of a task in the task tree from within the callbacks. Attributes ---------- task_name : str The name of the task this context is responsible for. task_id : int The identifier of the task this context is responsible for. It uniquely identifies the task among its siblings. max_subtasks : int or None The maximum number of children tasks for this task. 0 means it's a leaf. None means the maximum number of subtasks is not known in advance. sequential_subtasks : bool Whether this context's subtasks are sequential. When True, children contexts' have consecutive integer task_ids starting from 0. estimator_name : str The name of the estimator that holds this context. parent : CallbackContext or None The parent context of this context. None if this context is the root. root_uuid : uuid.UUID instance The UUID of the root context. All contexts in the same task tree have the same root UUID that is used to identify the task tree itself. init_time : datetime.datetime The time when the context was initialised, in the UTC timezone. source_estimator_name : str or None The name of the estimator that holds the parent task this task was merged with. None if it was not merged with another context. source_task_name : str or None The task name of the parent task this task was merged with. None if it was not merged with another context. c||}t|dg|_|jj|_||_||_||_||_ d|_ tj |_ tjt j|_i|_d|_d|_t-|dr,|j}|||jdz|_nd|_|S)a#Private constructor to create a root context. Parameters ---------- estimator : estimator instance The estimator this context is responsible for. task_name : str The name of the root task. task_id : int Identifier for the root task. max_subtasks : int or None The maximum number of subtasks that can be children of the root task. None means the maximum number of subtasks is not known in advance. 0 means it's a leaf. sequential_subtasks : bool Whether the root context has sequential subtasks. If True, children contexts created via `subcontext` will have automatically assigned consecutive integer task_ids starting from 0. _skl_callbacksN_parent_callback_ctxr)__new__getattr _callbacks __class____name__estimator_name task_nametask_id max_subtaskssequential_subtasksparentuuiduuid4 root_uuidrnowrutc init_time _children_mapsource_estimator_namesource_task_namehasattrr _merge_with_propagation_depth)cls estimatorrrrrnew_ctx parent_ctxs `/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/sklearn/callback/_callback_context.py_from_estimatorzCallbackContext._from_estimatorIs6++c""%Y0@"EE!*!4!=%!+&9# JLL$L66 "(,%#' 94 5 5 + #7J    + + +)3)F)JG & &)*G &cp||}|j|_|j|_|j|_||_||_||_||_|j|_tj tj |_ d|_i|_d|_d|_|||S)avPrivate constructor to create a sub-context. Parameters ---------- parent_context : `CallbackContext` instance The parent context of the new context. task_name : str The name of the task this context is responsible for. task_id : int The identifier of the task this context is responsible for. max_subtasks : int or None The maximum number of tasks that can be children of the task this context is responsible for. 0 means it's a leaf. None means the maximum number of subtasks is not known in advance. sequential_subtasks : bool Whether this context's subtasks are sequential. If True, children contexts created via `subcontext` will have automatically assigned consecutive integer task_ids starting from 0. N)rrrr'rrrrrrrrr r!rr"r#r$ _add_child)r(parent_contextrrrrr*s r, _from_parentzCallbackContext._from_parents6++c""+6!/!>%3%F"%!+&9#*4$L66 "(,%#'  !!'***r.c#ZK|V|jD] }|Ed{V dS)z1Pre-order depth-first traversal of the task tree.N)r"values)selfcontexts r,__iter__zCallbackContext.__iter__sR )0022  G          r.cP|jjd|jd|jd|jdS)Nz(estimator_name=z , task_name=z , task_id=))rrrrr)r5s r,__repr__zCallbackContext.__repr__sN~& ' '"1 ' ' ' '| ' ' ' r.c 6|j|jvr(td|jd|jd|jd|jEt |j|jkr(td|jd|jd|jd||j|j<||_dS) z/Add `child_context` as a child of this context.zCallback context  of estimator z" already has a child with task_id=.Nz%Cannot add child to callback context z8 because it already has its maximum number of children (z).)rr" ValueErrorrrrlenr)r5 child_contexts r,r0zCallbackContext._add_childs  D$6 6 64DN44&44(0444    )D&''4+<<<===&=='+'8===  5B=01# r.c |jdkr0td|jd|jd|jd|jd |j|_|j|_|j|_||jj|j<|j|_|j|_ dS)aMerge this context with `other_context`. This method is called on a sub-estimator's root task to merge it with a meta-estimator's leaf task. The sub-estimator's task tree is therefore attached to the meta-estimator's task tree. The root node of the sub-estimator's task tree and the leaf node of the meta-estimator's task tree are both represented by a single node in this combined task tree. rz$Cannot merge callback context (task r<z) with callback context (task z#) because the latter is not a leaf.N) rr>rrrrrr"r$r#)r5 other_contexts r,r&zCallbackContext._merge_withs  % * *Ut~UU&UU&0UU!/UUU $* $, &0;? *4<8!. 7%2%A"""r.NrTc |jr%|#td|jd|d|jd|js%|#td|jd|d|jd|t |j}t |||||S)aCreate a context for a subtask of the current task. Parameters ---------- task_name : str, default="" The name of the subtask. task_id : int or None, default=None An identifier of the subtask. It must be distinct from the task_ids of its siblings. If None, task_id is automatically set to the next available integer task_id. max_subtasks : int or None, default=0 The maximum number of tasks that can be children of the subtask. 0 means it's a leaf. None means the maximum number of subtasks is not known in advance. sequential_subtasks : bool, default=True Whether the new context's subtasks are sequential. If True, children contexts of the new context, created via `subcontext`, will have automatically assigned consecutive integer task_ids starting from 0. Nz task_id for  z1 must be None if sequential_subtasks is True for r=z6 must be provided if sequential_subtasks is False for )r1rrrr)rr>rrr?r"r r2)r5rrrrs r, subcontextzCallbackContext.subcontexts2  # (;Et2EEYEE37>EEE ' GOIt2IIYII7;~III  ?$,--G++% 3 ,   r.c d}i}|jD]m}|t|dgvrtt||}d|jD}t |t t z x} r+td|d|jj d| dt d i} |D]} | |vr| d krK| d d } t| r | n| } | t|| nd } | |d <n6| | d }t|r |n|}||| <|| | | <|tt||||fi| z}o|S) aHelper to call the hook of all callbacks with their respective arguments. Provide the right arguments to each hook by inspecting their signatures. Any value that is a callable is replaced by what it returns to allow lazy loading of the arguments. Parameters ---------- estimator : estimator instance The estimator calling the callback hook. hook_name : str Name of the callback hook to call. **kwargs: dict Optional keyword arguments passed to the callback context. Returns ------- result : bool True if any hook call returned True. False otherwise. F_propagated_callbacksc<h|]}|j|jk|jS)kind KEYWORD_ONLYname).0ps r, z.CallbackContext._call_hooks..Es36Q^+++++r.zHook z of the callback z$ has parameters that are not valid: z. The valid parameters are: r=r reconstruction_attributesN)rr_cached_signature parametersr4setVALID_HOOK_PARAMS_OUT TypeErrorrrgetcallable_from_reconstruction_attributesbool)r5r) hook_namekwargsresultevaluated_argscallback signature params_namesdiff args_to_pass param_nameattrsnew_estvals r, _call_hookszCallbackContext._call_hooks!s., , H74)@"EEEE)'(I*F*FGGI"-4466L <((3/D+E+EEEt 5I558J8S55:>551555 L* F F ^33"%777 & +F M M+3E?? E %0E'9::$jjT::'/}}=cceee#58z2+9*+E Z(( d,),,YMM MM FF r.)rrr rQc<||d|||||S)aCall the `on_fit_task_begin` hook of the callbacks. Parameters ---------- estimator : estimator instance The estimator calling the callback hook. X : array-like or None, default=None The training data of the current task. y : array-like or None, default=None The training targets of the current task. metadata : dict or None, default=None A dictionary containing training metadata for the current task. reconstruction_attributes : dict or None, default=None A dictionary of the sufficient fitted attributes needed to construct a `fitted_estimator` from the current state of the estimator, i.e. an estimator instance ready to predict, transform, etc ... as if the fit had stopped at the beginning of this task. The `fitted_estimator` is the object that will be passed to the callbacks, if required. on_fit_task_beginr[rrr rQrhr5r)rrr rQs r,call_on_fit_task_beginz&CallbackContext.call_on_fit_task_beginms;@  )&?     r.c8||d||||S)adCall the `on_fit_task_end` hook of the callbacks. Parameters ---------- estimator : estimator instance The estimator calling the callback hook. X : array-like or None, default=None The training data of the current task. y : array-like or None, default=None The training targets of the current task. metadata : dict or None, default=None A dictionary containing training metadata of the current task. reconstruction_attributes : dict or None, default=None A dictionary of the sufficient fitted attributes needed to construct a `fitted_estimator` from the current state of the estimator, i.e. an estimator instance ready to predict, transform, etc ... as if the fit had stopped at the end of this task. The `fitted_estimator` is the object that will be passed to the callbacks, if required. Returns ------- stop : bool Whether or not to stop the current level of iterations at this end of this task. on_fit_task_endrkrlrms r,call_on_fit_task_endz$CallbackContext.call_on_fit_task_ends4L '&?     r.c #.Kdt|dgD}|r(td|jjdjd|d|_fdjDrz>CallbackContext.propagate_callback_context..s?   ($:;;   '   r.rzThe sub-estimator (z) of a meta-estimator (z() can't have auto-propagated callbacks (z0). Register them directly on the meta-estimator.clg|]0}t|tr|jj|jk.|1S)N)rtrmax_propagation_depthr')rNr_r5s r,ruz>CallbackContext.propagate_callback_context..sU" " " ($:;;" .6*X-KKK  LKKr. set_callbackszThe estimator z7 does not support callbacks. The callbacks attached to z* will not be propagated to this estimator.Ncg|]}|v| SrJrJ)rNcbcallbacks_to_propagates r,ruz>CallbackContext.propagate_callback_context..s/"""!777777r.) rrVrrrrrr%warningswarnrHrxr)r5 sub_estimator bad_callbackscurr_callbackskept_callbacksr{s` @r,propagate_callback_contextz*CallbackContext.propagate_callback_contexts$  #M3CRHH    Am&=&FAA$($7AA/<AAA .2 *" " " "  O" " "  " ('-*Q*Q ( M4!8!A448<8K444    &( " ! T)?D &$]4DbIIN 'M '.;Q*Q S S 3 EEE% =""""+:""" , +^<<222& =""""+:""" , +^<<2 2 2 2 2s C11#D)rCNrT)r __module__ __qualname____doc__ classmethodr-r2r7r:r0r&rFrhrnrqrrrJr.r,r r s511f77[7r--[-^    $$$,BBB:OS, , , , \JJJ`  "&(((((\  "&- - - - - ^E3E3^E3E3E3r.r ctj|}|D]\}}t||||S)aReturn a copy of the estimator as if it was fitted. Parameters ---------- estimator : estimator instance The estimator from which to make a ready-to-be-evaluated copy. reconstruction_attributes : dict A dictionary containing the necessary attributes to create a working fitted estimator from this instance. Returns ------- fitted_estimator : estimator instance The fitted copy of this estimator. )copyitemssetattr)r)rQ new_estimatorkeyrgs r,rYrYsM"Ii((M-3355))S sC(((( r.cF|j|gnt|j|gzS)aHelper function to get the path from the root context down to a given context. Parameters ---------- context : `CallbackContext` instance The context to get the path to. Returns ------- list of `CallbackContext` instances The list of the ancestors (itself included) of the given context. The list is ordered from the root context to the given context. )rget_context_path)r6s r,rr&s. > !  gn - - 9r.)r functoolsinspectrr| contextlibrrrsklearn.callback._baserrU lru_cacher`rRr rYrrJr.r,rs  %%%%%%''''''''999999CBB*'I'))'*;<<w3w3w3w3w3w3w3w3t.r.