o 5Bi1@sHUdZddlmZddlZddlZddlZddlmZddl m Z m Z mZ gdZ ejddZd ed<ejd d Zeed Zd dZddZddZ d6dddddZddddZddddZddZddZdd Z !d7d"d#Zd$d%Zd&d'Z ! d8d)d*Z ! d9d,d-Z! ! d:d.d/Z"ddd0d1Z#d2d3Z$d4d5Z%dS);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)array_api_compat)is_array_api_objsizenumpy)array_namespace_asarrayrSCIPY_ARRAY_APIFz str | bool SCIPY_DEVICEcpu)r r c Cstt|D]t}||}t|tjjrtdt|tjr!tdt|tjtj frD|j }t |tj sCt |tj sCtd|dqt|szzt|}Wn tyZtdw|j }t |tj svt |tj svd|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 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)arraysiarrayrmessagerU/var/www/html/Trade-python/venv/lib/python3.10/site-packages/scipy/_lib/_array_api.pycompliance_scipy%s2    r!cCs:d}z|||st|WdStyt|w)zCheck for NaNs or Infs.z#array must not contain infs or NaNsN)allisfinite ValueErrorr)rxpmsgrrr _check_finiteKs r'cGs,tdstSdd|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. r cSsg|]}|dur|qS)Nr).0rrrr ssz#array_namespace..)_GLOBAL_CONFIG np_compatr!rr)rrrr rUs  r)r% check_finitecCs|durt|}|jdvr(|durtj|||d}ntj|||d}||}n#z |j|||d}WntyJt|d}|j|||d}Ynw|rRt|||S)aSciPy-specific replacement for `np.asarray` with `order` and `check_finite`. 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. N>r!scipy._lib.array_api_compat.numpyT)orderr)rcopy)r__name__rrasarrayrr')rrr.r/r%r, coerced_xprrr rzs      rr%cCsD|durt|}||}|j|kr |j|dd}t|||d}|S)z9Recursively expand the dimension to have at least `ndim`.Nraxisndimr%)rr2r8 expand_dims atleast_nd)xr8r%rrr r:s  r:cCs|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/r%)rrr;r%rrr r/sr/cC |jdvS)N)rr-r1r4rrr is_numpy r?cCr=)N)cupyz scipy._lib.array_api_compat.cupyr>r4rrr is_cupyr@rBcCr=)N)torchz!scipy._lib.array_api_compat.torchr>r4rrr is_torchr@rDTcCsrd}|r t||||}|rd}|j|jksJ||r0d}|j|jks*J|t||||||j}|S)NTzDdtypes do not match. Actual: {actual.dtype} Desired: {desired.dtype}zDShapes do not match. Actual: {actual.shape} Desired: {desired.shape})_assert_matching_namespacer2rshape _check_scalar broadcast_to)actualdesiredr%check_namespace check_dtype check_shape__tracebackhide___msgrrr _strict_checks   rPcCsZd}t|tr |n|f}t|}|D]}t|}d|jd|j}||ks*J|qdS)NTz!Namespaces do not match. Actual: z Desired: )rtuplerr1)rIrJrN desired_spacearr arr_spacerOrrr rEsrEcCsrd}|jdks t|s dS|d}dt|dt|}||r'||s5||s1||r7J|dSdS)NTrzTypes do not match: Actual: z Desired: )rFr?typeisscalar)rIrJr%rNrOrrr rGs rGc Csd}|dur t|}t||||||d}t|r!|jj|||dSt|r:|dkr+dn|}|jj||dddd|dStjj|||dS)NTrKrLrM)err_msgrWrFrtolatol equal_nanrLr&)rrPrBtestingassert_array_equalrD assert_closer)rIrJrKrLrMrYr%rNrrr xp_assert_equal s raHz>c Csd} |dur t|}t||||||d}t|r#|jj|||||dSt|r<|dkr-dn|}|jj||||dd|dStjj|||||dS)NTrX)r[r\rYrWFrZ)rrPrBr^assert_allcloserDr`r) rIrJr[r\rKrLrMrYr%rNrrr xp_assert_closes$   rdc Csd}|dur t|}t||||||d}t|r"|jj||||dSt|r:|jjdkr0|}|jjdkr:|}t jj||||dS)NTrX)rYverboser ) rrPrBr^assert_array_lessrDdevicerUr r) rIrJrKrLrMrYrer%rNrrr xp_assert_less0s$     rhc Cs|durt|}t||d}|||j}t|d|d}|j||d}|j|dd}|jdd}|dkr?tj dt dd d }||dddf8}|j }| |j d rX||}||}||}td d t|jD}|j||dS)Nr4r7)rr5rz!Degrees of freedom <= 0 for slice) stacklevelgcomplex floatingcss |] \}}|dkr|VqdS)rjNr)r(r6lengthrrr [szcov..)rr/ result_typefloat64r:r2meanrFwarningswarnRuntimeWarningTisdtyperconjrQ enumeratesqueeze) r;r%XravgfactX_Tcaxesrrr covCs*  rcCs d|dS)Nz Providing z$ is only supported for numpy arrays.r)paramrrr xp_unsupported_param_msg_s rcCs||jdS)Nrl)rvrr<rrr is_complexcsr)NNN)TTT)TTTrWN)rbrTTTrWN)TTTrWTN)&__doc__ __future__rosrrrr scipy._librscipy._lib.array_api_compatrrr+__all__environgetr __annotations__r r*r!r'rrr:r/r?rBrDrPrErGrardrhrrrrrrr sR  & & &