§
    Øj™;  ã            
       ó  — d dl Z d dlZd dlZd dlmZmZ d dlmZm	Z	m
Z
 d dlmZ d dlmZ d dlmZ d dlmZmZ d dlmZ d d	lmZmZ d d
lmZ  ej        d ¬¦  «        Zd„ Zd„ Zd„ Zd„ Z d„ Z!d„ Z"ej#         $                    deed           ej%        ed         ed         dz  dz   f¦  «        j&        fg¦  «        d„ ¦   «         Z'ej#         $                    deed           ej%        ed         ed         dz  dz   f¦  «        j&        fg¦  «        d„ ¦   «         Z(d„ Z)d„ Z*d„ Z+ G d„ de	e¦  «        Z, G d„ de¦  «        Z-d„ Z. G d„ d e	e¦  «        Z/ej#         $                    d!d"d#g¦  «        d$„ ¦   «         Z0 G d%„ d&e¦  «        Z1d'„ Z2d(„ Z3 G d)„ d*e¦  «        Z4d+„ Z5ej#         $                    d,d-d.g¦  «        d/„ ¦   «         Z6 G d0„ d1e¦  «        Z7ej#         $                    d2ddg¦  «        d3„ ¦   «         Z8 ed#¬4¦  «        d5„ ¦   «         Z9dS )6é    N)Úconfig_contextÚdatasets)ÚBaseEstimatorÚTransformerMixinÚclone)ÚTransformedTargetRegressor)Úmake_regression)ÚDummyRegressor)ÚLinearRegressionÚOrthogonalMatchingPursuit)ÚPipeline)ÚFunctionTransformerÚStandardScaler©Úassert_allclose)Úrandom_statec                  óà  — t           \  } }t          t          ¦   «         t          ¦   «         t          j        t          j        ¬¦  «        }t          j        t          d¬¦  «        5  | 
                    | |¦  «         d d d ¦  «         n# 1 swxY w Y   t	          j        |j        d         f¦  «        }t          t          ¦   «         t          ¦   «         ¬¦  «        }t          j        t          d¬¦  «        5  | 
                    | ||¬¦  «         d d d ¦  «         n# 1 swxY w Y   t          t          j        ¬¦  «        }t          j        t          d	¬¦  «        5  | 
                    | |¦  «         d d d ¦  «         n# 1 swxY w Y   t          t          j        ¬
¦  «        }t          j        t          d¬¦  «        5  | 
                    | |¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )N)Ú	regressorÚtransformerÚfuncÚinverse_funczE'transformer' and functions 'func'/'inverse_func' cannot both be set.©Úmatchr   ©r   r   z:fit\(\) got an unexpected keyword argument 'sample_weight'©Úsample_weight)r   z=When 'func' is provided, 'inverse_func' must also be provided)r   z=When 'inverse_func' is provided, 'func' must also be provided)Úfriedmanr   r   r   ÚnpÚexpÚlogÚpytestÚraisesÚ
ValueErrorÚfitÚonesÚshaper   Ú	TypeError)ÚXÚyÚregrr   s       ú–/Users/tedpeters/Documents/Projects/Claude Projects/Misc Tools/notename-sheets/.venv/lib/python3.11/site-packages/sklearn/compose/tests/test_target.pyÚ%test_transform_target_regressor_errorr,      sÏ  € Ý�D€A€qå%Ý"Ñ$Ô$Ý"Ñ$Ô$ÝŒVÝ”Vð	ñ ô €Dõ 
ŒÝØUð
ñ 
ô 
ð ð ð 	�Š��A‰Œˆð	ð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð õ ”G˜QœW QœZ˜MÑ*Ô*€MÝ%Ý+Ñ-Ô-½>Ñ;KÔ;Kðñ ô €Dõ 
ŒÝØKð
ñ 
ô 
ð 4ð 4ð 	�Š��A ]ˆÑ3Ô3Ð3ð	4ð 4ð 4ñ 4ô 4ð 4ð 4ð 4ð 4ð 4ð 4øøøð 4ð 4ð 4ð 4õ &­2¬6Ð2Ñ2Ô2€DÝ	ŒÝØMð
ñ 
ô 
ð ð ð 	�Š��A‰Œˆð	ð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð õ &µ2´6Ð:Ñ:Ô:€DÝ	ŒÝØMð
ñ 
ô 
ð ð ð 	�Š��A‰Œˆð	ð ð ñ ô ð ð ð ð ð ð ð øøøð ð ð ð ð ð sH   Á%BÂBÂBÃ7DÄD Ä#D ÅE?Å?FÆFÆ?G#Ç#G'Ç*G'c                  óZ  — t           \  } }t          t          ¦   «         t          j        t          j        d¬¦  «        }t          j        t          d¬¦  «        5  | 	                    | |¦  «         d d d ¦  «         n# 1 swxY w Y   t          t          ¦   «         t          j        t          j        ¬¦  «        }| 
                    d¬¦  «         t          j        ¦   «         5  t          j        dt          ¦  «         | 	                    | |¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )	NT)r   r   r   Úcheck_inversez?The provided functions.* are not strictly inverse of each otherr   ©r   r   r   F)r.   Úerror)r   r   r   r   Úsqrtr    r!   ÚwarnsÚUserWarningr$   Ú
