+
    LV-jÛ(  ã                   ó¸   € ^ RI t^ RIHt ^RIHtHtHt ^RIH	t	 ^RI
Ht ^RIHt ^ RIHt ^ RIHtHtHt ^ RIHu Ht  ! R	 R
4      tR tRR ltR tRR ltR# )é    N)Ústats)Ú_get_pvalueÚ	_rankdataÚ_SimpleNormal)Ú
_morestats)Ú_broadcast_arrays)Ú_get_wilcoxon_distr)Ú_get_nan)Úarray_namespaceÚ
xp_promoteÚxp_sizec                   óV   a € ] tR t^t o R tR tR tR tR tR t	R t
R tR	 tR
tV tR# )ÚWilcoxonDistributionc                óÚ   € \         P                  ! V4      P                  \        R R7      pWn        \         P
                  ! V4       Uu/ uF  q"\        V4      bK  	  upV n        R# u upi )F©ÚcopyN)ÚnpÚasarrayÚastypeÚintÚnÚuniquer	   Ú_dists)Úselfr   Únis   && Úf/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/scipy/stats/_wilcoxon.pyÚ__init__ÚWilcoxonDistribution.__init__   sN   € Ü�JŠJ�q‹M× Ñ ¤¨5Ð Ó1ˆØŒÜ=?¿YºYÀq¼\ÓJ¹\°rÔ.¨rÓ2Ò2¹\ÑJˆŽùÒJs   Á
