o iQW@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,   dfdd 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&dd3d4d5Z3dgd7d8Z4dhd: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`daZEdid[e(dedBde(fdbdcZFdddeZGdS)ja6Utility 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 `)rangelenZ scipy.sparser/ ValueError isinstancenpmaZ MaskedArray TypeErrormatrixZndarrayZgenericdtypeZ issubdtypenumberZbool_r asanyarray)arraysiarrayr/msgr8messager@O/home/kim/smarthome/.venv/lib/python3.10/site-packages/scipy/_lib/_array_api.py_compliance_scipy9s:     rBr=xpreturncCs:d}z|||st|WdStyt|w)zCheck for NaNs or Infs.z#array must not contain infs or NaNsN)allisfiniter2r6)r=rCr>r@r@rA _check_finitehs rGr;cGs,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|qSNr@).0r=r@r@rA z#array_namespace..)_GLOBAL_CONFIG np_compatrBrr)r;Z_arraysr@r@rArrs  r)rC check_finitesubokr8order)KACFcopyrOrPcCs|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)rQr8rP)rQr8)r8rV)rrr4r=r:asarrayr6rG)r=r8rQrVrCrOrPZ coerced_xpr@r@rArs"  rrCxcCs|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)rVrC)rrrZrCr@r@rAr!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_namespacertypeZisscalarrXr8shape broadcast_to) actualdesiredrCr]r^r_r`__tracebackhide___msgr@r@rA _strict_checks4    rjcCsZd}t|tr |n|f}t|}|D]}t|}d|jd|j}||ks*J|qdS)NTz!Namespaces do not match. Actual: ra)r3tupler__name__)rfrgrhZ desired_spacearrZ arr_spacerir@r@rArbsrb)r]r^r_r`err_msgrCc Csd}|dur t|}t|||||||d\}}t|r$|jj|||dSt|r=|dkr.dn|}|jj||dddd|dStjj|||dS)NTr\)rornrFrtolatolZ equal_nanr^r>)rrjrtestingZassert_array_equalr assert_closer4) rfrgr]r^r_r`rorCrhr@r@rAr s  r)rqrrr]r^r_r`rorCc Csd} | dur t|} t||| ||||d\}}| |jd} |dur0| r0| |jjdd}n|dur6d}t| rE| jj|||||dSt | r^|dkrOdn|}| jj ||||dd |d St jj|||||dS) NTr\) real floatingcomplex floating?gHz>)rqrrrornFrp) rrjisdtyper8ZfinfoepsrrsZassert_allcloserrtr4) rfrgrqrrr]r^r_r`rorCrhZfloatingr@r@rAr$s2    r)r]r^r_r`roverboserCc Csd} |dur t|}t|||||||d\}}t|r%|jj||||dSt|r=|jjdkr3|}|jjdkr=|}t jj||||dS)NTr\)ror{r,) rrjrrsZassert_array_lessrr rcr,r4) rfrgr]r^r_r`ror{rCrhr@r@rAr Fs(     r cO6ddd| }}t||g|R||ddd|SzPBackwards compatible replacement. In new code, use xp_assert_close instead. rg? F)rrrqr^r_rrfrgdecimalargskwdsrqrrr@r@rAr_rcOr}r~rrr@r@rArhrrparamcCs d|dS)Nz Providing z$ is only supported for numpy arrays.r@)rr@r@rAr*qs r*cCs||jdSNrv)ryr8r[r@r@rArusrc 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.cSrHrIr@rr@r@rArKrLg?cSsg|]}|jqSr@rrr@r@rArKintegralcsg|] }|ds|qS)r)ry)rJr8rYr@rArK!s  cSsh|]}|jqSr@)rdrr@r@rA +rz'xp_broadcast_promote..rrz/Array shapes are incompatible for broadcasting.rPT)rV)rrXr8Z result_typeryr6r1r4Zbroadcast_shapesrdr2appendrreastype)rrrCrZ args_not_noneZ default_floatZdtypesr8Z float_dtypesZshapesrder?outrkwargsr@rYrAxp_broadcast_promotesR          rcCsX|durt|n|}|j}|||jr|||j}|S||dr*|||j}|S)Nru)rr8ryZfloat32rZ complex64Z complex128)rmrCZ arr_dtyper@r@rAxp_float_to_complexHs rcCst|r|S|jS)z@Query the namespace-dependent default floating-point dtype. )rZget_default_dtypeZfloat64rYr@r@rAxp_default_dtypeVsr)NNN)r|)rrI)H__doc__ostypesrtypingrrrr r4Z numpy.typingZnptZ scipy._librZscipy._lib.array_api_compatrrr(rNr r#r rr rr rrrrr__all__environgetrstrbool__annotations__rrMr-r.rBrGrrr!rjrbrrr rrr*rrrrintr$r"r'rkfloatr+r%r&r)rrrr@r@r@rAs   ,   / '  /    "    $ $        7