4[gHUdZddlmZddlZddlZddlmZddlmZm Z m Z ddl Z ddl mZddlmZddlmZmZm ZmZgdZej.j1d d Zd ed <ej.j1d d ZeedZe reZeej<zZd+dZd,dZ d-dZ! d.dd d d d/dZ"ddd0dZ#ddd1dZ$d2dZ%d2dZ&d2dZ'dZ( d3dZ)dZ*dZ+ d4dZ, d5dZ- d6d Z.ddd1d!Z/d7d"Z0d8d#Z1d9d$Z2d:d%Z3d;d&Z4 dd(Z6ddd?d)Z7ddd1d*Z8y)@a6Utility functions to use Python Array API compatible libraries. For the context about the Array API see: https://data-apis.org/array-api/latest/purpose_and_scope.html The SciPy use case of the Array API is described on the following page: https://data-apis.org/array-api/latest/use_cases.html#use-case-scipy ) annotationsN) ModuleType)AnyLiteral TYPE_CHECKING)array_api_compat)is_array_api_objsizenumpydevice)array_namespace_asarrayr r SCIPY_ARRAY_APIFz str | bool SCIPY_DEVICEcpu)rrcTtt|D]w}||}t|tjj r t dt|tjr t dt|tjtjfre|j}tj|tjrtj|tjrt d|dt|r tj|}|j}tj|tjs5tj|tjsd|d}t ||||<z|S#t $r t dwxYw)aRaise exceptions on known-bad subclasses. The following subclasses are not supported and raise and error: - `numpy.ma.MaskedArray` - `numpy.matrix` - NumPy arrays which do not have a boolean or numerical dtype - Any array-like which is neither array API compatible nor coercible by NumPy - Any array-like which is coerced by NumPy to an unsupported dtype z8Inputs of type `numpy.ma.MaskedArray` are not supported.z0Inputs of type `numpy.matrix` are not supported.zAn argument has dtype `z3`; only boolean and numerical dtypes are supported.zCAn argument is neither array API compatible nor coercible by NumPy.z1An argument was coerced to an unsupported dtype `)rangelen isinstancenpma MaskedArray TypeErrormatrixndarraygenericdtype issubdtypenumberbool_r asanyarray)arraysiarrayrmessages P/var/www/html/bid-api/venv/lib/python3.12/site-packages/scipy/_lib/_array_api.pycompliance_scipyr'/sb3v; q  eRUU.. /VW W ryy )NO O ebjj"**5 6KKEMM%3r}}UBHH7U"9%CS!TUU!%( 7 e,KKEMM%3r}}UBHH7UGyQGH ((F1I1 2 M 7!677 7s FF'cd} |j|j|s t|y#t$r t|wxYw)zCheck for NaNs or Infs.z#array must not contain infs or NaNsN)allisfinite ValueErrorr)r$xpmsgs r& _check_finiter.UsI /Cvvbkk%()S/ !* os +0ActdstS|Dcgc]}|| }}t|}tj|Scc}w)a0Get the array API compatible namespace for the arrays xs. Parameters ---------- *arrays : sequence of array_like Arrays used to infer the common namespace. Returns ------- namespace : module Common namespace. Notes ----- Thin wrapper around `array_api_compat.array_namespace`. 1. Check for the global switch: SCIPY_ARRAY_API. This can also be accessed dynamically through ``_GLOBAL_CONFIG['SCIPY_ARRAY_API']``. 2. `compliance_scipy` raise exceptions on known-bad subclasses. See its definition for more details. When the global switch is False, it defaults to the `numpy` namespace. In that case, there is no compliance check. This is a convenience to ease the adoption. Otherwise, arrays must comply with the new rules. r)_GLOBAL_CONFIG np_compatr'rr )r"r$_arrayss r&r r _sL4 + ,"(>&E,=u&G>w'G  + +W 55 ?s??)r, check_finitesubokc| t|}|jdvrc|durtj||||}n3|rtj|||}ntj |||}|j |}n |j |||}|r t|||S#t $r1t|j d}|j |||}YIwxYw)a`SciPy-specific replacement for `np.asarray` with `order`, `check_finite`, and `subok`. Memory layout parameter `order` is not exposed in the Array API standard. `order` is only enforced if the input array implementation is NumPy based, otherwise `order` is just silently ignored. `check_finite` is also not a keyword in the array API standard; included here for convenience rather than that having to be a separate function call inside SciPy functions. `subok` is included to allow this function to preserve the behaviour of `np.asanyarray` for NumPy based inputs. >r !scipy._lib.array_api_compat.numpyT)orderrr4)r7r)rcopy)r __name__rr$r!asarrayrr.)r$rr7r8r,r3r4 coerced_xps r&rrs0 z U # {{DD 4<HHU%uEJE MM%uEBEJJuE?E 5! FJJuEJ=E eR L F(A7J&&uE&EE FsB$$7CCr,c| t|}|j|}|j|kr!|j|d}t |||}|S)z9Recursively expand the dimension to have at least `ndim`.raxisndimr,)r r;rB expand_dims atleast_nd)xrBr,s r&rDrDsN z Q  1 Avv} NN11N % qt + Hc8| t|}t|d|S)a3 Copies an array. Parameters ---------- x : array xp : array_namespace Returns ------- copy : array Copied array Notes ----- This copy function does not offer all the semantics of `np.copy`, i.e. the `subok` and `order` keywords are not used. T)r8r,)r rrEr,s r&r8r8s"* z Q  ADR ((rFc|jdvS)N)r r6r:r=s r&is_numpyrK ;;H HHrFc|jdvS)N)cupyz scipy._lib.array_api_compat.cupyrJr=s r&is_cupyrOs ;;F FFrFc|jdvS)N)torchz!scipy._lib.array_api_compat.torchrJr=s r&is_torchrRrLrFc|jdvS)N)z jax.numpyzjax.experimental.array_apirJr=s r&is_jaxrTs ;;E EErFcd}|r t|||j|}|rd7m_ UD KK BKK$8KK' G0DMHLM F0D%9rFc Dd}| t|}t||||||}t|r|jj |||St |r+|dk(rdn|}|jj ||dddd|Stjj |||S)NTr]r^r_)err_msgrFrtolatol equal_nanr^r-)r rbrOtestingassert_array_equalrR assert_closer)r[r\r]r^r_rnr,r`s r&xp_assert_equalrw&s z V $FGR(3NGr{zz,,VWg,NN ""R-$Wzz&&vwQQRV38g'G G :: ( (' ( JJrFc d} | t|}t||||||}|j|jd} |.| r,|j |jj dzdz}n|d}t |r |jj|||||St|r+|dk(rdn|}|jj||||dd | Stjj|||||S) NTrm)z real floatingcomplex floatingg?gHz>)rqrrrnroFrp) r rbisdtyperfinfoepsrOrtassert_allcloserRrvr) r[r\rqrrr]r^r_rnr,r`floatings r&xp_assert_closer9s z V $FGR(3NGzz&,,(MNH |xx %))3.2 r{zz))&'/3W*F F "!R-$Wzz&&vwT155g'W W :: % %fgD+/ & BBrFcd}| t|}t||||||}t|r|jj ||||St |rR|j jdk7r|j}|j jdk7r|j}tjj ||||S)NTrm)rnverboser) r rbrOrtassert_array_lessrRr rjrr) r[r\r]r^r_rnrr,r`s r&xp_assert_lessrVs z V $FGR(3NGr{zz++FG4;W,N N " ==   &ZZ\F >>  % 'kkmG :: ' '07 ( JJrFcN| t|}t||}|j||j}t |d|}|j ||}|j |d}|jddz }|dkrtjdtd d }||dddfz}|j}|j|jd r|j|}||z}||z}td t!|jD}|j#||S) Nr=rArr?rz!Degrees of freedom <= 0 for slice) stacklevelgryc32K|]\}}|dk(s |yw)rNri).0r@lengths r& zcov..sL*<,$! *D qy 9$ 4QWA ##C zz#))/0ggcl CAIA L)AGG*<L LD ::ad: ##rFcd|dS)Nz Providing z$ is only supported for numpy arrays.ri)params r&xp_unsupported_param_msgrs y D EErFc:|j|jdS)Nry)r{rrHs r& is_complexrs ::agg1 22rFcg}t|rj|dgz }ddl}|jj}t d|D] }|d|gz } |j j jr|dgz }|St|rDddl }|jjj}t d|D] }|d|gz } |St|rddl }|jd}t d|D] }|d|gz } |jd}t d|D] }|d |gz } |jd } t d| D] }|d |gz } |SdgS) z>   * * , w G 99$$335q(#A %s } $G$ ""5"1q'"A $qc | #G#""5"1q'"A $qc | #G#""5"1q'"A $qc | #G# 6MrFczt|rddl}|jSt|rddl}|jSddl}|S)z Return the `scipy` namespace for alternative backends, where it exists, such as `cupyx.scipy` and `jax.scipy`. Useful for ad hoc dispatching. Default: return `scipy` (this package). rN)rOcupyxscipyrTr)r,rrrs r&scipy_namespace_forrs4r{{{ bzyy LrFct||}t|dr|j||S|j||\}}||k|j |z}|j |||}|j dk(r|dS|S)Nminimumrri)r hasattrrbroadcast_arraysisnanwhererB)x1x2r,r#ress r& xp_minimumrs R Br9zz"b!!  R (FB bBHHRL A ((1b" Chh!m3r7,,rFc| t|n|}|j||j|j||j}}t|dr|j |||S|j |||\}}}|j|d}||k}||||<||kD}||||<|j dk(r|dS|S)NrclipT)r8rri)r r;rrrrrB) rEminmaxr,abyiaibs r&xp_cliprs "z rB ::c: )2::c:+IqAr6wwq!Q!!!Q*GAq! 14  A QB bEAbE QB bEAbEFFaK1R5&Q&rFc| t|n|}tt|j}|j |}||gz}|j ||S)N)r listrrBpop permute_dims)rEsourcer,rtemps r&xp_moveaxis_to_endrsP !#  B aff D 88F D 4&=D ??1d ##rFcr| t||n|}|j|}|j|dk\|| S)Nr)r absr)rrr,abs_x1s r& xp_copysignrs:$&JR BB VVBZF 88B!GVfW --rFcZ| t|n|}t|r|j|S|j||j}|j d|j }|j|dkD||}|j|dk| |}|j|dk(d|z|}|S)Nrrr)r rKsign full_likenanr;rr)rEr,rones r&xp_signrs!z rB|wwqz <<266 "D **Qagg* &C 88AE3 %D 88AEC4 &D 88AFAcE4 (D KrF)r"zlist[ArrayLike]returnz list[Array])r$Arrayr,rrNone)r"rrr)NNN)r$ ArrayLikerrr7z"Literal['K', 'A', 'C', 'F'] | Noner8z bool | Noner,ModuleType | Noner3boolr4rrr)rErrBintr,rrr)rErr,rrr)r,rrr)TTT)TTTroN)NrTTTroN)TTTroTN)rrrstr)rErr,rrr)r,rrzlist[str] | list[None])r,rrr)rrrrrr)NN) rint | float | Array | NonerrrErr,rrr)rErrrr,rrr)rrrrr,rrr)9__doc__ __future__rosrtypesrtypingrrrr r numpy.typingnpt scipy._librscipy._lib.array_api_compatr r r1r __all__environgetr__annotations__rr0rrr'r.r rrDr8rKrOrRrTrbrWrYrwrrrrrrrrrrrrrirFr&rs2# ..' <!jjnn->FFzz~~ne4 '   E %I#L"6N48 0 !%"00020 0  000 0f@D /3)6IGIF GK* 0M2HL59K&IMGKB:GKBFJ(.2$8F3!H.- +/*. ' !% '('( ' '  ' */ '4!% $  $ $  $*/ $EI.59 rF