A(c                ó\   € V P                   V,          pVR V^,            P                  4       # ©N©r   Úsum©r   Úkr   Úpmfss   &&& r   Ú_cdf1ÚWilcoxonDistribution._cdf1   s(   € Ø�{‰{˜1�~ˆØ�F�Q˜•Uˆ|×ÑÓ!Ð!ó    c                ó\   € \         P                  ! V P                  \        .R 7      ! W4      # ©)Úotypes)r   Ú	vectorizer&   Úfloat©r   r$   r   s   &&&r   Ú_cdfÚWilcoxonDistribution._cdf   s   € Ü�|Š|˜DŸJ™J´¨wÕ7¸Ó=Ð=r(   c                óL   € V P                   V,          pW1R  P                  4       # r    r!   r#   s   &&& r   Ú_sf1ÚWilcoxonDistribution._sf1   s   € Ø�{‰{˜1�~ˆØ�Bˆx�|‰|‹~Ðr(   c                ó\   € \         P                  ! V P                  \        .R 7      ! W4      # r*   )r   r,   r2   r-   r.   s   &&&r   Ú_sfÚWilcoxonDistribution._sf   s   € Ü�|Š|˜DŸI™I¬u¨gÕ6°qÓ<Ð<r(   c                óX   € V P                   V P                   ^,           ,          ^,          # ©é   )r   )r   s   &r   ÚmeanÚWilcoxonDistribution.mean"   s   € Ø�v‰v˜Ÿ™ !�Õ$ qÕ(Ð(r(   c                óà   € \         P                  ! V4      P                  \        R R7      pV P	                  4       p\         P
                  ! VP                  \         P                  R7      pWV3# )Fr   ©Údtype)r   r   r   r   r:   ÚemptyÚshapeÚfloat64©r   r$   ÚmnÚouts   &&  r   Ú_prepÚWilcoxonDistribution._prep%   sL   € Ü�JŠJ�q‹M× Ñ ¤¨5Ð Ó1ˆØ�Y‰Y‹[ˆÜ�hŠh�q—w‘w¤b§j¡jÔ1ˆØ�cˆzÐr(   c                ó¤   a € S P                  V4      w  rp\        P                  ! W8*  VS P                  3S P                  V 3R  l4      R,          # )c                 óD   <€ ^SP                  V ^,           V4      ,
          # r8   )r5   ©r$   r   r   s   &&€r   Ú<lambda>Ú*WilcoxonDistribution.cdf.<locals>.<lambda>0   s   ø€ ˜˜TŸX™X a¨¥c¨1Ó-Ö-r(   © )rE   ÚxpxÚapply_wherer   r/   rB   s   f&  r   ÚcdfÚWilcoxonDistribution.cdf+   sH   ø€ Ø—Z‘Z “]‰
ˆˆsÜ�ŠØ‰G�a˜Ÿ™�[Ø�I‰IÜ-ó/ð 02õ3ð 	3r(   c                ó¤   a € S P                  V4      w  rp\        P                  ! W8*  VS P                  3S P                  V 3R  l4      R,          # )c                 óD   <€ ^SP                  V ^,
          V4      ,
          # r8   )r/   rI   s   &&€r   rJ   Ú)WilcoxonDistribution.sf.<locals>.<lambda>7   s   ø€ ˜˜TŸY™Y q¨¥s¨AÓ.Ö.r(   rL   )rE   rM   rN   r   r5   rB   s   f&  r   ÚsfÚWilcoxonDistribution.sf2   sH   ø€ Ø—Z‘Z “]‰
ˆˆsÜ�ŠØ‰G�a˜Ÿ™�[Ø�H‰HÜ.ó0ð 13õ4ð 	4r(   )r   r   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r&   r/   r2   r5   r:   rE   rO   rT   Ú__static_attributes__Ú__classdictcell__)Ú__classdict__s   @r   r   r      s8   ø‡ € òKò
"ò>òò=ò)òò3÷4ð 4r(   r   c           	      ó  € \        W4      p\        WR VR7      w  r\        P                  ! V4      R,          pRp\        P                  ! VP
                  \        P                  4      '       d   VP                  ^ 8w  d   \        V4      h\        V4      pRp\        \        RR4      ;'       g    \        P                  P                  p	 Vf   T p
M\        W3WgR7      w  rW,
          p
VP                  W¦R4      p
RpTe5   T P                  T,          TP                  T,          8w  d   \        T4      hRpTP!                  T
P
                  R	4      '       g   \        T4      h\#        T4      P%                  4       p0 RmpR
T R2pY,9  d   \        T4      hR R0pRT R2pY=9  d   \        T4      h\#        T4      P%                  4       p0 RmpRT R2pYN9  d   \        T4      h\'        T\(        P*                  4      '       g   0 RmpRT R2pY_9  d   \        T4      hTR8X  d   R MRpTP-                  T
^ 8H  RR7      pTR8X  d   T
P                  R,          ^28”  d   RpY¢Y4YVTTT3	#   T	 d   pT	! T4      ThRp?ii ; i)T)Úforce_floatingÚxpz`axis` must be an integer.z<`axis` must be compatible with the shape(s) of `x` (and `y`)Ú	AxisErrorN)Úaxisr_   z3`x` and `y` must have the same length along `axis`.z<`x` (and `y`, if provided) must be an array of real numbers.zreal floatingz`zero_method` must be one of Ú.Fz`correction` must be one of z`alternative` must be one of ÚautoÚ
asymptoticz`method` must be one of z- or an instance of `stats.PermutationMethod`.©ra   rL   éÿÿÿÿ>   ÚprattÚwilcoxÚzsplit>   ÚlessÚgreaterú	two-sided>   rc   Úexactrd   )r   r   r   r   Ú
issubdtyper>   ÚintegerÚndimÚ
ValueErrorr   ÚgetattrÚ
exceptionsr`   r   Úmoveaxisr@   ÚisdtypeÚstrÚlowerÚ
isinstancer   ÚPermutationMethodÚcount_nonzero)ÚxÚyÚzero_methodÚ
correctionÚalternativeÚmethodra   r_   Úmessager`   ÚdÚeÚzero_methodsÚcorrectionsÚalternativesÚmethodsÚoutput_zÚn_zeros   &&&&&&&           r   Ú_wilcoxon_ivrŠ   :   se  € Ü	˜Ó	€BÜ�a¨4°BÔ7�D€Aä�:Š:�dÓ˜BÕ€DØ*€GÜ�=Š=˜Ÿ™¤R§Z¡Z×0Ò0°D·I±IÀ´NÜ˜Ó!Ð!Üˆt‹9€DàL€GÜœ˜K¨Ó.×IÐI´"·-±-×2IÑ2I€Ið(ØŠ9Ø‰Aä$ a V°$Ô>‰DˆAØ•ˆAØ�K‰K˜ Ó$ˆð D€GØ‚}˜Ÿ™ �¨!¯'©'°$­-Ô7Ü˜Ó!Ð!àL€GØ�:‰:�a—g‘g˜×/Ò/Ü˜Ó!Ð!ä�kÓ"×(Ñ(Ó*€KÚ0€LØ-¨l¨^¸1Ð=€GØÔ&Ü˜Ó!Ð!à˜�-€KØ,¨[¨M¸Ð;€GØÔ$Ü˜Ó!Ð!ä�kÓ"×(Ñ(Ó*€KÚ3€LØ-¨l¨^¸1Ð=€GØÔ&Ü˜Ó!Ð!ä�fœe×5Ñ5×6Ò6Ú1ˆØ-¨g¨Yð 7?ð ?ˆàÔ Ü˜WÓ%Ð%Ø Ô-‰t°5€Hð
 ×Ñ˜a 1™f¨4ÐÓ0€FØ�Ô˜AŸG™G B�K¨"Ô,Øˆà˜:°FÀ(ÈFÐTVÐVÐVøðW ô (Ù˜Ó  aÐ'ûð(ús   Ã1I/ É/JÉ6	I?É?Jc                ó€  € V P                   pV ^ 8H  pVR8X  d9   \        P                  ! V 4      V,          P                  VP                  RR7      p VP                  V 4      pVP                  VRR7      pVP                  V P                  R,          V,
          V4      p\        VP                  V 4      RRVR7      w  ršVP                  W”RR7      VP                  W¤RR7      r©VP                  VP                  V ^ 8„  V4      V	,          RR7      pVP                  VP                  V ^ 8  V4      V	,          RR7      pVP                  V
^ 8H  4      pVR8X  d‰   VP                  VP                  WT4      V	,          RR7      ^,          p\        P                  ! V4      R	,          P                  V4      p\        P                  ! V4      R	,          P                  V4      pWˆR
,           ,          R,          pWˆR
,           ,          RV,          R
,           ,          pVR8X  Ed.   VP                  VP                  VRR7      V4      p\        P                  ! V4      R	,          P                  VVR
,           ,          R,          4      p\        P                  ! V4      R	,          P                  VVR
,           ,          RV,          R
,           ,          4      pVP                  V4      p\        P                  ! V4      R,          P                  VP                  VRR7      4      p\        P                  ! V
4      V,          P                  R4      p
VP                  V
^,          V
,
          RR7      pVP!                  VV^,          ,
          ^,          4      pVR9   d   W¿,
          V,          pMVP                  pW¼VVW�3# )r   rh   Tr   re   Úaverage)Úreturn_tiesr_   Fri   .g      ð?g      Ð?g       @rg   g        rf   ).r   )rd   rc   )r>   rM   ÚatÚsetÚnanÚisnanrz   r   r@   r   Úabsr"   ÚanyÚaddÚsubtractÚ
zeros_likeÚsqrt)r‚   r€   r}   r_   r>   Úi_zerosÚi_nanÚn_nanÚcountÚrÚtÚr_plusÚr_minusÚhas_tiesÚr_zero_2rC   Úser‰   Ú	t_i_zerosÚtie_correctÚzs   &&&$                 r   Ú_wilcoxon_statisticr¦   {   sò  € Ø�G‰G€EØ�A‰v€Gà�hÔô
 �FŠF�1‹I�gÕ×"Ñ" 2§6¡6°Ð"Ó5ˆà�H‰H�Q‹K€EØ×Ñ˜U¨ÐÓ,€EØ�I‰I�a—g‘g˜b•k EÕ)¨5Ó1€Eä�R—V‘V˜A“Y 	°tÀÔC�D€AØ�9‰9�Q Eˆ9Ó*¨B¯I©I°aÀU¨IÓ,K€qà�V‰V�B—I‘I˜a !™e UÓ+¨aÕ/°bˆVÓ9€FØ�f‰f�R—Y‘Y˜q 1™u eÓ,¨qÕ0°rˆfÓ:€Gà�v‰v�a˜1‘f‹~€Hà�hÔð —6‘6˜"Ÿ)™) GÓ3°aÕ7¸b�6ÓAÀAÕEˆÜ—’˜“ Õ$×(Ñ(¨Ó2ˆÜ—&’&˜“/ #Õ&×*Ñ*¨8Ó4ˆà	˜"•*Õ	 Õ	$€BØ	˜"•*Õ	  e¥¨b¥Õ	1€Bà�gÕð —‘˜2×+Ñ+¨G¸"Ð+Ó=¸uÓEˆÜ�VŠV�B‹Z˜�_×%Ñ% f°¸µÕ&<¸tÕ&CÓDˆÜ�VŠV�B‹Z˜�_×%Ñ% f°¸µÕ&<ÀÀVÅÈbÕ@PÕ&QÓRˆð
 —M‘M 'Ó*ˆ	Ü—F’F˜9Ó% fÕ-×1Ñ1°"·&±&¸Àr°&Ó2JÓKˆ	Ü�FŠF�1‹I�iÕ ×$Ñ$ RÓ(ˆà—&‘&˜˜A� �¨�&Ó+€KØ	�‰�"�{ 1•}Õ$¨Õ*Ó	+€Bð Ð'Ô'Ø�[˜BÕ‰à�F‰Fˆà˜B  5Ð2Ð2r(   c                 óH   € VR 8X  d   ^# VR8X  d   R# VP                  V 4      # )rk   rj   rf   )Úsign)r¥   r   r_   s   &&&r   Ú_correction_signr©   ¼   s(   € Ø�iÔÙØ	˜Ô	Øˆ	à�w‰w�q‹zÐr(   c           	      ó‚  aaa€ \        WSW4SV4      pVw	  por4orip
o\        V4      ^ 8X  d4   \        VSR7      p\        P                  ! W»R7      pSR8X  d   W¼n        V# \        VSSSR7      w  rÞppppSR8X  dC   V'       g   V
^ 8”  g   RoM0VP                  R,          ^8:  d   \        P                  ! 4       oMRoSR8X  dj   V'       dI   \        VVSR7      p\        P                  ! V4      R,          P                  VR,          V,          4      p\        V\        4       VSR7      pEM\SR8X  Ed   \!        V4      p\"        P$                  ! V4      pVR8X  d'   VP'                  \"        P(                  ! V4      4      pM©VR	8X  d'   VP+                  \"        P,                  ! V4      4      pM|^\"        P.                  ! VP+                  \"        P,                  ! V4      4      VP'                  \"        P(                  ! V4      4      4      ,          p\"        P0                  ! V^ ^4      pSP%                  VVP2                  R
7      pM?\        P4                  ! V3VVV3R l3RR/SP7                  4       BRVRR/B P8                  pVR8X  d   SP/                  WÞ4      MTpVR8X  d   SR8X  d   SP;                  V4      ) MTpVP<                  ^ 8X  d
   VR,          MTpVP<                  ^ 8X  d
   VR,          MTp\        P                  ! VVR7      pV	'       d"   VP<                  ^ 8X  d
   VR,          MTVn        V# )r   ©r_   )Ú	statisticÚpvaluerd   rc   rm   .g      à?rj   rk   r=   c                 ó0   <€ \        V SSSR 7      ^ ,          # )r«   )r¦   )r‚   r€   r_   r}   s   &€€€r   rJ   Ú_wilcoxon_nd.<locals>.<lambda>  s   ø€ Ô/°°6¸;È2ÔNÈqÖQr(   Úpermutation_typeÚsamplesr   ra   rl   rf   rL   )rŠ   r   r
   r   ÚWilcoxonResultÚ
zstatisticr¦   r@   r   ry   r©   rM   rŽ   r•   r   r   r   r   r   rO   ÚceilrT   ÚfloorÚminimumÚclipr>   Úpermutation_testÚ_asdictr­   r’   rp   )r{   r|   r}   r~   r   r€   ra   Útempr‚   rˆ   r‰   ÚNaNÚresrž   rŸ   r¢   r¥   r›   r    r¨   ÚpÚdistÚ	r_plus_npr¬   r_   s   &&f&&f&                 @r   Ú_wilcoxon_ndrÀ   Å   s¸  ú€ ô ˜˜k¨:ÀFÈDÓQ€DØRVÑO€A€{�J¨V°TÀVÈRäˆqƒz�Q„Ü�q˜RÔ ˆÜ×'Ò'°#ÔBˆØ�\Ô!Ø ŒNØˆ
ä.AØ	ˆ6�; 2ô/Ñ+€F�R˜˜E 8ð �Ôß˜F QœJØ‰FØ�W‰W�R�[˜BÔô
 ×,Ò,Ó.‰Fð "ˆFà�ÔßÜ# A {°rÔ:ˆDÜ—’�q“	˜#•×'Ñ'¨¨s­
°R­Ó8ˆAÜ˜œ=›?¨K¸BÔ?ŠØ	�7Õ	Ü# EÓ*ˆô —J’J˜vÓ&ˆ	Ø˜&Ô Ø—‘œŸš Ó+Ó,‰AØ˜IÔ%Ø—‘œŸš Ó+Ó,‰Aà”B—J’J˜tŸw™w¤r§x¢x°	Ó':Ó;Ø#Ÿx™x¬¯ª°	Ó(:Ó;ó=õ =ˆAä—’˜˜1˜aÓ ˆAØ�J‰J�q §¡ˆJÓ(‰ä×"Ò"ØˆDÖQñ.à&ð.à*0¯.©.Ó*:ñ.ð $ð.ð +-ò.÷ /5©fð 	
ð 0;¸KÔ/G�—
‘
˜6Ô+ÈV€IØ" kÔ1°fÀÔ6Lˆ�‰�‹‰
ÐST€Aà!*§¡°1Ô!4�	˜"–¸)€IØ—‘˜1”ˆˆ"Ž !€AÜ
×
#Ò
#¨iÀÔ
B€CßØ"#§&¡&¨A¤+˜˜2ž°1ˆŒØ€Jr(   )rh   )Nrh   Trl   rc   r   )Únumpyr   Úscipyr   Ú	_stats_pyr   r   r   Ú r   Ú_axis_nan_policyr   Ú
_hypotestsr	   Úscipy._lib._utilr
   Úscipy._lib._array_apir   r   r   Úscipy._lib.array_api_extraÚ_libÚarray_api_extrarM   r   rŠ   r¦   r©   rÀ   rL   r(   r   Ú<module>rÌ      sL   ðÛ å ß <Ñ <Ý Ý /Ý +Ý %ß FÑ Fß (Ð (÷*4ñ *4òZ>WôB>3òBöKr(   