
    > j?n                        S SK r S SKrS SKJr  S SKJr  S SKJr  S SKr	S SK
JrJrJrJrJrJrJrJrJr  S SKJs  Jr  SSKJr  SSKJr  SS	KJr  \ " S
 S5      5       rSS.S jrS rS rS r S r!S r"S r#S r$\" SSS/S9r%/ \" \RL                  S\" SS/SS9SSS9P\" \RN                  S\" SS/SS9SSS9P\" \RP                  S\" SS/SS9SSS9P\" \RR                  S\%SS9P\" \RT                  S\%\"SS9P\" \RV                  S\" SS/SS9SSSS 9P\" \RX                  S!\" SSS/S9SS0SS"9P\" \RZ                  S!\" SS/SS9SS9P\" \R\                  S!\" SS/SS9SS9P\" \R^                  S!\" SS/SS9SS9P\" \R`                  S\" SS/SS9SS9P\" \Rb                  S!\ S#9P\" \Rd                  S!\!SSSS$.S%9P\" \Rf                  S!\" SS/SS9SS9P\" \Rh                  S\" SS/SS9SSS&9P\" \Rj                  S\" SS/SS9SS9P\" \Rl                  S\" SS/SS9SS9P\" \Rn                  S\" SS/SS9SS9P\" \Rp                  S\" SS/SS9SS9P\" \Rr                  S5      P\" \Rt                  S5      P\" \Rv                  S5      P\" \Rx                  S\" SSS'/SS9SS9P\" \Rz                  S5      P\" \R|                  S\" SS/SS9SS9P\" \R~                  S\" SS/SS9SS9P\" \R                  S\" SS/SS9SS9P\" \R                  S\" SS/SS9SS9P\" \R                  S\" SSS/S9SS9P\" \R                  S5      P\" \R                  S!\" SSS/S9SS(SSS)9P\" \R                  S\" SS/SS9SS9P\" \R                  S\" SS/SS9SS9P\" \R                  S\" SS/SS9SS9P\" \R                  S\" SSS/S9SS9P\" \R                  S!5      P\" \R                  S!SS0S*9P\" \R                  S!\" SS/SS9SS9P\" \R                  S!\" SS/SS9SS9P\" \R                  S5      P\" \R                  S\" SSS/S9SS9P\" \R                  S\" SS/SS9SS9P\" \R                  S\" SS/SS9SS9P\" \R                  S!\" SS/SS9SS9P\" \R                  S\" SS/S9SS0SSS+9P\" \R                  S!\" SS/SS9SS9P\" \R                  S!\" SS/SS9SS9P\" \R                  S5      P\" \R                  S5      P\" \R                  S5      P\" \R                  S5      P\" \R                  S\" SSS'/SS9S'S,0SS-9P\" \R                  S\" SSS'/SS9S'S.0SS-9P\" \R                  S\" SSS'/SS9S'S/0S9P\" \R                  S\" SSS'/SS9S'S00SS-9P\" \R                  S\" SSS'/SS9S'S10S9P\" \R                  S\" SSS'/SS9S'S20SS-9P\" \R                  S!\" SSS/S9SS9P\" \R                  S5      P\" \R                  S\" SS/SS9SS9P\" \R                  S5      P\" \R                  S\" SS/SS9SSS39P\" \R                  S!SSS/SS/SS/S$.SS/SS/SS/SS/S4.S5SSSS69	P\" \R                  S\" SS/SS9S7SSS89P\" \R                  S\" SS/SS9S7SSS89P\" \R                  S\" SS/SS9S7SSS89P\" \R                  S5      P\" \R                  S5      P\" \R                  S!\" SS/SS9SSS"9P\" \R                  S!\" SS/SS9SSS"9P\" \R                  S!\" SS/SS9S(SSS89P\" \R                  S!\" SSS/S9SSS&9P\" \R                  S!\" SS/SS9SS9P\" \R                  S!S(SS'S(0SS(SS9.SS:9P\" \R                  SSS;0S9P\" \R                  S\" SS/SS9SS9P\" \R                  S!\S#9P\" \R                  S\" SS/SS9SS9P\" \R                  S\" SSS'/SS9SS<9P\" \R                  S\" SS/SS9SSS&9P\" \R                  S\" SS/S9SS9P\" \R                  S!\%\#SS9P\" \R                  S!\" SS/S=/S>9\$SS9P\" \R                  S\" SS/SS9SSS&9P\" \R                  S!5      P\" \R                  S!\S#9P\" \R                  S\" SSS'/SS9S'S?0SS-9P\" \R                  S\" SSS'/SS9S'S@0SS-9P\" \R                  S!\" SS/SS9SS(0S(SSS)9P\" \R                  S!SSS(SA.SSB9P\" \GR                   S\" SS/SS9SS9P7r\" 5       GR                  \ V s0 s H&  o GR                  GR
                  U GR                  _M(     sn 5        \pr\ V s/ s H  o GR                  GR
                  PM     sn S;/-   rgs  sn f s  sn f )C    N)Callable)	dataclass)
