UgH>UdZddlmZddlZddlZddlmZddlmZm Z m Z ddl Z ddl mZddlmZddlmZmZm ZmZgdZejd d Zd ed <ejd d ZeedZe r eZeejzZdMdZdNdZ dOdZ! dPdd d ddQd'Z"dd(dRd,Z#dd(dSd-Z$dTd.Z%dTd/Z&dTd0Z'd1Z( dUd3Z)d4Z*d5Z+ dVd7Z, dWd8Z- dXd9Z.dd(dSd:Z/dYd=Z0dZd>Z1d[d@Z2d\dAZ3d]dDZ4 d^dd(d_dHZ5dd(d`dJZ6dd(dadKZ7dd(dSdLZ8dS)ba6Utility 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)rrarrayslist[ArrayLike]return list[Array]cFtt|D]}||}t|tjjrt dt|tjrt dt|tjtj frY|j }tj |tj s2tj |tj st d|dt|s tj|}n#t $rt dwxYw|j }tj |tj s4tj |tj sd|d}t ||||<|S)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)riarrayr!messages T/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/scipy/_lib/_array_api.pycompliance_scipyr*/s3v;;  q  eRU. / / PVWW W ry ) ) PNOO O ebj"*5 6 6 KEM%33 Ur}UBH7U7U U!T%!T!T!TUUU!%((  7 e,, 7 7 7!6777 7KEM%33 )r}UBH7U7U )HHHH (((F1I Ms D##D=r'ArrayxprNonecd} |||st|dS#t$rt|wxYw)zCheck for NaNs or Infs.z#array must not contain infs or NaNsN)allisfinite ValueErrorr)r'r,msgs r) _check_finiter3Usi /Cvvbkk%(()) "S// ! " " oos 7=Ac|tdstSd|D}t|}tj|S)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. rcg|]}||SN).0r's r) z#array_namespace..}s>>>E,=u,=,=,=)_GLOBAL_CONFIG np_compatr*rr )r_arrayss r)r r _sF4 + ,>>&>>>Gw''G  +W 55r:)r, check_finitesubok ArrayLiker!rorder"Literal['K', 'A', 'C', 'F'] | Nonecopy bool | NoneModuleType | Noner>boolr?c|t|}|jdvrd|durtj||||}n1|rtj|||}ntj|||}||}nd ||||}nJ#t $r=t|d}||||}YnwxYw|rt|||S)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. N>r !scipy._lib.array_api_compat.numpyT)rAr!r?)rAr!)r!rC)r __name__rr'r%asarrayrr3)r'r!rArCr,r>r? coerced_xps r)rrs%0 z U # # {DDD 4<<HU%uEJJJEE  @M%uEBBBEEJuE???E 5!! FJJuEJ==EE F F F(A77J&&uE&EEEEE F!eR   LsBAC C r,xndimintc|t|}||}|j|kr)||d}t |||}|S)z9Recursively expand the dimension to have at least `ndim`.NraxisrOr,)r rKrO expand_dims atleast_nd)rNrOr,s r)rVrVs\ z Q   1 Av}} NN11N % % qt + + + Hr:cH|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. NT)rCr,)r rrNr,s r)rCrCs+* z Q   ADR ( ( ((r:c|jdvS)N)r rHrJrMs r)is_numpyr[ ;H HHr:c|jdvS)N)cupyz scipy._lib.array_api_compat.cupyrZrMs r)is_cupyr_s ;F FFr:c|jdvS)N)torchz!scipy._lib.array_api_compat.torchrZrMs r)is_torchrbr\r:c|jdvS)N)z jax.numpyzjax.experimental.array_apirZrMs r)is_jaxrds ;E EEr:Tchd}|rt||||}|r,d|jd|j}|j|jks J||r=d|jd|j}|j|jks J|t ||||||j}|S)NTzdtypes do not match. Actual: Desired: zShapes do not match. Actual: )_assert_matching_namespacerKr!shape _check_scalar broadcast_to)actualdesiredr,check_namespace check_dtype check_shape__tracebackhide___msgs r) _strict_checkrrs4"67333jj!!G3X XXXX|w},,,d,,,+X XXXX|w},,,d,,,fgr***oogv|44G Nr:cd}t|tr|n|f}t|}|D]3}t|}d|jd|j}||ks J|4dS)NTz!Namespaces do not match. Actual: rf)rtupler rJ)rkrlrp desired_spacearr arr_spacerqs r)rgrgs!&%00 ?VVviF#G,,M00#C(( 5$-55)255M)))4)))) 00r:cVd}|jdkst|sdS|d}dt|dt|}||r||s4||s||r J|dSdS)NTr7zTypes do not match: Actual: z Desired: )rhr[typeisscalar)rkrlr,rprqs r)riri s}(2,,bkG UDLL U Ud7mm U UD KK  MBKK$8$8MKK''M02 G0D0DMMHLMMMMMMMr:c bd}|t|}t||||||}t|r|j|||St |r+|dkrdn|}|j||dddd|Stj|||S)NTrmrnro)err_msgr{rFrtolatol equal_nanrnr2)r rrr_testingassert_array_equalrb assert_closer)rkrlrmrnror~r,rps r)xp_assert_equalr&s z V $ $FGR(3NNNGr{{Gz,,VWg,NNN "G"R--$$Wz&&vwQQRV38g'GG G : ( (' ( J JJr:c d} |t|}t||||||}||jd} |(| r&||jjdzdz}n|d}t |r|j|||||St|r+|dkrdn|}|j ||||dd | Stj|||||S) NTr})z real floatingcomplex floatingg?gHz>)rrr~r{Fr) r rrisdtyper!finfoepsr_rassert_allcloserbrr) rkrlrrrmrnror~r,rpfloatings r)xp_assert_closer9sI z V $ $FGR(3NNNGzz&,(MNNH ||xx %%)3.2 r{{Wz))&'/3W*FF F "W!R--$$Wz&&vwT155g'WW W : % %fgD+/ & B BBr:cd}|t|}t||||||}t|r|j||||St |rH|jjdkr|}|jjdkr|}tj||||S)NTr})r~verboser) r rrr_rassert_array_lessrbr ryrr) rkrlrmrnror~rr,rps r)xp_assert_lessrVs z V $ $FGR(3NNNGr{{$z++FG4;W,NN N "$ =  & &ZZ\\F > % ' 'kkmmG : ' '07 ( J JJr:c|t|}t||}|||j}t |d|}|||}||d}|jddz }|dkrtj dtd d }||dddfz}|j }| |j d r||}||z}||z}td t!|jD}|||S) NrMrTr!rRrz!Degrees of freedom <= 0 for slice) stacklevelgrc3,K|]\}}|dk |VdS)rNr7)r8rSlengths r) zcov..s*LL,$!  LLr:)r rC result_typefloat64rVrKmeanrhwarningswarnRuntimeWarningTrr!conjrt enumeratesqueeze) rNr,Xr!avgfactX_Tcaxess r)covrjsG z Q   Q2A NN1bj ) )E11$$$A 1E ""A ''!!'  C 71:>D qyy 9$ 4 4 4 4QQQWA #C zz#)/00ggcll CAIA LL)AG*<*<LLL L LD ::ad: # ##r:paramstrcd|dS)Nz Providing z$ is only supported for numpy arrays.r7)rs r)xp_unsupported_param_msgrs E E E EEr:c8||jdS)Nr)rr!rXs r) is_complexrs ::ag1 2 22r:list[str] | list[None]cg}t|re|dgz }ddl}|j}t d|D] }|d|gz } |jjr|dgz }|St|r@ddl }|jj }t d|D] }|d|gz } |St|rddl }|d}t d|D] }|d|gz } |d}t d|D] }|d |gz } |d } t d| D] }|d |gz } |SdgS) z  * * , ,  w G  9$3355q(## % %A   } $GG   ""5"11q'"" $ $A  q | #GG""5"11q'"" $ $A  q | #GG""5"11q'"" $ $A  q | #GG 6Mr:cvt|r ddl}|jSt|r ddl}|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)r_cupyxscipyrdr)r,rrrs r)scipy_namespace_forrsPr{{ { bzz yLLL Lr:x1x2c0t||}t|dr|||S|||\}}||k||z}||||}|jdkr|dn|S)Nminimumrr7)r hasattrrbroadcast_arraysisnanwhererO)rrr,r&ress r) xp_minimumrs R Br9"zz"b!!!  R ( (FB bBHHRLL A ((1b"  Ch!mm3r77,r:minint | float | Array | Nonemaxc|t|n|}|||j|||j}}t|dr||||S||||\}}}||d}||k}||||<||k}||||<|jdkr|dn|S)NrclipT)rCrr7)r rKr!rrrrO) rNrrr,abyiaibs r)xp_cliprs "z   rB ::c: ) )2::c:+I+IqAr6 wwq!Q!!!Q**GAq! 14   A QB bEAbE QB bEAbEFaKK1R55Q&r:sourcec|t|n|}tt|j}||}||gz}|||Sr6)r listrrOpop permute_dims)rNrr,rtemps r)xp_moveaxis_to_endrs_ !#    B af  D 88F  D 4&=D ??1d # ##r:c|t||n|}||}||dk|| S)Nr)r absr)rrr,abs_x1s r) xp_copysignrsE$&JR BB VVBZZF 88B!GVfW - --r:c|t|n|}t|r||S|||j}|d|j}||dk||}||dk| |}||dkd|z|}|S)Nrrr)r r[sign full_likenanrKr!r)rNr,rones r)xp_signrs!z   rB||wwqzz <<26 " "D **Qag* & &C 88AE3 % %D 88AEC4 & &D 88AFAcE4 ( (D Kr:)rrrr)r'r+r,rrr-)rr+rr)NNN)r'r@r!rrArBrCrDr,rEr>rFr?rFrr+)rNr+rOrPr,rErr+)rNr+r,rErr+)r,rrrF)TTT)TTTr{N)NrTTTr{N)TTTr{TN)rrrr)rNr+r,rrrF)r,rrr)r,rrr)rr+rr+rr+)NN) rrrrrNr+r,rErr+)rNr+rrPr,rErr+)rr+rr+r,rErr+)9__doc__ __future__rosrtypesrtypingrrrr r numpy.typingnpt scipy._librscipy._lib.array_api_compatr r r<r __all__environgetr__annotations__rr;r+r@r*r3r rrVrCr[r_rbrdrrrgrirrrrrrrrrrrrrr7r:r)rs8#""""" ..........'''''' < ; ;!jnn->FFFFFFz~~ne44 '  & E %I####L"6"6"6"6N48 0 !%"000000f@D      /3))))))6IIIIGGGGIIIIFFF GK* 0 0 0MMM2HL59KKKK&IMGKBBBB:GKBFJJJJ(.2$$$$$$8FFFF3333!!!!H.---- +/*. ' !% ''''''4!% $ $ $ $ $ $EI......59        r: