
    -ji                    4   S SK Jr  S SKrS SKJr  S SKrS SKJr  \(       a  S SK	J
r
  SrSrSrS	rS
rSrSr\" \\\\\/5      rSrSrSrSrSr\" S\\\\\/5      rSrSrSrSrSrSrSr\" \\\\\\\/5      r Sr!Sr"Sr#Sr$Sr%Sr&\" S\!\"\#\$\%\&/5      r'S r(S!r)S"r*S#r+S$r,S%r-S&r.S'r/\" \(\)\*\+\,\-\.\//5      r0S(r1S)r2S*r3S+r4S,r5S-r6S.r7S/r8\" S\1\2\3\4\5\6\7\8/	5      r9S0r:S1r;S2r<S3r=S4r>S5r?S6r@\" \:\;\<\=\>\?\@/5      rAS7rBS8rCS9rDS:rES;rFS<rGS=rH\" S\B\C\D\E\F\G\H/5      rIS@S> jrJS@S? jrKg)A    )annotationsN)TYPE_CHECKING)
Polynomial)Callableg   `
?gin?ghn?gi}ԿgOQ*ghf#wg g	yw?g6UL?gk"t?g"]!?g aBCо   g8uYcgM ?gø@׿g Q_?g(>=cg땗Y6*?g?
8ag#f>?g3oJ?g\c?g5c`&?gQk?g5WT?g5d`4g`s4g&A%g
0Ogf"(Ldg\ggWeRTg\#gvp3@g!jR5a@gq){@gh(!+@gpW!z@g,H([@gJHG@gbt﮿gJo94gޅpgZ_U	1gC]dg(ugY.jag?8ܛN8~gQ&V>@g
"\t@gQ@gj$h@g;@gct}@gb-qBp6c                f   [        U R                  5      S:X  d   S5       e[        R                  " U S/S/S/S//:  SS9n[        R                  " U 5      nS	 S
 S S S /n[        U5       H?  u  pE[        R                  " X:H  5      S   =nR                  (       d  M3  U" X   5      X&'   MA     U$ )Nr   zInput must be a 1D array.g      0>g      ?g      ?gm۶m@r   )axisc                    S[         -   U -  $ Nr   )efx)xs    S/home/james-whalen/.local/lib/python3.13/site-packages/optuna/samplers/_tpe/_erf.py<lambda>$_erf_right_non_big.<locals>.<lambda>v   s    1s7a-    c                D    U S[        X -  =n5      [        U5      -  -   -  $ r   )ppqq)r   zs     r   r   r   w   s!    !q215ja>BqE112r   c                H    [         [        U S-
  =n5      [        U5      -  -   $ r   )erxpaqa)r   ss     r   r   r   x   s    #A:1A..r   c                    S[         R                  " X -  =n* S-
  [        SU-  =n5      [        U5      -  -   5      U -  -
  $ Nr   g      ?)npexprasar   r   r   s      r   r   r   {   ?    !bffAEzq]V3ba!enr!u6LLMPQQQr   c                    S[         R                  " X -  =n* S-
  [        SU-  =n5      [        U5      -  -   5      U -  -
  $ r   )r   r   rbsbr!   s      r   r   r   |   r"   r   )lenshaper   count_nonzero
empty_like	enumeratenonzerosize)r   bin_indsouterf_approx_in_each_binbin_idxerf_approx_in_bintarget_indss          r   _erf_right_non_bigr3   p   s    qww<1999vh	D6H:%N NUVWH
--
C2. 	RQH '00F&G"::h&9:1==KCCC0@C 'H Jr   c                p   U R                   S:  ab  [        R                  " U R                  5        Vs/ s H  n[        R
                  " U5      PM     sn5      R                  U R                  5      $ [        R                  " U 5      R                  5       n[        R                  " U5      ) n[        R                  " US[        R                  5      n[        R                  " X2S:  -  5      S   n[        X%   5      XE'   [        R                  " U 5      UR                  U R                  5      -  $ s  snf )Ni  g      ?   r   )r,   r   asarrayravelmatherfreshaper'   absisnanwherenanr+   r3   sign)r   va
is_not_nanr.   non_big_indss         r   r9   r9      s    vv}zz	:	1488A;	:;CCAGGLL
q	A((1+J
((:sBFF
+C::jE23A6L*1?;C771:AGG,,, ;s    D3)r   
np.ndarrayreturnrD   )L
__future__r   r8   typingr   numpyr   numpy.polynomialr   collections.abcr   r   r   pp0pp1pp2pp3pp4r   qq1qq2qq3qq4qq5r   pa0pa1pa2pa3pa4pa5pa6r   qa1qa2qa3qa4qa5qa6r   ra0ra1ra2ra3ra4ra5ra6ra7r   sa1sa2sa3sa4sa5sa6sa7sa8r    rb0rb1rb2rb3rb4rb5rb6r$   sb1sb2sb3sb4sb5sb6sb7r%   r3   r9    r   r   <module>r      s  " #     ' ( ! ! !!!!!c3S)*    !Cc3,- " ! ! !c3S#s34      Cc3S12 "!      c3S#sC89!Cc3S#s;< "!     c3S#s34 Cc3S#67*	-r   