Q(hI$ddlZddlmZddlmZddlZddlmZddl m Z m Z m Z m Z mZmZddlmZmZddlmZddlmZdd lmZdd lmZmZmZdd lmZd,d Ze fd Z!eedZ"d-dZ#dZ$Gdde%Z&Gdde e Z'Gdde e Z(Gdde e Z)Gdde e Z*Gddee Z+Gdde Z,Gd d!eZ-Gd"d#eeZ.Gd$d%e e e Z/Gd&d'e e e Z0Gd(d)e e e Z1Gd*d+e ee Z2y).N) defaultdict)partial)assert_array_equal) BaseEstimatorClassifierMixinMetaEstimatorMixinRegressorMixinTransformerMixinclone)_Scorermean_squared_error)BaseCrossValidator)GroupsConsumerMixin)SIMPLE_METHODS)MetadataRouter MethodMappingprocess_routing)_check_partial_fit_first_callc dtj}|dj}|dj}t|dst d|_|s8|j Dcic]\}}t|tr|dk7r||}}}|j ||j|ycc}}w)zUtility function to store passed metadata to a method of obj. If record_default is False, kwargs whose values are "default" are skipped. This is so that checks on keyword arguments whose default was not changed are skipped. _recordsc ttSN)rlistc/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/sklearn/tests/metadata_routing_common.pyz!record_metadata..*s ;t+<rdefaultN) inspectstackfunctionhasattrrritems isinstancestrappend)objrecord_defaultkwargsr"calleecallerkeyvals rrecord_metadatar0s MMOE 1X  F 1X  F 3 #"#<= #LLN Sc3'C9,< H  LL ''/  s%"B,c ^t|dtj|tj|t}|D]}t |j t |j k(s)Jd|j d|j |j D]r\}}||} ||vr)| 'tj| |jr6Jt| tjr t| |_| |urdJd| d|d|y)aCheck whether the expected metadata is passed to the object's method. Parameters ---------- obj : estimator object sub-estimator to check routed params for method : str sub-estimator's method where metadata is routed to, or otherwise in the context of metadata routing referred to as 'callee' parent : str the parent method which should have called `method`, or otherwise in the context of metadata routing referred to as 'caller' split_params : tuple, default=empty specifies any parameters which are to be checked as being a subset of the original values **kwargs : dict passed metadata rz Expected z vs Nz . Method: ) getattrdictgetrsetkeysr%npisinallr&ndarrayr) r)methodparent split_paramsr+ all_recordsrecordr.valuerecorded_values rcheck_recorded_metadatarB4s2( Z(,,VTV<@@PQ6;;=!S KKM&   : v{{}oT&++- 9 : !,,. QJC#C[Nl"~'Aww~u599;;;nbjj9&~u='%/Q">"2$ugZxPQ/ Q QrF)r*c Xt|tr0|D]*\}}| ||vr||}nd}t|j|,y|gn|}tD]U}||vrt ||}|j jDcgc]\}}t|ts||} }}| sUJycc}}w)aCheck if a metadata request dict is empty. One can exclude a method or a list of methods from the check using the ``exclude`` parameter. If metadata_request is a MetadataRouter, then ``exclude`` can be of the form ``{"object" : [method, ...]}``. N)exclude) r&rassert_request_is_emptyrouterrr2requestsr%r') metadata_requestrDname route_mapping_excluder;mmrpropaliaspropss rrErEbs"N3#3 L D-"tw"4= #M$8$8( K  L ObG  W  &/ #||113 e%%):   y  s>B&c|jD]"\}}t||}|j|k(r"JtDcgc] }||vs| }}|D]#}t t||js#Jycc}wr)r%r2rGrlen)request dictionaryr;rGrL empty_methodss rassert_request_equalrUs&,,.(gv&||x'''(+9UF*eZdZdZd dZdZd dZdZdZdZ d Z y) NonConsumingClassifier5A classifier which accepts no metadata on any method.c||_yr)alpha)rYrs rrizNonConsumingClassifier.__init__s  rcntj||_tj||_|Sr)r7uniqueclasses_ ones_likecoef_rYrsrts rrxzNonConsumingClassifier.fits% !  \\!_  rNc|Srr)rYrsrtclassess rruz"NonConsumingClassifier.partial_fitr\rc$|j|Sr)r~rYrss rdecision_functionz(NonConsumingClassifier.decision_functions||Arctjt|f}d|dt|dzd|t|dzd|S)Nrzrrr)r7emptyrQ)rYrsy_preds rr~zNonConsumingClassifier.predictsCQ * !}Q1 !s1v{} rctjt|df}tjddg|dt|dzddf<tjddg|t|dzdddf<|S)Nrrz?)r7rrQasarray)rYrsy_probas r predict_probaz$NonConsumingClassifier.predict_probask((#a&!-$&JJSz$: #a&A+ q !$&JJSz$:A! q !rc$|j|Sr)rrs rpredict_log_probaz(NonConsumingClassifier.predict_log_probas!!!$$r)rr) r`rarbrrirxrurr~rrrrrrrs(?  %rrc"eZdZdZdZdZdZy)NonConsumingRegressorrc|Srrrs rrxzNonConsumingRegressor.fitr\rc|Srrrs rruz!NonConsumingRegressor.partial_fitr\rc>tjt|Sr)r7onesrQrs rr~zNonConsumingRegressor.predictswws1vrN)r`rarbrrxrur~rrrrrs?rrcReZdZdZd dZ d dZd dZd dZd dZd dZ d d Z d d Z y)ConsumingClassifieraA classifier consuming metadata. Parameters ---------- registry : list, default=None If a list, the estimator will append itself to the list in order to have a reference to the estimator later on. Since that reference is not required in all tests, registration can be skipped by leaving this value as None. alpha : float, default=0 This parameter is only used to test the ``*SearchCV`` objects, and doesn't do anything. Nc ||_||_yr)rrg)rYrgrs rrizConsumingClassifier.__init__s   rc|j|jj|t|||t|||Srl)rgr(rqr)rYrsrtrrnros rruzConsumingClassifier.partial_fits@ == $ MM  &#    &dG4 rc|j|jj|t|||tj||_tj ||_|Srl)rgr(rqr7rrrrrrs rrxzConsumingClassifier.fitsR == $ MM  &#    !  \\!_  rct|||tjt|fd}d|t|dzdd|dt|dz|S)Nrmint8)r{dtyperrrrqr7rrQrYrsrnroy_scores rr~zConsumingClassifier.predict sV#   ((#a&&9!"A! !" #a&A+rct|||tjt|df}tjddg|dt|dzddf<tjddg|t|dzdddf<|S)Nrmrrzrr)rqr7rrQr)rYrsrnrors rrz!ConsumingClassifier.predict_proba)s|#   ((#a&!-$&JJSz$: #a&A+ q !$&JJSz$:A! q !rc`t|||tjt|dfS)Nrmrrzr|rYrsrnros rrz%ConsumingClassifier.predict_log_proba2s*#   xxs1vqk**rct|||tjt|f}d|t|dzdd|dt|dz|S)Nrmrzrrrrrs rrz%ConsumingClassifier.decision_function8sT#   ((#a&+!"A! !" #a&A+rc t|||yrrrrs rrzConsumingClassifier.scoreArr)Nrrr) r`rarbrrirurxr~rrrrrrrrrs6 ! EN  + rrc8eZdZdZddZd dZd dZd dZd dZy) ConsumingTransformera~A transformer which accepts metadata on fit and transform. Parameters ---------- registry : list, default=None If a list, the estimator will append itself to the list in order to have a reference to the estimator later on. Since that reference is not required in all tests, registration can be skipped by leaving this value as None. Nc||_yrrfrhs rrizConsumingTransformer.__init__Trjrc~|j|jj|t|||d|_|S)NrmT)rgr(rqfitted_rrs rrxzConsumingTransformer.fitWs; == $ MM  &#     rc(t||||dzSrrrs r transformzConsumingTransformer.transforma#   1u rclt||||j||||j|||Srl)rqrxrrrs r fit_transformz"ConsumingTransformer.fit_transformgsF $   xx1MHxMWW ]XX  rc(t||||dz Srrrs rinverse_transformz&ConsumingTransformer.inverse_transformsrrrrrNN) r`rarbrrirxrrrrrrrrHs  !  rrc(eZdZdZddZddZddZy) "ConsumingNoFitTransformTransformerzA metadata consuming transformer that doesn't inherit from TransformerMixin, and thus doesn't implement `fit_transform`. Note that TransformerMixin's `fit_transform` doesn't route metadata to `transform`.Nc||_yrrfrhs rriz+ConsumingNoFitTransformTransformer.__init__rjrcp|j|jj|t||||Srl)rgr(r0rrs rrxz&ConsumingNoFitTransformTransformer.fits/ == $ MM  &MHM rc"t||||Srl)r0rs rrz,ConsumingNoFitTransformTransformer.transformsMHMrrNNNr)r`rarbrrirxrrrrrrzsQ!rrc*eZdZdfd ZfdZxZS)ConsumingScorercBt|tdid||_y)Nrr~) score_funcsignr+response_method)superrir rg)rYrg __class__s rrizConsumingScorer.__init__s' )"i  ! rc |j|jj|t|fi||jdd}t||||||S)Nrnrn)rgr(rqr4r_score)rY method_callerclfrsrtr+rnrs rrzConsumingScorer._scoresU == $ MM  &#D3F3 ?D9 w~mS!Qm~TTrr)r`rarbrir __classcell__)rs@rrrs! UUrrc,eZdZddZddZddZd dZy) ConsumingSplitterNc||_yrrfrhs rrizConsumingSplitter.__init__rjrc#K|j|jj|t|||t|dz}t t d|}t t |t|}||f||fyw)N)groupsrorr)rgr(rqrQrrange)rYrsrtrro split_index train_indices test_indicess rsplitzConsumingSplitter.splitsw == $ MM  &#D(K!fk U1k23 E+s1v67 M))\))sBBcy)Nrr)rYrsrtrros r get_n_splitszConsumingSplitter.get_n_splitssrc#Kt|dz}ttd|}tt|t|}||yw)Nrr)rQrr)rYrsrtrrrrs r_iter_test_indicesz$ConsumingSplitter._iter_test_indicessE!fk U1k23 E+s1v67 sA A rr)NNNNr)r`rarbrirrrrrrrrs! *rrc"eZdZdZdZdZdZy) MetaRegressorz(A meta-regressor which is only a router.c||_yr) estimator)rYrs rrizMetaRegressor.__init__s "rc t|dfi|}t|jj||fi|jj|_yNrx)rr rrx estimator_rYrsrt fit_paramsparamss rrxzMetaRegressor.fitsC u; ;3%/33AqQFBBnn(?..eE.JC  rNr`rarbrrirxrrrrrrs2#Rrrc,eZdZdZddZddZdZdZy)WeightedMetaRegressorz*A meta-regressor which is also a consumer.Nc ||_||_yrrrgrYrrgs rrizWeightedMetaRegressor.__init__"  rc |j|jj|t||t|dfd|i|}t |j j ||fi|j j |_|SNrrxrnrgr(r0rr rrxr)rYrsrtrnrrs rrxzWeightedMetaRegressor.fitss == $ MM  &M: uXMXZX3%/33AqQF {G6FG*t  **1M0B0B0L0LMMrct|jjj|jt jddjddS)Nrrxrr)rr)rrr`rrrr^s rrz$MetaTransformer.get_metadata_routingsTDNN$;$;<@@(((? SeS , S KS 8 A  rr)r`rarbrrirxrrrrrrr s$' N rr)Tr)3r! collectionsr functoolsrnumpyr7 numpy.testingr sklearn.baserrrr r r sklearn.metrics._scorerr r sklearn.model_selectionrsklearn.model_selection._splitr sklearn.utils._metadata_requestsrsklearn.utils.metadata_routingrrrsklearn.utils.multiclassrr0tuplerBrqrErUrrWrdrrrrrrrrrr rrrrr$s5#,@6> C0,?Dg(QV&oeL::  ++\ %_m %F NM Q/=Qh/+]/d*UgU"+-?6& $. D/-8 (*:M r