set_paramsÚwarningsÚcatch_warningsÚsimplefilter©r(   r)   r*   s      r+   Ú*test_transform_target_regressor_invertibler9   <   s�  € Ý�D€A€qÝ%Ý"Ñ$Ô$ÝŒWÝ”VØð	ñ ô €Dõ 
ŒÝØQð
ñ 
ô 
ð ð ð 	�Š��A‰Œˆð	ð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð õ
 &Ý"Ñ$Ô$­2¬7ÅÄðñ ô €Dð 	‡O‚O %€OÑ(Ô(Ð(å	Ô	 Ñ	"Ô	"ð ð ÝÔ˜g¥{Ñ3Ô3Ð3Ø�Š��A‰Œˆðð ð ñ ô ð ð ð ð ð ð ð øøøð ð ð ð ð ð s$   ÁA<Á<B ÂB Ã"1D Ä D$Ä'D$c                 óŠ   — t          j        | d¬¦  «        }t          j        | d¬¦  «        }t          | |z
  |z  |¦  «         d S )Nr   )Úaxis)r   ÚmeanÚstdr   )r)   Úy_predÚy_meanÚy_stds       r+   Ú_check_standard_scaledrA   S   sI   € ÝŒW�Q˜QÐÑÔ€FÝŒF�1˜1ÐÑÔ€EÝ�Q˜‘Z 5Ñ(¨&Ñ1Ô1Ð1Ð1Ð1ó    c                 ó,   — t          | dz   |¦  «         d S ©Né   r   )r)   r>   s     r+   Ú_check_shifted_by_onerF   Y   s   € Ý�A˜‘E˜6Ñ"Ô"Ð"Ð"Ð"rB   c            	      óº  — t           \  } }t          t          ¦   «         t          j        t          j        ¬¦  «        }|                     | |¦  «                             | ¦  «        }|j         	                    | 
                    dd¦  «        ¦  «                             ¦   «         }t          t          j        |¦  «        |¦  «         t          ||j                             | 
                    dd¦  «        ¦  «                             ¦   «         ¦  «         |j        |j        k    sJ ‚t          ||                     |j                             | ¦  «        ¦  «        ¦  «         t          ¦   «                              | |                     |¦  «        ¦  «        }t          |j        j                             ¦   «         |j                             ¦   «         ¦  «         d S )Nr/   éÿÿÿÿrE   )r   r   r   r   r    r   r$   ÚpredictÚtransformer_Ú	transformÚreshapeÚsqueezer   Úinverse_transformr&   r   Ú
regressor_r   Úcoef_Úravel©r(   r)   r*   r>   Úy_tranÚlrs         r+   Ú)test_transform_target_regressor_functionsrU   ]   sx  € Ý�D€A€qÝ%Ý"Ñ$Ô$­2¬6ÅÄðñ ô €Dð �XŠX�a˜‰^Œ^×#Ò# AÑ&Ô&€FàÔ×(Ò(¨¯ª°2°qÑ)9Ô)9Ñ:Ô:×BÒBÑDÔD€FÝ•B”F˜1‘I”I˜vÑ&Ô&Ð&ÝØ	ˆ4Ô×.Ò.¨v¯~ª~¸bÀ!Ñ/DÔ/DÑEÔE×MÒMÑOÔOñô ð ð Œ7�f”lÒ"Ð"Ð"Ð"Ý�F˜D×-Ò-¨d¬o×.EÒ.EÀaÑ.HÔ.HÑIÔIÑJÔJÐJå	Ñ	Ô	×	Ò	  4§9¢9¨Q¡<¤<Ñ	0Ô	0€BÝ�D”OÔ)×/Ò/Ñ1Ô1°2´8·>²>Ñ3CÔ3CÑDÔDÐDÐDÐDrB   c                  ó–  — t           d         } t          j        t           d         t           d         dz  dz   f¦  «        j        }t	          t          ¦   «         t          j        t          j        ¬¦  «        }|                     | |¦  «         	                    | ¦  «        }|j
                             |¦  «        }t          t          j        |¦  «        |¦  «         t          ||j
                             |¦  «        ¦  «         |j        |j        k    sJ ‚t          ||                     |j         	                    | ¦  «        ¦  «        ¦  «         t          ¦   «                              | |                     |¦  «        ¦  «        }t          |j        j                             ¦   «         |j                             ¦   «         ¦  «         d S )Nr   rE   é   r/   )r   r   ÚvstackÚTr   r   r    r   r$   rI   rJ   rK   r   rN   r&   r   rO   r   rP   rQ   rR   s         r+   Ú5test_transform_target_regressor_functions_multioutputrZ   p   sd  € Ý�Œ€AÝ
