Q(hQWUdZddlZddlmZddlmZmZmZddlZ ddl mZ ddl m Z ddlmZmZmZmZmZmZmZmZmZgdZej@jCdd Z"e#e$ze%d<ej@jCd d Z&e"e&d Z'eZ(ee%d <e(e jRzZ)ee%d<dZ*de(deddfdZ+de(defdZ, dJdd d dde)dededdzde$dzdedzde$de$de(fdZ-ddde(dedzde(fd Z.d!d!d!d!d"d#Z/d$Z0d!d!d!d!d%dd&d'Z1ddd!d!d!d!d%dd(d)Z2d!d!d!d!d%d!dd*d+Z3dKd,Z4dLd-Z5d.ede#fd/Z6de(dede$fd0Z7dede8e#e8dzfd1Z9dededzfd2Z:ddde(d3e;dedzde(fd4Zdd d9dd:de(d;e;e?e;zdzdZAddde(dedzde(fd?ZBddde(dedzde(fd@ZCdAddBdCe(dDe(d;e;dedzde(f dEZDd d ddFdGZEdMdCe(dedzde(fdHZFdIZGy)Na6Utility 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 N) ModuleType)AnyLiteral TypeAlias)array_api_compat) is_array_api_objsizenumpydeviceis_numpy_namespaceis_cupy_namespaceis_torch_namespaceis_jax_namespaceis_array_api_strict_namespace)_asarrayarray_namespaceassert_almost_equalassert_array_almost_equalget_xp_devicesis_array_api_strict is_complexis_cupyis_jaxis_numpyis_torchSCIPY_ARRAY_API SCIPY_DEVICEscipy_namespace_forxp_assert_closexp_assert_equalxp_assert_lessxp_copy xp_copysign xp_devicexp_moveaxis_to_endxp_ravelxp_realxp_signxp_sizexp_take_along_axisxp_unsupported_param_msgxp_vector_normrFrcpu)rrArray ArrayLikectt|D]}||}ddlm}||r d}t |t |t jjr tdt |t jr tdt |t jt jzre|j}t j|t jrt j|t j rtd|dt#|r t j$|}|j}t j|t js5t j|t j sd |d}t||||<|S#t$r tdwxYw) 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 r)issparsezSparse arrays/matrices are not supported by this function. Perhaps one of the `scipy.sparse.linalg` functions would work instead.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 scipy.sparser1 ValueError isinstancenpma MaskedArray TypeErrormatrixndarraygenericdtype issubdtypenumberbool_r asanyarray)arraysiarrayr1msgr>messages S/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/scipy/_lib/_array_api.py_compliance_scipyrI9s3v; !q ) E?)CS/ ! eRUU.. /VW W ryy )NO O eRZZ"**4 5KKEMM%3r}}UBHH7U"9%CS!TUU!%( 7 e,KKEMM%3r}}UBHH7UGyQGH ((F1IC!D M 7!677 7s +F//GrExpreturncd} |j|j|s t|y#t$r t|wxYw)zCheck for NaNs or Infs.z#array must not contain infs or NaNsN)allisfiniter5r:)rErJrFs rH _check_finiterOhsI /Cvvbkk%()S/ !* os +0ArCctdstS|Dcgc]}|| }}t|}tj|Scc}w)a1Get 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_compatrIrr)rCrE_arrayss rHrrrsK4 + ,"(>E,=u>G>(G  + +W 55 ?s??)rJ check_finitesubokr>order)KACFcopyrTrUc| t|}t|rR|durtj||||}nI|rtj|||}n.tj |||}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. T)rVr>rU)rVr>)r>r[)rrr7rErBasarrayr:rO)rEr>rVr[rJrTrU coerced_xps rHrrs0 z U #| 4<HHU%uEJE MM%uEBEJJuE?E FJJuEJ=E eR L F(A7J&&uE&EE Fs,B7C  C rJxc8| 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)r[rJ)rrrarJs rHr"r"s", z Q  ADR ((Tcheck_namespace check_dtype check_shapecheck_0dcd}|r t||t|rg|redt|dt|}|j|r|j|s)|j|s|j|rJ||j |}|j |}|rshape broadcast_to) actualdesiredrJrfrgrhri__tracebackhide___msgs rH _strict_checkrus."673|7<. T']O=V$W)= F+BKK4H QLP QJZZ Fjj!G/ ~[X||w}},2d2,/ ~[X||w}},2d2,oogv||4G 7?rdcd}t|tr|n|f}t|}|D]5}t|}d|jd|j}||k(r0J|y)NTz!Namespaces do not match. Actual: rk)r6tupler__name__)rqrrrs desired_spacearr arr_spacerts rHrlrls{!&%0VviF#G,M0#C( $--./)2235M)/4/) 0rd)rfrgrhrierr_msgrJc Ld}| t|}t|||||||\}}t|r|jj |||St |r+|dk(rdn|}|jj ||dddd|Stjj |||S)NTre)r}r|rFrtolatol equal_nanrgrF)rrurtestingassert_array_equalr assert_closer7) rqrrrfrgrhrir}rJrss rHr r s z V $#_[OFG r{zz,,VWg,NN ""R-$Wzz&&vwQQRV38g'G G :: ( (' ( JJrd)rrrfrgrhrir}rJc d} | t|} t||| ||||\}}| j|jd} |.| r,| j |jj dzdz}n|d}t | r | jj|||||St| r+|dk(rdn|}| jj||||dd | Stjj|||||S) NTre) real floatingcomplex floating?gHz>)rrr}r|Fr) rruisdtyper>finfoepsrrassert_allcloserrr7) rqrrrrrfrgrhrir}rJrsfloatings rHrr$s  z V $#'[(OFG zz&,,(MNH |xx %))3.2 r{zz))&'/3W*F F "!R-$Wzz&&vwT155g'W W :: % %fgD+/ & BBrd)rfrgrhrir}verboserJc d} | t|}t|||||||\}}t|r|jj ||||St |rR|j jdk7r|j}|j jdk7r|j}tjj ||||S)NTre)r}rr-) rrurrassert_array_lessrr rmr-r7) rqrrrfrgrhrir}rrJrss rHr!r!Fs z V $#_[OFG r{zz++FG4;W,N N " ==   &ZZ\F >>  % 'kkmG :: ' '07 ( JJrdcBddd| zz}}t||g|||ddd|SzPBackwards compatible replacement. In new code, use xp_assert_close instead. rg? F)rrrgrhrrqrrdecimalargskwdsrrs rHrr_HCgX&$D 67 *  * $4UPU *$( **rdcBddd| zz}}t||g|||ddd|Srrrs rHrrhrrdparamcd|dS)Nz Providing z$ is only supported for numpy arrays.)rs rHr+r+qs y D EErdc:|j|jdSNr)rr>rcs rHrrus ::agg1 22rdcg}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 $ 6Mrdct|rddl}|jSt|rddl}|jSt |r|Sy)aReturn the `scipy`-like namespace of a non-NumPy backend That is, return the namespace corresponding with backend `xp` that contains `scipy` sub-namespaces like `linalg` and `special`. If no such namespace exists, return ``None``. Useful for dispatching. rN)rcupyxscipyrrr)rJrrs rHrrs;r{{{ bzyy| rdsourcec| t|n|}tt|j}|j |}||gz}|j ||SN)rlistr2ndimpop permute_dims)rarrJaxestemps rHr%r%sP !#  B aff D 88F D 4&=D ??1d ##rdx1x2cr| t||n|}|j|}|j|dk\|| S)Nr)rabswhere)rrrJabs_x1s rHr#r#s:$&JR BB VVBZF 88B!GVfW --rdcp| t|n|}t|r|j|S|j|}|j d|j }|j |dkD||}|j |dk| |}|j |j||j|z|}|S)N)r>r) rrsign zeros_liker^r>risnannan)rarJrones rHr(r(s!z rB|wwqz == D **Qagg* &C 88AE3 %D 88AEC4 &D 88BHHQKT 2D Krd)axiskeepdimsordrJrrrc:| t|n|}trdt|dr|jj ||||S|dk7r t d|j |j||z||dzStjj||||S)Nlinalg)rrrrzonly the Euclidean norm (`ord=2`) is currently supported in `xp_vector_norm` for backends not implementing the `linalg` extension.)rrr)rrr) rrhasattrr vector_normr5sumconjr7norm)rarrrrJs rHr,r,s "z rB 2x 99((c(R Rax !66"''!*q.th6GL Lyy~~aSth~GGrdcD| t|n|}|j|dS)N))rreshapercs rHr&r&s% "z rB ::a rdc~| t|n|}|j|jdr|j|S|Sr)rrr>realrcs rHr'r's9 "z rBAGG-?@2771:GaGrdr)rrJrzindicesc| t|n|}t|r|j|||St|r t d|j |||S)N)dimz2Array API standard does not define take_along_axis)rrtake_along_dimrNotImplementedErrortake_along_axis)rzrrrJs rHr*r*s\ "$ B|  g4 88 R !"VWW!!#w55rd)ensure_writeableforce_floatingrJc|t|n|}|Dcgc]}| t|dn|}}|Dcgc]}|| }}|jdj}|Dcgc]}|j}} |j|}|r$|j |dr|j ||}|Dchc]}|j} } t| dk7rtj| n|dj} g}|D]}||j||j| k7r%t|rddini}|j|| fi|}|j|k7s|r|j||d }|j||Scc}wcc}wcc}w#t $rV|Dcgc]}|j |ds|ncc}w} }| r|jg| |}n|r|}n|j|}Y?wxYwcc}w#t$r} d} t| | d} ~ wwxYw) NT)rUg?integralrrz/Array shapes are incompatible for broadcasting.rU)r[)rrr^r> result_typerr:ror3r7broadcast_shapesr5appendrrpastype)rrrJrarg args_not_none default_floatdtypesr> float_dtypesshapesroerGoutkwargss rHxp_broadcast_promoters:#%:$ 2BOS T#/Xc &s B TD T$(%!zz%<>> > "BNN@L@-@E !E"BNNF+E,2 )C!q()sQE9E>E>F25F+G*/G/ G'F103G'&G'/ H 8HH c | t|n|}|j}|j||jr|j ||j }|S|j|dr|j ||j }|S)Nr)rr>rfloat32r complex64 complex128)rzrJ arr_dtypes rHxp_float_to_complexrHss!# B I zz)RZZ(iiR\\* J I /iiR]]+ JrdcPt|r|jS|jS)z@Query the namespace-dependent default floating-point dtype. )rget_default_dtypefloat64r`s rHxp_default_dtyperVs&|##%%zzrd)NNN))r)H__doc__ostypesrtypingrrrr r7 numpy.typingnpt scipy._librscipy._lib.array_api_compatrr r)rRr r$r rr rrrrrrr__all__environgetrstrbool__annotations__rrQr.r/rIrOrrr"rurlr rr!rrr+rrrrintr%r#r(rwfloatr,r&r'r*rrrrrdrHrs **'    !jjnn->FtFzz~~ne4 '  ys}}, 9,,^J4"6U"6z"6N48 , !%",,,)*T1,Tk ,   ,,, ,^26)u)Z$.)%):#'Dd< 09=$ $tRDK0.24 $$4BD8<#dBQUJ2**FCFCF3%3Z3D3!z!d3i$t*&<!HJ:+<8!% $  $ $   $*/ $EI.E.u. T0A.U.59 u  T 1 U 48$)&'+/ HeHuSz)D0H!He H"D( H5: H46:  d!2 e 59HuH T 1HUH$&/3 6E 6 % 6  6&, 69> 6 27uQU4n U  T(9 U rd