Q3jUddlZddlZddlZddlZddlZddlmZddlmZmZddl m Z gdZ ejejZGddZdZd Zy) N)contextmanager)datetimetimezone)AutoPropagatedCallback)Xymetadatafitted_estimatorceZdZdZedZedZdZdZdZ dZ dd Z d Z ddddd d Z ddddd d ZedZy)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|j|}t|dg|_|jj|_||_||_||_||_ d|_ tj|_ tjt j"|_i|_d|_d|_t-|dr3|j.}|j1||j2dz|_|Sd|_|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 O/DATA/.local/lib/python3.12/site-packages/sklearn/callback/_callback_context.py_from_estimatorzCallbackContext._from_estimatorIs6++c"%Y0@"E!*!4!4!=!=%!+&9# JJL$LL6 "(,%#' 94 5 #77J    +)3)F)F)JG &*+G &c|j|}|j|_|j|_|j|_||_||_||_||_|j|_tjtj|_ d|_i|_d|_d|_|j%||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"+66!/!>!>%3%F%F"%!+&9#*44$LL6 "(,%#'  !!'*r.c#jK||jjD] }|Ed{y7w)z1Pre-order depth-first traversal of the task tree.N)r"values)selfcontexts r,__iter__zCallbackContext.__iter__s0 ))002G  3 s '313c|jjd|jd|jd|jdS)Nz(estimator_name=z , task_name=z , task_id=))rrrrr)r5s r,__repr__zCallbackContext.__repr__sL~~&&'("1145*+||nA ' r.c |j|jvr3td|jd|jd|jd|j Ut |j|j k\r3td|jd|jd|j d||j|j<||_y) 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 6#DNN#3>&&'((0014     )D&&'4+<+<<77G~&&'(''+'8'8&9=  5B=001# r.c |jdk7r@td|jd|jd|jd|jd |j|_|j |_|j |_||jj|j <|j|_|j|_ y)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  % % *6t~~6H&&'(&003> //00SU $** $,, &00;? **4<<8!. 7 7%2%A%A"r.Nc @|jr+|)td|jd|d|jd|js+|)td|jd|d|jd|t |j }t j|||||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. z 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  # #(;t2231YK@337>>2B!E ''GOt2231YK@77;~~6FaI  ?$,,-G++% 3 ,  r.c d}i}|jD]`}|t|dgvrtt||}|jj Dchc]'}|j |j k(r |j)} }t| ttz x} r0td|d|jjd| dtd i} | D]z} | |vrl| dk(r<|jd d } t| r| n| } | t|| nd }||d<n+|j| d }t|r|n|}||| <|| | | <||t!t||||fi| z}c|Scc}w) 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_callbackszHook z of the callback z$ has parameters that are not valid: z. The valid parameters are: r=r reconstruction_attributesN)rr_cached_signature parametersr4kind KEYWORD_ONLYnamesetVALID_HOOK_PARAMS_OUT TypeErrorrrgetcallable_from_reconstruction_attributesbool)r5r) hook_namekwargsresultevaluated_argscallback signaturep params_namesdiff args_to_pass param_nameattrsnew_estvals r, _call_hookszCallbackContext._call_hooks!s.H74)@"EE)'(I*FGI#--4466A66Q^^+6  <(3/D+EEEtEI;&78J8J8S8S7TU::>@12!5 L* ^3"%77 & +F M+3E? %0E'9:$jjT:'/}ce#58z2+9*+E Z('+* d,),YM M FU(\ Ms,E7)rrr rHc4|j|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_beginrUrrr rHrcr5r)rrr rHs r,call_on_fit_task_beginz&CallbackContext.call_on_fit_task_beginms1@  )&?   r.c0|j|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_endrfrgrhs r,call_on_fit_task_endz$CallbackContext.call_on_fit_task_ends/L '&?    r.c #~Kt|dgDcgc](}t|tr|jj*}}|r3t d|jjd|j d|d||_|jDcgc]9}t|tr'|j|j|jkr|;}}|rHt|dsAA .2 *!OO" +($:;..6**X-K-KK + "  "'-*Q MM !8!8!A!A BC88<8K8K7LM34  &( " !)?D &$]4DbIN 'M ' '.;Q*Q S 3 %,::":!77:" , ++^<2i (" 4"&,::":!77:"" , ++^<2sMF=-E3A F= >E8 A4F=>FF= E=#F=F: F! F::F=)NrT)r __module__ __qualname____doc__ classmethodr-r2r7r:r0r&rErcrirlrryr.r,r r s1f77r--^  $,B:OS, \J`  "&(\  "&- ^E3E3r.r cztj|}|jD]\}}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)rH new_estimatorkeyrbs r,rSrSs;"IIi(M-335S sC(6 r.cR|j|gSt|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&s4 >> !  gnn - 9r.)r functoolsinspectrrq contextlibrrrsklearn.callback._baserrO lru_cacherZrIr rSrrr.r,rs`  %'9C*'I'')'*;*;<w3w3t.r.