a h1@s8UdZddlmZddlZddlZddlZddlmZddl m Z m Z mZ gdZ ejddZd ed<ejd d Zeed Zd dZddZddZd6dddddZddddZddddZddZddZdd Zd7d"d#Zd$d%Zd&d'Zd8d)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|tjjr0tdnt|tjrDtdt|tjtj fr|j }t |tj st |tj std|dq t|s zt|}WntytdYn0|j }t |tj st |tj sd|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 isinstancenpmaZ MaskedArray TypeErrormatrixZndarrayZgenericdtypeZ issubdtypenumberZbool_rZ asanyarray)arraysiarrayrmessagerQ/var/www/html/assistant/venv/lib/python3.9/site-packages/scipy/_lib/_array_api.pycompliance_scipy%s,      rcCs@d}z|||st|Wnty:t|Yn0dS)zCheck for NaNs or Infs.z#array must not contain infs or NaNsN)allisfinite ValueErrorr)rxpmsgrrr _check_finiteKs   r!cGs,tds tSdd|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 sz#array_namespace..)_GLOBAL_CONFIG np_compatrrr)rrrrrUs r)r check_finitecCs|durt|}|jdvrP|dur4tj|||d}ntj|||d}||}nFz|j|||d}Wn0tyt|d}|j|||d}Yn0|rt|||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*rr'Z coerced_xprrrrzs     rrcCsD|durt|}||}|j|kr@|j|dd}t|||d}|S)z9Recursively expand the dimension to have at least `ndim`.Nraxisndimr)rr-r2Z expand_dims atleast_nd)xr2rrrrr3s  r3cCs|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)rrr4rrrrr*sr*cCs |jdvS)N)rr(r,r.rrris_numpysr7cCs |jdvS)N)Zcupyz scipy._lib.array_api_compat.cupyr6r.rrris_cupysr8cCs |jdvS)N)Ztorchz!scipy._lib.array_api_compat.torchr6r.rrris_torchsr9TcCsrd}|rt||||}|r8d}|j|jks8J||r`d}|j|jksTJ|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_namespacer-rshape _check_scalarZ broadcast_to)actualdesiredrcheck_namespace check_dtype check_shape__tracebackhide___msgrrr _strict_checks   rDcCsZd}t|tr|n|f}t|}|D]0}t|}d|jd|j}||ks$J|q$dS)NTz!Namespaces do not match. Actual: z Desired: )rtuplerr,)r=r>rBZ desired_spaceZarrZ arr_spacerCrrrr:sr:cCsnd}|jdkst|sdS|d}dt|dt|}||rN||sj||sb||rjJ|dS)NTrzTypes do not match: Actual: z Desired: )r;r7typeZisscalar)r=r>rrBrCrrrr<sr<c Csd}|durt|}t||||||d}t|rB|jj|||dSt|rt|dkrVdn|}|jj||dddd|dStjj|||dS)NTr?r@rA)err_msgrGrFrtolatolZ equal_nanr@r )rrDr8testingZassert_array_equalr9 assert_closer)r=r>r?r@rArIrrBrrrxp_assert_equal s rOHz>c Csd} |durt|}t||||||d}t|rF|jj|||||dSt|rx|dkrZdn|}|jj||||dd|dStjj|||||dS)NTrH)rKrLrIrGFrJ)rrDr8rMZassert_allcloser9rNr) r=r>rKrLr?r@rArIrrBrrrxp_assert_closes$   rQc Csd}|durt|}t||||||d}t|rD|jj||||dSt|rt|jjdkr`|}|jjdkrt|}t jj||||dS)NTrH)rIverboser ) rrDr8rMZassert_array_lessr9ZdevicerFr r) r=r>r?r@rArIrRrrBrrrxp_assert_less0s$     rSc Cs|durt|}t||d}|||j}t|d|d}|j||d}|j|dd}|jdd}|dkr~tj dt dd d }||dddf8}|j }| |j d r||}||}||}td d t|jD}|j||dS)Nr.r1)rr/rz!Degrees of freedom <= 0 for slice) stacklevelgcomplex floatingcss|]\}}|dkr|VqdS)rUNr)r"r0lengthrrr [r$zcov..)rr*Z result_typeZfloat64r3r-meanr;warningswarnRuntimeWarningTisdtyperZconjrE enumerateZsqueeze) r4rXrZavgZfactZX_TcZaxesrrrcovCs*  rccCs d|dS)Nz Providing z$ is only supported for numpy arrays.r)paramrrrxp_unsupported_param_msg_srecCs||jdS)NrW)r_rr5rrr is_complexcsrf)NNN)TTT)TTTrGN)rPrTTTrGN)TTTrGTN)&__doc__ __future__rosr[rrZ scipy._librZscipy._lib.array_api_compatrrr&__all__environgetr __annotations__r r%rr!rrr3r*r7r8r9rDr:r<rOrQrSrcrerfrrrrsP  & & &