Œ	•8˜A”;¥¨¤¨qÑ 0°1Ñ 4Ð5Ñ6Ô6Ô8€AÝ%Ý"Ñ$Ô$­2¬6ÅÄðñ ô €Dð �XŠX�a˜‰^Œ^×#Ò# AÑ&Ô&€FàÔ×(Ò(¨Ñ+Ô+€FÝ•B”F˜1‘I”I˜vÑ&Ô&Ð&Ý�A�tÔ(×:Ò:¸6ÑBÔBÑCÔCÐCØŒ7�f”lÒ"Ð"Ð"Ð"Ý�F˜D×-Ò-¨d¬o×.EÒ.EÀaÑ.HÔ.HÑIÔIÑJÔJÐJå	Ñ	Ô	×	Ò	  4§9¢9¨Q¡<¤<Ñ	0Ô	0€BÝ�D”OÔ)×/Ò/Ñ1Ô1°2´8·>²>Ñ3CÔ3CÑDÔDÐDÐDÐDrB   zX,yrE   rW   c                 ó  — t          d„ d„ ¬¦  «        }t          t          ¦   «         |¬¦  «        }|                     | |¦  «                             | ¦  «        }|j        |j        k    sJ ‚|j                             |¦  «        }t          ||¦  «         |j        |j        k    sJ ‚t          ||j         
                    |¦  «                             ¦   «         ¦  «         t          ¦   «         }t          |¦  «        }|                     | |                     |¦  «        ¦  «         |                     | ¦  «        }t          || 
                    |¦  «        ¦  «         t          |j        j        |j        ¦  «         d S )Nc                 ó   — | dz   S rD   © ©Úxs    r+   ú<lambda>z@test_transform_target_regressor_1d_transformer.<locals>.<lambda>‹   s
   € �q˜1‘u€ rB   c                 ó   — | dz
  S rD   r]   r^   s    r+   r`   z@test_transform_target_regressor_1d_transformer.<locals>.<lambda>‹   s
   € °Q¸±U€ rB   ©r   r   r   )r   r   r   r$   rI   r&   rJ   rK   rF   r   rN   rM   r   Úfit_transformrO   rP   ©	r(   r)   r   r*   r>   rS   rT   Útransformer2Ú	y_lr_preds	            r+   Ú.test_transform_target_regressor_1d_transformerrg   ‚   sd  € õ &Øˆ_¨?¨?ðñ ô €Kõ &Ý"Ñ$Ô$°+ðñ ô €Dð �XŠX�a˜‰^Œ^×#Ò# AÑ&Ô&€FØŒ7�f”lÒ"Ð"Ð"Ð"àÔ×(Ò(¨Ñ+Ô+€FÝ˜!˜VÑ$Ô$Ð$ØŒ7�f”lÒ"Ð"Ð"Ð"å�A�tÔ(×:Ò:¸6ÑBÔB×JÒJÑLÔLÑMÔMÐMå	Ñ	Ô	€BÝ˜Ñ%Ô%€LØ‡F‚Fˆ1ˆl×(Ò(¨Ñ+Ô+Ñ,Ô,Ð,Ø—
’
˜1‘”€IÝ�F˜L×:Ò:¸9ÑEÔEÑFÔFÐFÝ�D”OÔ)¨2¬8Ñ4Ô4Ð4Ð4Ð4rB   c           	      óê  — t          ¦   «         }t          t          ¦   «         |¬¦  «        }|                     | |¦  «                             | ¦  «        }|j        |j        k    sJ ‚|j        dk    r/|j                             | 	                    dd¦  «        ¦  «        }n|j                             |¦  «        }t          ||                     ¦   «         ¦  «         |j        |j        k    sJ ‚t          ||j                             |¦  «                             ¦   «         ¦  «         t          ¦   «         }t          |¦  «        }|j        dk    r |                     | |                     | 	                    dd¦  «        ¦  «                             ¦   «         ¦  «         |                     | ¦  «         	                    dd¦  «        }|                     |¦  «                             ¦   «         }	nS|                     | |                     |¦  «        ¦  «         |                     | ¦  «        }|                     |¦  «        }	t          ||	¦  «         t          |j        j        |j        ¦  «         d S )Nr   rE   rH   )r   r   r   r$   rI   r&   ÚndimrJ   rK   rL   rA   rM   r   rN   r   rc   rO   rP   )
r(   r)   r   r*   r>   rS   rT   re   rf   Úy_pred2s
             r+   Ú.test_transform_target_regressor_2d_transformerrk   ¡   s  € õ !Ñ"Ô"€KÝ%Ý"Ñ$Ô$°+ðñ ô €Dð �XŠX�a˜‰^Œ^×#Ò# AÑ&Ô&€FØŒ7�f”lÒ"Ð"Ð"Ð"à„v�‚{€{ØÔ"×,Ò,¨Q¯YªY°r¸1Ñ-=Ô-=Ñ>Ô>ˆˆàÔ"×,Ò,¨QÑ/Ô/ˆÝ˜1˜fŸnšnÑ.Ô.Ñ/Ô/Ð/ØŒ7�f”lÒ"Ð"Ð"Ð"å�A�tÔ(×:Ò:¸6ÑBÔB×JÒJÑLÔLÑMÔMÐMå	Ñ	Ô	€BÝ˜Ñ%Ô%€LØ„v�‚{€{Ø
�Šˆq�,×,Ò,¨Q¯YªY°r¸1Ñ-=Ô-=Ñ>Ô>×FÒFÑHÔHÑIÔIÐIØ—J’J˜q‘M”M×)Ò)¨"¨aÑ0Ô0ˆ	Ø×0Ò0°Ñ;Ô;×CÒCÑEÔEˆˆà
�Šˆq�,×,Ò,¨QÑ/Ô/Ñ0Ô0Ð0Ø—J’J˜q‘M”Mˆ	Ø×0Ò0°Ñ;Ô;ˆå�F˜GÑ$Ô$Ð$Ý�D”OÔ)¨2¬8Ñ4Ô4Ð4Ð4Ð4rB   c                  ó€  — t           d         } t          j        t           d         t           d         dz  dz   f¦  «        j        }t	          ¦   «         }t          t          ¦   «         |¬¦  «        }|                     | |¦  «                             | ¦  «        }|j	        |j	        k    sJ ‚|j
                             |¦  «        }t          ||¦  «         |j	        |j	        k    sJ ‚t          ||j
                             |¦  «                             ¦   «         ¦  «         t          ¦   «         }t!          |¦  «        }|                     | |                     |¦  «        ¦  «         |                     | ¦  «        }t          ||                     |¦  «        ¦  «         t          |j        j        |j        ¦  «         d S )Nr   rE   rW   r   )r   r   rX   rY   r   r   r   r$   rI   r&   rJ   rK   rA   r   rN   rM   r   rc   rO   rP   rd   s	            r+   Ú:test_transform_target_regressor_2d_transformer_multioutputrm   Æ   s‚  € õ 	�Œ€AÝ
Œ	•8˜A”;¥¨¤¨qÑ 0°1Ñ 4Ð5Ñ6Ô6Ô8€AÝ Ñ"Ô"€KÝ%Ý"Ñ$Ô$°+ðñ ô €Dð �XŠX�a˜‰^Œ^×#Ò# AÑ&Ô&€FØŒ7�f”lÒ"Ð"Ð"Ð"àÔ×(Ò(¨Ñ+Ô+€FÝ˜1˜fÑ%Ô%Ð%ØŒ7�f”lÒ"Ð"Ð"Ð"å�A�tÔ(×:Ò:¸6ÑBÔB×JÒJÑLÔLÑMÔMÐMå	Ñ	Ô	€BÝ˜Ñ%Ô%€LØ‡F‚Fˆ1ˆl×(Ò(¨Ñ+Ô+Ñ,Ô,Ð,Ø—
’
˜1‘”€IÝ�F˜L×:Ò:¸9ÑEÔEÑFÔFÐFÝ�D”OÔ)¨2¬8Ñ4Ô4Ð4Ð4Ð4rB   c                  ól  — t           d         } t          j        t           d                              ddd¦  «        g d¢¦  «        }d„ }d„ }t	          ||¬¦  «        }t          t          ¦   «         |¬¦  «        }|                     | |¦  «                             | ¦  «        }|j	        |j	        k    sJ ‚d S )	Nr   rE   rH   )rE   é   rW   c                 óD   — |                       | j        d         d¦  «        S )Nr   rH   ©rL   r&   ©Údatas    r+   Úflatten_dataz?test_transform_target_regressor_3d_target.<locals>.flatten_dataç   s   € Ø�|Š|˜DœJ qœM¨2Ñ.Ô.Ð.rB   c                 óF   — |                       | j        d         dd¦  «        S )Nr   rH   rW   rq   rr   s    r+   Úunflatten_datazAtest_transform_target_regressor_3d_target.<locals>.unflatten_dataê   s   € Ø�|Š|˜DœJ qœM¨2¨qÑ1Ô1Ð1rB   rb   r   )
r   r   ÚtilerL   r   r   r   r$   rI   r&   )r(   r)   rt   rv   r   r*   r>   s          r+   Ú)test_transform_target_regressor_3d_targetrx   à   sÅ   € õ 	�Œ€AÝ
Œ•˜”×#Ò# B¨¨1Ñ-Ô-¨y¨y¨yÑ9Ô9€Að/ð /ð /ð2ð 2ð 2õ &¨<ÀnÐUÑUÔU€KÝ%Ý"Ñ$Ô$°+ðñ ô €Dð �XŠX�a˜‰^Œ^×#Ò# AÑ&Ô&€FØŒ7�f”lÒ"Ð"Ð"Ð"Ð"Ð"rB   c                  óÞ  — t           d         } t          j        t           d         t           d         dz  dz   g¦  «        }d„ }d„ }t          ||d¬¦  «        }|                     | |¦  «         |                     | ¦  «        }|j        dk    sJ ‚d	„ }t          ||d¬¦  «        }|                     | |¦  «         |                     | ¦  «        }|j        dk    sJ ‚t          ||¦  «         d S )
Nr   rE   rW   c                 ó�   — t          j        | d d …df         dz  | d d …df         dz  z   ¦  «        }|d d …t           j        f         S ©Nr   rW   rE   )r   r1   Únewaxis)r)   Úouts     r+   r   z=test_transform_target_regressor_multi_to_single.<locals>.funcù   sK   € ÝŒg�a˜˜˜˜1˜”g ‘l Q q q q¨! t¤W°¡\Ñ1Ñ2Ô2ˆØ�1�1�1•b”j�=Ô!Ð!rB   c                 ó   — | S ©Nr]   ©r)   s    r+   r   zEtest_transform_target_regressor_multi_to_single.<locals>.inverse_funcý   ó   € ØˆrB   F)r   r   r.   )éd   rE   c                 ód   — t          j        | d d …df         dz  | d d …df         dz  z   ¦  «        S r{   )r   r1   r€   s    r+   r   z=test_transform_target_regressor_multi_to_single.<locals>.func  s7   € ÝŒw�q˜˜˜˜A˜”w !‘| a¨¨¨¨1¨¤g°¡lÑ2Ñ3Ô3Ð3rB   )r   r   Ú	transposer   r$   rI   r&   r   )r(   r)   r   r   ÚttÚy_pred_2d_funcÚy_pred_1d_funcs          r+   Ú/test_transform_target_regressor_multi_to_singlerˆ   õ   s  € Ý�Œ€AÝ
Œ•h˜q”k¥H¨Q¤K°1Ñ$4°qÑ$8Ð:Ñ;Ô;€Að"ð "ð "ðð ð õ 
$Ø ¸Eð
ñ 
ô 
€Bð ‡F‚Fˆ1ˆa�L„L€LØ—Z’Z ‘]”]€NØÔ 8Ò+Ð+Ð+Ð+ð4ð 4ð 4õ 
$Ø ¸Eð
ñ 
ô 
€Bð ‡F‚Fˆ1ˆa�L„L€LØ—Z’Z ‘]”]€NØÔ 8Ò+Ð+Ð+Ð+å�N NÑ3Ô3Ð3Ð3Ð3rB   c                   ó"   — e Zd Zdd„Zd„ Zd„ ZdS )ÚDummyCheckerArrayTransformerNc                 ó>   — t          |t          j        ¦  «        sJ ‚| S r   ©Ú
isinstancer   Úndarray©Úselfr(   r)   s      r+   r$   z DummyCheckerArrayTransformer.fit  s   € Ý˜!�RœZÑ(Ô(Ð(Ð(Ð(ØˆrB   c                 ó>   — t          |t          j        ¦  «        sJ ‚|S r   rŒ   ©r�   r(   s     r+   rK   z&DummyCheckerArrayTransformer.transform  ó   € Ý˜!�RœZÑ(Ô(Ð(Ð(Ð(ØˆrB   c                 ó>   — t          |t          j        ¦  «        sJ ‚|S r   rŒ   r’   s     r+   rN   z.DummyCheckerArrayTransformer.inverse_transform  r“   rB   r   )Ú__name__Ú
__module__Ú__qualname__r$   rK   rN   r]   rB   r+   rŠ   rŠ     sF   € € € € € ðð ð ð ðð ð ðð ð ð ð rB   rŠ   c                   ó*   ‡ — e Zd Zdˆ fd„	Zˆ fd„Zˆ xZS )ÚDummyCheckerListRegressorNc                 óx   •— t          |t          ¦  «        sJ ‚t          ¦   «                              |||¦  «        S r   )r�   ÚlistÚsuperr$   )r�   r(   r)   r   Ú	__class__s       €r+   r$   zDummyCheckerListRegressor.fit$  s3   ø€ Ý˜!�TÑ"Ô"Ð"Ð"Ð"Ý‰wŒw�{Š{˜1˜a Ñ/Ô/Ð/rB   c                 ót   •— t          |t          ¦  «        sJ ‚t          ¦   «                              |¦  «        S r   )r�   r›   rœ   rI   )r�   r(   r�   s     €r+   rI   z!DummyCheckerListRegressor.predict(  s/   ø€ Ý˜!�TÑ"Ô"Ð"Ð"Ð"Ý‰wŒw�Š˜qÑ!Ô!Ð!rB   r   )r•   r–   r—   r$   rI   Ú__classcell__©r�   s   @r+   r™   r™   #  sV   ø€ € € € € ð0ð 0ð 0ð 0ð 0ð 0ð"ð "ð "ð "ð "ð "ð "ð "ð "rB   r™   c                  óp  — t           \  } }t          t          ¦   «         t          ¦   «         d¬¦  «        }|                     |                      ¦   «         |                     ¦   «         ¦  «         |                     |                      ¦   «         ¦  «         t          j        t          ¦  «        5  |                     | |                     ¦   «         ¦  «         d d d ¦  «         n# 1 swxY w Y   t          j        t          ¦  «        5  |                     | ¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )NF)r   r   r.   )
r   r   rŠ   r™   r$   ÚtolistrI   r!   r"   ÚAssertionError)r(   r)   r…   s      r+   Ú.test_transform_target_regressor_ensure_y_arrayr¤   -  sz  € õ �D€A€qÝ	#Ý0Ñ2Ô2Ý+Ñ-Ô-Øð
ñ 
ô 
€Bð
 ‡F‚Fˆ1�8Š8‰:Œ:�q—x’x‘z”zÑ"Ô"Ð"Ø‡J‚Jˆq�xŠx‰zŒzÑÔÐÝ	Œ•~Ñ	&Ô	&ð ð Ø
�Šˆq�!—(’(‘*”*ÑÔÐðð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð å	Œ•~Ñ	&Ô	&ð ð Ø
�
Š
�1‰Œˆðð ð ñ ô ð ð ð ð ð ð ð øøøð ð ð ð ð ð s$   Â/)C$Ã$C(Ã+C(ÄD+Ä+D/Ä2D/c                   ó.   — e Zd ZdZdd„Zd	d„Zd„ Zd„ ZdS )
ÚDummyTransformerz;Dummy transformer which count how many time fit was called.r   c                 ó   — || _         d S r   ©Úfit_counter)r�   r©   s     r+   Ú__init__zDummyTransformer.__init__B  s   € Ø&ˆÔÐÐrB   Nc                 ó&   — | xj         dz  c_         | S rD   r¨   r�   s      r+   r$   zDummyTransformer.fitE  s   € ØÐÔ˜AÑÐÔØˆrB   c                 ó   — |S r   r]   r’   s     r+   rK   zDummyTransformer.transformI  r�   rB   c                 ó   — |S r   r]   r’   s     r+   rN   z"DummyTransformer.inverse_transformL  r�   rB   )r   r   )r•   r–   r—   Ú__doc__rª   r$   rK   rN   r]   rB   r+   r¦   r¦   ?  s`   € € € € € ØEÐEð'ð 'ð 'ð 'ðð ð ð ðð ð ðð ð ð ð rB   r¦   r.   FTc                 ó¤   — t           \  }}t          t          ¦   «         | ¬¦  «        }|                     ||¦  «         |j        j        dk    sJ ‚d S )N)r   r.   rE   )r   r   r¦   r$   rJ   r©   )r.   r(   r)   Úttrs       r+   Ú)test_transform_target_regressor_count_fitr±   P  s\   € õ �D€A€qÝ
$Ý$Ñ&Ô&°mðñ ô €Cð ‡G‚GˆAˆq�M„M€MØÔÔ'¨1Ò,Ð,Ð,Ð,Ð,Ð,rB   c                   ó    ‡ — e Zd Zdˆ fd„	Zˆ xZS )Ú DummyRegressorWithExtraFitParamsNTc                 óR   •— |rJ ‚t          ¦   «                              |||¦  «        S r   )rœ   r$   )r�   r(   r)   r   Úcheck_inputr�   s        €r+   r$   z$DummyRegressorWithExtraFitParams.fit]  s*   ø€ ð ÐÐˆÝ‰wŒw�{Š{˜1˜a Ñ/Ô/Ð/rB   ©NT)r•   r–   r—   r$   rŸ   r    s   @r+   r³   r³   \  s=   ø€ € € € € ð0ð 0ð 0ð 0ð 0ð 0ð 0ð 0ð 0ð 0rB   r³   c                  óÀ   — t           \  } }t          t          ¦   «         t          ¦   «         ¬¦  «        }|                     | |d¬¦  «         |j        j        dk    sJ ‚d S )Nr   F©rµ   rE   )r   r   r³   r¦   r$   rJ   r©   r8   s      r+   Ú3test_transform_target_regressor_pass_fit_parametersr¹   d  sh   € Ý�D€A€qÝ%Ý2Ñ4Ô4ÕBRÑBTÔBTðñ ô €Dð 	‡H‚HˆQ�˜u€HÑ%Ô%Ð%ØÔÔ(¨AÒ-Ð-Ð-Ð-Ð-Ð-rB   c                  óø   — t           \  } }t          t          ¦   «         t          ¦   «         ¬¦  «        }dt	          ¦   «         fd|fg}t          |¦  «        } |j        | |fi ddi¤Ž |j        j        dk    sJ ‚d S )Nr   Ú	normalizeÚestÚest__check_inputFrE   )	r   r   r³   r¦   r   r   r$   rJ   r©   )r(   r)   r*   Ú
estimatorsÚpips        r+   Ú.test_transform_target_regressor_route_pipelinerÀ   n  s˜   € Ý�D€A€qå%Ý2Ñ4Ô4ÕBRÑBTÔBTðñ ô €Dð ¥Ñ 0Ô 0Ð1°E¸4°=ÐA€Jå
�:Ñ
Ô
€CØ€C„GˆAˆqÐ0Ð0Ð'¨Ð/Ð0Ð0Ð0àÔÔ(¨AÒ-Ð-Ð-Ð-Ð-Ð-rB   c                   ó    ‡ — e Zd Zdˆ fd„	Zˆ xZS )Ú$DummyRegressorWithExtraPredictParamsTc                 ó\   •— d| _         |rJ ‚t          ¦   «                              |¦  «        S r¶   )Úpredict_calledrœ   rI   )r�   r(   rµ   r�   s      €r+   rI   z,DummyRegressorWithExtraPredictParams.predict}  s.   ø€ ð #ˆÔØÐÐˆÝ‰wŒw�Š˜qÑ!Ô!Ð!rB   )T)r•   r–   r—   rI   rŸ   r    s   @r+   rÂ   rÂ   |  s=   ø€ € € € € ð"ð "ð "ð "ð "ð "ð "ð "ð "ð "rB   rÂ   c                  óâ   — t           \  } }t          t          ¦   «         t          ¦   «         ¬¦  «        }|                     | |¦  «         |                     | d¬¦  «         |j        j        sJ ‚d S )Nr   Fr¸   )r   r   rÂ   r¦   r$   rI   rO   rÄ   r8   s      r+   Ú=test_transform_target_regressor_pass_extra_predict_parametersrÆ   …  sq   € å�D€A€qÝ%Ý6Ñ8Ô8ÕFVÑFXÔFXðñ ô €Dð 	‡H‚HˆQ��N„N€NØ‡L‚L� €LÑ&Ô&Ð&ØŒ?Ô)Ð)Ð)Ð)Ð)Ð)rB   Úoutput_formatÚpandasÚpolarsc                 ó¼  — t          j        ¦   «         \  }}t          j        |¦  «        dz   }t	          | ¬¦  «        5  t          j        ¦   «         5  t          j        d¦  «         t          t          ¦   «         t          j
        t          j        ¬¦  «                             ||¦  «         ddd¦  «         n# 1 swxY w Y   ddd¦  «         dS # 1 swxY w Y   dS )z©Test that TransformedTargetRegressor will not raise warnings if
    set_config(transform_output="pandas"/"polars") is set globally; regression test for
    issue #29361.rE   )Útransform_outputr0   r/   N)r   r	   r   Úabsr   r5   r6   r7   r   r   r    r   r$   )rÇ   r(   r)   s      r+   Ú@test_transform_target_regressor_not_warns_with_global_output_setrÍ   ‘  sC  € õ
 Ô#Ñ%Ô%�D€A€qÝ