ModuleType)	array_namespacescipy_namespace_foris_numpyis_dask	is_marray
xp_promotexp_capabilitiesSCIPY_ARRAY_APIget_native_namespace_name   )_basic)_spfun_stats)_ufuncsc                      \ rS rSr% \\S'   \\S'   Sr\\/\4   S-  \S'   Sr\\	\	S-  /\S-  4   S-  \S'   Sr
\\\4   S-  \S'   Sr\\   S-  \S'   S	r\\\   -  \\\\   4   -  \S
'   Sr\\S'   Sr\\\\   4   \\   -  S-  \S'   Sr\\\\   4   \\   -  S-  \S'   Sr\\S'   S	r\\S'   Sr\\S'   \S 5       rS rS r\S 5       r\R:                  " S5      S 5       rSrg)	_FuncInfo   funcn_argsNr   generic_implalt_names_mapint_onlyFpositive_onlyTis_ufuncpython_int_onlyscalar_or_0d_onlytest_large_intsproduces_0dtorch_nativec                 .    U R                   R                  $ N)r   __name__selfs    |/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/scipy/special/_support_alternative_backends.pyname_FuncInfo.nameK   s    yy!!!    c                 ,    [        U R                  5      $ r#   )hashr   r%   s    r'   __hash___FuncInfo.__hash__P   s    DIIr*   c                 b    [        U[        5      =(       a    U R                  UR                  :H  $ r#   )
isinstancer   r   )r&   others     r'   __eq___FuncInfo.__eq__S   s!    %+G		UZZ0GGr*   c                   ^  T R                   [        5       ;   a$  SS Kn[        UR                  T R                   5      $ [
        (       aD  [        R                  " T R                  5      U 4S j5       nSUl	        T R                   Ul
        UnOT R                  nT R                  =(       d
    [        5       nU" U5      nXSL d   eU$ )Nr   c                  B   > [        U 6 nTR                  U5      " U 0 UD6$ r#   )r   _wrapper_for)argskwargsxpr&   s      r'   wrapped"_FuncInfo.wrapper.<locals>.wrapped_   s(    $d+((,d=f==r*   zscipy.special)r(   globalsscipy.specialgetattrspecialr   	functoolswrapsr   
__module____qualname__r   )r&   scipyr:   r   capabilitiescap_funcs   `     r'   wrapper_FuncInfo.wrapperV   s    99	! !5==$))44?__TYY'> (> "1G#'99G D99D++@/@  %r*   i  c                    [        U5      (       a  U R                  $ [        U5      n[        XU R                  U R
                  S9nUb  U$ U R                  b  U R                  X5      nUb  U$ [        U5      (       a   [        5       U R                     nXAS.S jnU$ [        U5      (       a   [        5       U R                     nXAS.S jnU$ U R                  nXAS.S jnU$ )Nr   )_fr9   c                     U Vs/ s H  n[        USU5      PM     nnU " U0 UD6n[        R                  " [        R                  S U 5       5      nUR                  XgS9$ s  snf )Ndatac              3   <   #    U  H  n[        US S5      v   M     g7f)maskFN)r>   ).0args     r'   	<genexpr>4_FuncInfo._wrapper_for.<locals>.f.<locals>.<genexpr>   s     (UPTfe)D)DPTs   )rO   )r>   r@   reduceoperatoror_asarray)rK   r9   r7   r8   rQ   	data_argsoutrO   s           r'   f!_FuncInfo._wrapper_for.<locals>.f   se    BFG$3WS&#6$	G).v. ''(UPT(UWzz#z11	 Hs   A!c                 R    UR                   " [        R                  " U 40 UD6/UQ76 $ r#   )
map_blocksr@   partial)rK   r9   r7   r8   s       r'   rZ   r[      s&    }}Y%6%6r%DV%DLtLLr*   c                     U Vs/ s H  n[         R                  " U5      PM     nnU " U0 UD6nUR                  U5      $ s  snf r#   )nprW   )rK   r9   r7   r8   rQ   rY   s         r'   rZ   r[      sB     044tBJJsOtD4d%f%C::c?" 5s    A )
r   r   r   _get_native_funcr(   r   r   r
   r<   r	   )r&   r9   spxrZ   rK   s        r'   r6   _FuncInfo._wrapper_fors   s    B<<99 ""%Rdiit?Q?QR=H (!!"*A}R== 499%B 2 H2;; 499%B M H YY 	# r*    ) r$   rB   rC   __firstlineno__r   __annotations__intr   r   r   r   dictstrr   tupleboolr   r   r   r   r   r    r!   propertyr(   r-   r2   rG   r@   	lru_cacher6   __static_attributes__rd   r*   r'   r   r      s    N K >BOXxj(23d:A  (	Z$&'D8 
 ,0M4S>D(/#'HeDkD ' BGM4%+%S%+-=(>>F Hd DHOT#uT{*+eDk9D@GEItCt,-d;dBI OT K L$" "H  8 8 8r*   r   rJ   c                    Uc  0 nUR                  [        U 5      U5      nU(       a  [        UR                  US 5      OS nUc(  [	        U S5      (       a  [        U R                  US 5      nU$ )Nr?   )getr   r>   r?   hasattr)r9   rb   f_namer   rZ   s        r'   ra   ra      se    8<fEF.1VT*tAyWR++ BJJ-Hr*   c                     U S.S jnU$ )Nr9   c                H  ^ [        U5      (       a   [        U R                  UR                  5      OUm[        XSSUS9u  pU S:  US:  -  nUR	                  U 5      UR	                  U5      -  n[
        R                  " X4) -  X4U4S jUR                  S9n[
        R                  " U5      U S:H  US:  -     R                  S5      n[
        R                  " U5      UR                  U 5      UR                  U5      -  X4-  -     R                  UR                  5      nU$ )NT)	broadcastforce_floatingr9   r   c                 P   > U TR                  U 5      TR                  U5      -
  -  $ r#   )log)xymxps     r'   <lambda>/_rel_entr.<locals>.__rel_entr.<locals>.<lambda>   s    cggaj3771:56r*   )
fill_value)r	   r   _metar   isinfxpxapply_whereinfatsetisnannan)rz   r{   r9   xy_posxy_infresr|   s         @r'   
__rel_entr_rel_entr.<locals>.__rel_entr   s    3:2;;oaggqww/B!$tKa%AE"!rxx{*ooWF6vv
 ffSk16a1f-.2215ffSk"((1+3vGHLLRVVT
r*   rd   )r9   rb   r   s      r'   	_rel_entrr      s    ! " r*   c                     U S.S jnU$ )Nrt   c                    [        XSUS9u  p[        R                  " SSS9   XR                  U5      -  nS S S 5        UR	                  U S:H  SW5      $ ! , (       d  f       N$= f)NT)rw   r9   ignore)divideinvalidg        )r   r`   errstatery   where)rz   r{   r9   temps       r'   __xlogy_xlogy.<locals>.__xlogy   sT    !t;[[(;vvay=D <xxRT** <;s   A
A#rd   )r9   rb   r   s      r'   _xlogyr      s     +
 Nr*   c                 :   ^ ^ [        T US5      mTc  g UU 4S jnU$ )Ngammaincc                    > T" U S-  US-  5      nU S:H  US:  -  nTR                  USU5      nTR                  U 5      TR                  U5      -  nTR                  UTR                  U5      $ )N   r   g      ?)r   r   r   )vrz   r   rO   r   r9   s       r'   __chdtr_chdtr.<locals>.__chdtr   sl    q1ua!e$Q1q5!hhtR%xx{RXXa[(xxbffc**r*   ra   )r9   rb   r   r   s   `  @r'   _chdtrr      s(    
  C4H+ Nr*   c                 :   ^ ^ [        T US5      mTc  g UU 4S jnU$ )N	gammainccc                    > TR                  US:  T" U S-  US-  5      S5      nUS:H  U S:H  -  TR                  U5      -  TR                  U 5      -  U S:*  -  nTR                  UTR                  U5      nU$ )Nr   r   r   )r   r   r   )r   rz   r   i_nanr   r9   s       r'   __chdtrc_chdtrc.<locals>.__chdtrc   sx    hhqAvy1ac2A6q&Q!V$3bhhqkAQ!VLhhubffc*
r*   r   )r9   rb   r   r   s   `  @r'   _chdtrcr      s(    
 !S+6I
 Or*   c                 4   ^ [        XS5      mTc  g U4S jnU$ )Nbetaincc                    > T" XSU-
  5      $ )Nr   rd   )abrz   r   s      r'   
__betaincc_betaincc.<locals>.__betaincc  s    qQqS!!r*   r   )r9   rb   r   r   s      @r'   	_betainccr      s$    r	2G" r*   c                 :   ^ ^ [        T US5      mTc  g UU 4S jnU$ )Nr   c                 j   > XS-  U -   -  nT" U S-  SU5      S-  nTR                  US:  USU-
  5      $ )Nr   g      ?r   r   )r   )dftrz   tailr   r9   s       r'   __stdtr_stdtr.<locals>.__stdtr  sE    q&2+rAvsA&*xxAtQX..r*   r   )r9   rb   r   r   s   `  @r'   _stdtrr   	  s%    r3	2G/
 Nr*   c                 x   ^ ^^^ [        T US5      =(       d    [        T U5      mTc  g SSKJmJm  UUUU 4S jnU$ )Nstdtrr   )bracket_root	find_rootc                 |   > U4S jnT" UTR                  U5      X4S9nT" X#R                  X4S9nUR                  $ )Nc                    > T" X5      U-
  $ r#   rd   )r   r   pr   s      r'   fun(_stdtrit.<locals>.__stdtrit.<locals>.fun"  s    5<!#33r*   )r7   )
zeros_likebracketrz   )	r   r   r   res_bracketres_rootr   r   r   r9   s	        r'   	__stdtrit_stdtrit.<locals>.__stdtrit!  s=    3"3a(8wGS"5"5RGDzzr*   )ra   r   scipy.optimize.elementwiser   r   )r9   rb   r   r   r   r   s   `  @@@r'   _stdtritr     s;    Rg.A&S/E }B  r*   T	jax.numpycupy)cpu_only
exceptions   F)r   r   jax_jit)FTF)r   r!   )r!   )r   r!   )r   r   r!   r   )r   r!   )r   )r   r   torch)r   r   )r   r!   r   )TF)r   r   r   r!   )r   )r   r   r!   	bessel_j0)r   r   	bessel_j1modified_bessel_k0scaled_modified_bessel_k0modified_bessel_k1scaled_modified_bessel_k1)r!   r   )array_api_strictnumpyz
dask.arraymarray)FT)r   r   r   r   r   r   r!   )TTF)r   r   r!   )r   r   )r   r   r   r    r   r   digamma)r   )r   zno scipy.optimize support)r   r   skip_backends	bessel_y0	bessel_y1)r   r   )r   r   r   )r@   rU   collections.abcr   dataclassesr   typesr   r   r`   scipy._lib._array_apir   r   r   r	   r
   r   r   r   r   scipy._lib.array_api_extra_libarray_api_extrar    r   r   r   r   ra   r   r   r   r   r   r   r   _needs_betaincbdtrbdtrcbdtrir   betaincc
betaincinvbetalnbinomboxcoxboxcox1pcbrtchdtrchdtrcchdtricosdgcosm1cotdgellipkellipkm1entrerferfcerfcxerfinvexp1exp10exp2exprelexpiexpitexpnfdtrfdtrcfdtrigammar   r   gammainccinvgammaincinvgammalngammasgngdtrgdtrchuberhyp1f1
inv_boxcoxinv_boxcox1pi0i0ei1i1ej0j1k0k0ek1k1ekl_divlog_ndtrloggammalogitlpmvmultigammalnnbdtrnbdtrcnbdtrindtrndtripdtrpdtrcpdtripochpseudo_huber	polygammapsiradianrel_entrrgammasincsindgspencer   stdtrittandgxlog1pyxlogyy0y1ynzetazetac_special_funcsr<   updater   r$   rG   r   __all__)nfos   0r'   <module>rC     s>     $ !     ) (    Y Y Yx 8< 	*&"
2 !$K;PQqavh	
 &Eq qvh	
 &Eq" qvh	
 &E#q2 gooq.uE3q4 g>	 "5q8 Avh	
 T9qH 6;2GH #D)IqV qvh	
 Wqf vh	
 gqv !vh	
 wqF avh	
 GqV gmmQV4WqX gnnag%)tLNYq^ vh	
 _qn qvh	
 Eoq~ qvh	
 qN qvh	
 Oq^ vh	
 _qn !vh	
 oq~ gllAq@ gkk1AqB gllACqD qvw&7	
 EqT gnna UqV avh	
 Wqf qvh	
 gqv avh	
 wqF vh	
 GqV a6;2GHWq` gmmQaqb a6;2GH cqt avh	
 uqD qvh	
 EqT qvh	
 Uqd q6;2GHeqn g"oqp 1"D)qqz avh	
 {qJ Qvh	
 KqZ gooq![q\ !6;2GH]qf avh	
 gqv qvh	
 wqF qvh	
 GqV ;-@"D)5	Wqr Avh	
 sqB	 avh	
 C	qR	 gjj!S	qT	 gkk1U	qV	 gjj!W	qX	 gkk1Y	qZ	 

Avw&7	
 ,e[	qj	 

Avw&7	
 ,ek	qz	 

Avw&7	
  45{	qJ
 Qvw&7	
  ;<K
q\
 

Avw&7	
  45]
ql
 Qvw&7	
  ;<m
q|
 6;2GH}
qF g"GqH !vh	
 IqX gmmQYqZ avh	
 [ql !!1DMT]
 "'T] $-dm	
 mqN qvh	
 %DOq` vh	
 %Daqr vh	
 %DsqD gllAEqF gmmQGqH avh	
 IqX qvh	
 Yqh qvh	
 diqz a6;2GHE{qD avh	
 EqT !me!(- 8d&3$G#	Uq` gkk1[),DEaqb vh	
 cqr g	:sqt vh	
 uqD Qvw&7	
 EqT qvh	
 EUqd ;-@eqn gmmaf "oqr vhEF	
 EsqB qvh	
 ECqR gooq!SqT gmmQV4UqV 

Avw&7	
 ,eWqf 

Avw&7	
 ,egqv 

Avh	
 }-EwqH Q$(=AIqR qvh	
 Sql 	  NKNS((##S[[0NK L (6
788
79+
E L
 8s   -g+!g0