o  h#V@sUdZddlZddlmZddlmZmZmZddlZ ddl mZ ddl m Z ddlmZmZmZmZmZmZmZmZmZgdZej !dd Z"e#e$Be%d<ej !d d Z&e"e&d Z'eZ(ee%d <e(e j)BZ)ee%d<ddZ*de(deddfddZ+de(defddZ,   dddd d dde)dededdBde$dBdedBde$de$de(fd d!Z-dd"d#e(dedBde(fd$d%Z.d&d&d&d&d'd(d)Z/d*d+Z0d&d&d&d&d,dd-d.d/Z1ddd&d&d&d&d,dd0d1d2Z2d&d&d&d&d,d&dd3d4d5Z3ded7d8Z4dfd:d;Z5dZ6d#e(dede$fd?d@Z7dede8e#e8dBfdAdBZ9dededBfdCdDZ:dd"d#e(dEe;dedBde(fdFdGZdd dNddOd#e(dPe;e?e;BdBdQe$dRe;e@BdedBde(f dSdTZAdd"d#e(dedBde(fdUdVZBdd"d#e(dedBde(fdWdXZCdYddZd[e(d\e(dPe;dedBde(f d]d^ZDd d dd_d`daZEdgd[e(dedBde(fdbdcZFdS)ha6Utility 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 ArrayLikec Cstt|D]}||}ddlm}||rd}t|t|tjjr't dt|tj r1t dt|tj tj BrT|j }t|tjsSt|tjsSt d|dqt|szt|}Wn t yjt dw|j }t|tjst|tjsd |d}t ||||<q|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 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.sparser/ ValueError isinstancenpma MaskedArray TypeErrormatrixndarraygenericdtype issubdtypenumberbool_r asanyarray)arraysiarrayr/msgr<messagerFI/var/www/vscode/kcb/lib/python3.10/site-packages/scipy/_lib/_array_api.py_compliance_scipy9s:     rHrCxpreturncCs:d}z|||st|WdStyt|w)zCheck for NaNs or Infs.z#array must not contain infs or NaNsN)allisfiniter3r8)rCrIrDrFrFrG _check_finitehs rMrAcGs,tdstSdd|D}t|}tj|S)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. rcSg|]}|dur|qSNrF).0rCrFrFrG z#array_namespace..)_GLOBAL_CONFIG np_compatrHrr)rA_arraysrFrFrGrrs  r)rI check_finitesubokr<order)KACFcopyrVrWcCs|durt|}t|r.|durtj||||d}n7|r%tj|||d}n,tj|||d}n#z |j|||d}WntyPt|d}|j|||d}Ynw|rXt|||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. NT)rXr<rW)rXr<)r<r])rrr5rCr@asarrayr8rM)rCr<rXr]rIrVrW coerced_xprFrFrGrs"  rrIxcCs|durt|}t|d|dS)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)r]rI)rrrbrIrFrFrGr!sr!Tcheck_namespace check_dtype check_shapecheck_0dc Csd}|r t||t|r3|r3dt|dt|}||r%||s3||s/||r3J|||}||}|rSd|jd|j}|j|jksSJ||rid|jd|j}|j|jksiJ||||j}||fS)NTz$Array-ness does not match: Actual: z Desired: zdtypes do not match. Actual: Desired: zShapes do not match. Actual: )_assert_matching_namespacertypeisscalarr_r<shape broadcast_to) actualdesiredrIrerfrgrh__tracebackhide___msgrFrFrG _strict_checks4    rscCsZd}t|tr |n|f}t|}|D]}t|}d|jd|j}||ks*J|qdS)NTz!Namespaces do not match. Actual: ri)r4tupler__name__)rorprq desired_spacearr arr_spacerrrFrFrGrjsrj)rerfrgrherr_msgrIc Csd}|dur t|}t|||||||d\}}t|r$|jj|||dSt|r=|dkr.dn|}|jj||dddd|dStjj|||dS)NTrd)rzryrFrtolatol equal_nanrfrD)rrsrtestingassert_array_equalr assert_closer5) rorprerfrgrhrzrIrqrFrFrGr s  r)r|r}rerfrgrhrzrIc Csd} | dur t|} t||| ||||d\}}| |jd} |dur0| r0| |jjdd}n|dur6d}t| rE| jj|||||dSt | r^|dkrOdn|}| jj ||||dd |d St jj|||||dS) NTrd) real floatingcomplex floating?gHz>)r|r}rzryFr{) rrsisdtyper<finfoepsrrassert_allcloserrr5) rorpr|r}rerfrgrhrzrIrqfloatingrFrFrGr$s2    r)rerfrgrhrzverboserIc Csd} |dur t|}t|||||||d\}}t|r%|jj||||dSt|r=|jjdkr3|}|jjdkr=|}t jj||||dS)NTrd)rzrr,) rrsrrassert_array_lessrr rkr,r5) rorprerfrgrhrzrrIrqrFrFrGr Fs(     r cO6ddd| }}t||g|R||ddd|SzPBackwards compatible replacement. In new code, use xp_assert_close instead. rg? F)r}r|rfrgrrorpdecimalargskwdsr|r}rFrFrGr_rcOrrrrrFrFrGrhrrparamcCs d|dS)Nz Providing z$ is only supported for numpy arrays.rF)rrFrFrGr*qs r*cCs||jdSNr)rr<rcrFrFrGrusrc Cs@g}t|r1|dg7}ddl}|j}td|D] }|d|g7}q|jjr/|dg7}|St|rQddl }|jj }td|D] }|d|g7}qD|St |rddl }|jdd}td|D] }|d|g7}qd|jdd}td|D] }|d |g7}qz|jd d} td| D] }|d |g7}q|SdgS) z.cSrNrOrFrrFrFrGrQrRg?cSsg|]}|jqSrFrrrFrFrGrQintegralcsg|] }|ds|qS)r)r)rPr<rarFrGrQ!s  cSsh|]}|jqSrF)rmrrFrFrG +rz'xp_broadcast_promote..rrz/Array shapes are incompatible for broadcasting.rWT)r])rr_r< result_typerr8r1r5broadcast_shapesrmr3appendrrnastype)rrrIr args_not_none default_floatdtypesr< float_dtypesshapesrmerEoutrkwargsrFrarGxp_broadcast_promotesR          rcCsX|durt|n|}|j}|||jr|||j}|S||dr*|||j}|S)Nr)rr<rfloat32r complex64 complex128)rwrI arr_dtyperFrFrGxp_float_to_complexHs r)NNN)r)rrO)G__doc__ostypesrtypingrrrr r5 numpy.typingnpt scipy._librscipy._lib.array_api_compatrrr(rTr r#r rr rr rrrrr__all__environgetrstrbool__annotations__rrSr-r.rHrMrrr!rsrjrrr rrr*rrrrintr$r"r'rtfloatr+r%r&r)rrrFrFrFrGs   ,   / '  /    "    $ $         7