Œˆq‰	Œ	�A‰€AÝ	¨Ð	7Ñ	7Ô	7ð ð ÝÔ$Ñ&Ô&ð 	ð 	ÝÔ! 'Ñ*Ô*Ð*Ý&Ý*Ñ,Ô,µ2´6ÍÌðñ ô çŠc�!�Q‰iŒiˆið		ð 	ð 	ñ 	ô 	ð 	ð 	ð 	ð 	ð 	ð 	øøøð 	ð 	ð 	ð 	ðð ð ñ ô ð ð ð ð ð ð ð øøøð ð ð ð ð ð s6   ¾CÁAB9Â-CÂ9B=	Â=CÃ B=	ÃCÃCÃCc                   ó$   — e Zd ZdZd„ Zd„ Zd„ ZdS )ÚValidateDimensionRegressorzLA regressor that expects the target to have a specific number of dimensions.c                 ó   — || _         d S r   ©ri   )r�   ri   s     r+   rª   z#ValidateDimensionRegressor.__init__£  s   € ØˆŒ	ˆ	ˆ	rB   c                 ó*   — |j         | j         k    sJ ‚d S r   rÑ   r�   s      r+   r$   zValidateDimensionRegressor.fit¦  s   € ØŒv˜œÒ"Ð"Ð"Ð"Ð"Ð"rB   c                 ó   — d S r   r]   r’   s     r+   rI   z"ValidateDimensionRegressor.predict©  s   € ØˆrB   N)r•   r–   r—   r®   rª   r$   rI   r]   rB   r+   rÏ   rÏ      sG   € € € € € ØVÐVðð ð ð#ð #ð #ðð ð ð ð rB   rÏ   ri   c                 óØ   — t          j        ddd¬¦  «        \  }}| dk    r|                     dd¦  «        }t          t	          | ¦  «        ¬¦  «        }|                     ||¦  «         d	S )
z�Check that TransformedTargetRegressor internally preserves the shape of the input

    non-regression test for issue #26530.
    é
   é   é*   )Ú	n_samplesÚ
n_featuresr   rW   rH   rE   )r   N)r   r	   rL   r   rÏ   r$   )ri   r(   r)   r*   s       r+   Ú5test_transform_target_regressor_preserves_input_shaperÚ   ­  si   € õ Ô#¨b¸QÈRÐPÑPÔP�D€A€qØˆq‚y€yØ�IŠI�b˜!ÑÔˆå%Õ0JÈ4Ñ0PÔ0PÐQÑQÔQ€DØ‡H‚HˆQ��N„N€N€N€NrB   )Úenable_metadata_routingc                  ó²   — t          ¦   «         \  } }t          ¦   «         }|                     | |t          j        | j        d         f¬¦  «        ¬¦  «         dS )z:Test that metadata request is set on the default regressorr   )r&   r   N)r	   r   r$   r   r%   r&   )r(   r)   r°   s      r+   ÚBtest_transform_target_regressor_metadata_routing_default_estimatorrÝ   »  sQ   € õ ÑÔ�D€A€qÝ
$Ñ
&Ô
&€CØ‡G‚GˆAˆq¥¤¨q¬w°q¬z¨mÐ <Ñ <Ô <€GÑ=Ô=Ð=Ð=Ð=rB   ):r5   Únumpyr   r!   Úsklearnr   r   Úsklearn.baser   r   r   Úsklearn.composer   Úsklearn.datasetsr	   Úsklearn.dummyr
   Úsklearn.linear_modelr   r   Úsklearn.pipeliner   Úsklearn.preprocessingr   r   Úsklearn.utils._testingr   Úmake_friedman1r   r,   r9   rA   rF   rU   rZ   ÚmarkÚparametrizerX   rY   rg   rk   rm   rx   rˆ   rŠ   r™   r¤   r¦   r±   r³   r¹   rÀ   rÂ   rÆ   rÍ   rÏ   rÚ   rÝ   r]   rB   r+   ú<module>rë      sc  ðØ €€€à Ð Ð Ð Ø €€€à ,Ð ,Ð ,Ð ,Ð ,Ð ,Ð ,Ð ,Ø ?Ð ?Ð ?Ð ?Ð ?Ð ?Ð ?Ð ?Ð ?Ð ?Ø 6Ð 6Ð 6Ð 6Ð 6Ð 6Ø ,Ð ,Ð ,Ð ,Ð ,Ð ,Ø (Ð (Ð (Ð (Ð (Ð (Ø LÐ LÐ LÐ LÐ LÐ LÐ LÐ LØ %Ð %Ð %Ð %Ð %Ð %Ø EÐ EÐ EÐ EÐ EÐ EÐ EÐ EØ 2Ð 2Ð 2Ð 2Ð 2Ð 2à"ˆ8Ô"°Ð2Ñ2Ô2€ð&ð &ð &ðRð ð ð.2ð 2ð 2ð#ð #ð #ðEð Eð Eð&Eð Eð Eð$ „×ÒØ	ˆH�x ”{ I B¤I¨x¸¬{¸HÀQ¼KÈ1Ñ<LÈqÑ<PÐ.QÑ$RÔ$RÔ$TÐUÐVñô ð5ð 5ñô ð5ð8 „×ÒØ	ˆH�x ”{ I B¤I¨x¸¬{¸HÀQ¼KÈ1Ñ<LÈqÑ<PÐ.QÑ$RÔ$RÔ$TÐUÐVñô ð5ð 5ñô ð5ðD5ð 5ð 5ð4#ð #ð #ð*4ð 4ð 4ð@ð ð ð ð Ð#3°]ñ ô ð ð"ð "ð "ð "ð " ñ "ô "ð "ðð ð ð$ð ð ð ð Ð'¨ñ ô ð ð" „×Ò˜¨5°$¨-Ñ8Ô8ð-ð -ñ 9Ô8ð-ð0ð 0ð 0ð 0ð 0 ~ñ 0ô 0ð 0ð.ð .ð .ð.ð .ð .ð"ð "ð "ð "ð "¨>ñ "ô "ð "ð	*ð 	*ð 	*ð „×Ò˜¨8°XÐ*>Ñ?Ô?ðð ñ @Ô?ðð
ð 
ð 
ð 
ð 
 ñ 
ô 
ð 
ð „×Ò˜ ! Q Ñ(Ô(ð
ð 
ñ )Ô(ð
ð €¨Ð-Ñ-Ô-ð>ð >ñ .Ô-ð>ð >ð >rB   