\ciUdZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZmZddlmZddlZddlm Zdd lmZmZmZmZmZmZmZmZmZ m!Z"m#Z$m%Z&m'Z(m)Z*m+Z,m-Z.dd l/m0Z0dd l1m2Z2dd l3m4Z4m5Z5m6Z6dd l7m8Z8ddl9m:Z;gdZd<e=ej?zZ?ee>d<de=de ddfdZ@ d%ddddde?dededdzdeAdzde dzdeAdeAde=fdZBddd e=de dzde=fd!ZCd e=dejDfd"ZEe d#ZFe e e>d$<ede ded%fd&ZGdd'd(ZHd)d)d)d)d*d+ZId,ZJd)d)d)d)d-dd.d/ZKddd)d)d)d)d-dd0d1ZLd2d)d)d)d)d-dd3d4ZMd)d)d)d)d-d)dd5d6ZNddd8ZOded:ZPd;edeQfd<ZRd e=de deAfd=ZSde deQfd>ZTde de dzfd?ZUddd@ddAd e=dBeVeWeVzdzdCeAdDeVeXzde dzde=f dEZYddd e=de dzde=fdFZZdfdGZ[ddHdIZ\dddJdKZ]dfdLe=de dzde=fdMZ^dNZ_dOZ`de dzdPee?de=fdQZadRZbdgdSZcdTZdejedUGdVdWZf dhdYZgdfdZZhddXdXdddddXdXdd)dd[ d\Zidd]d^Zjdd]d_Zkdd]d`ZliZmdae=dedbfdcZndS)ia6Utility 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) Generator)contextmanager) ContextVar) ModuleType)AnyLiteral TypeAlias)Iterable)is_array_api_obj is_lazy_arrayis_numpy_array is_cupy_arrayis_torch_array is_jax_array is_dask_arraysizenumpydeviceis_numpy_namespaceis_cupy_namespaceis_torch_namespaceis_jax_namespaceis_dask_namespaceis_array_api_strict_namespace)_compat_module_name)lazy_xp_function)array_namespaceSCIPY_ARRAY_API SCIPY_DEVICE) FunctionDoc)array_api_extra)"_asarrayrassert_almost_equalassert_array_almost_equal default_xp eager_warnsr is_marrayis_array_api_strict is_complexis_cupyis_jaxis_numpyis_torch np_compatget_native_namespace_namerrscipy_namespace_forxp_assert_closexp_assert_equalxp_assert_lessxp_copy xp_devicexp_ravelxp_sizexp_unsupported_param_msgxp_vector_normxp_capabilitiesxp_result_type xp_promotemake_xp_test_casemake_xp_pytest_marksmake_xp_pytest_paramArray ArrayLikearrayxpreturncx|||sd}t|dS)zCheck for NaNs or Infs.z#array must not contain infs or NaNsN)allisfinite ValueError)rBrCmsgs e/var/www/html/mdtn/previsions/meteo_cartes/venv/lib/python3.11/site-packages/scipy/_lib/_array_api.py _check_finiterKEs= 66"++e$$ % %3ooF)rC check_finitesubokdtypeorder)KACFcopyrMrNc|t|}t|rO|durtj||||}n|rtj|||}n|tj|||}nd ||||}nJ#t $r=t|d}||||}YnwxYw|rt|||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)rPrOrN)rPrO)rOrU)rr,nprB asanyarrayasarray TypeErrorrK)rBrOrPrUrCrMrN coerced_xps rJr"r"Ks0 z U # #|| F 4<<HU%uEJJJEE  @M%uEBBBEEJuE???EE FJJuEJ==EE F F F(A77J&&uE&EEEEE F!eR   Ls1B ACCrCxcH|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. NT)rUrC)rr"r^rCs rJr4r4zs+, z Q   ADR ( ( ((rLct|}t|r|St|r|St |r&|St|r>tj | || ddStj |dS)aCopies a possibly on device array to a NumPy array. This function is intended only for converting alternative backend arrays to numpy arrays within test code, to make it easier for use of the alternative backend to be isolated only to the function being tested. `_xp_copy_to_numpy` should NEVER be used except in test code for the specific purpose mentioned above. In production code, attempts to copy device arrays to NumPy arrays should fail, or else functions may appear to be working on the GPU when they actually aren't. Parameters ---------- x : array Returns ------- ndarray CPU_DEVICE)rT)rU) rr,rUr*getr-cpurr(rXrZDevicer`s rJ_xp_copy_to_numpyrfs&   B||vvxxr{{uuww||uuww}}2 z JJqAmatchcddl}ddlm}t|st |st |r|||S|||dn|S)zpytest.warns context manager if arrays of specified namespace are always eager. Otherwise, context manager that *ignores* specified warning. rN) ignore_warnsrm)pytestscipy._lib._utilrpr,r(r*warns) warning_typernrCrrrps rJr&r&s} MMM------||7*2..7'"++7||L|666 < %-BBU K K KKrLTcheck_namespace check_dtype check_shapecheck_0dcd}|: t}n#t$rt|}YnwxYw|rt |||t |r|rdt |dt |}||r||s4||s||r J|||}||}|r,d|j d|j }|j |j ks J||rct|r(| | d|j d|j }|j |j ks J|| ||j }|||fS)NTz$Array-ness does not match: Actual: z Desired: zdtypes do not match. Actual: Desired: zShapes do not match. Actual: )rhrc LookupErrorr_assert_matching_namespacer,typeisscalarrZrOis_daskcompute_chunk_sizesshape broadcast_to) actualdesiredrCrwrxryrz__tracebackhide___msgs rJ _strict_checkrs z *#''))BB * * * ))BBB *8"67B777||QQ=<<==-1']]==V$$ QW)=)= Q F++ Q46KK4H4H Q QLP Q Q QZZ  Fjj!!G3X XXXX|w},,,d,,,3 2;; *  & & ( ( (  ' ' ) ) )X XXXX|w},,,d,,,oogv|44G 7B s  <<cd}t|}d|jd|j}||ks J|t|}d|jd|j}||ks J|dS)NTzNamespace of desired array does not match expectations set by the `default_xp` context manager or by the `xp`pytest fixture. Desired array's space: z Expected namespace: z=Namespace of actual and desired arrays do not match. Actual: r|)r__name__)rrrCrdesired_arr_spaceractual_arr_spaces rJr~r~s'00 1'8&@ 1 1$&;  1 1D  " " "D " " "&v.. &'0 & &  & &D r ! ! !4 ! ! ! ! !rLrq)rwrxryrzerr_msgrCc Jd}t|||||||\}}}t|r|j|||St |r+|dkrdn|}|j||dddd|St j|||S)NTrv)rrqrFrtolatol equal_nanrxrI)rr*testingassert_array_equalr- assert_closerX) rrrwrxryrzrrCrs rJr2r2s'_[FGR r{{Gz,,VWg,NNN "G"R--$$Wz&&vwQQRV38g'GG G : ( (' ( J JJrL)rrrwrxryrzrrCc d} t||| ||||\}}} | |jd} |(| r&| |jjdzdz}n|d}t | r| j|||||St| r+|dkrdn|}| j ||||dd | Stj|||||S) NTrv real floatingcomplex floating?gHz>)rrrrqFr) risdtyperOfinfoepsr*rassert_allcloser-rrX) rrrrrwrxryrzrrCrfloatings rJr1r1.sA''[(FGR zz&,(MNNH ||xx %%)3.2 r{{Wz))&'/3W*FF F "W!R--$$Wz&&vwT155g'WW W : % %fgD+/ & B BBrL)nulprwrxryrzrrCc d} t|||||||\}}}tt||f\}}tj|||S)NTrv)r)rmaprfrXrassert_array_almost_equal_nulp) rrrrwrxryrzrrCrs rJxp_assert_close_nulprNsn''[(FGR +fg->??OFG : 4 4VW4 4 P PPrL)rwrxryrzrverboserCc d} t|||||||\}}}t|r|j||||St |rH|jjdkr|}|jjdkr|}tj||||S)NTrv)rrrd) rr*rassert_array_lessr-rrrdrX) rrrwrxryrzrrrCrs rJr3r3]s'_[FGR r{{$z++FG4;W,NN N "$ =  & &ZZ\\F > % ' 'kkmmG : ' '07 ( J JJrLcDddd| zz}}t||g|R||ddd|SzPBackwards compatible replacement. In new code, use xp_assert_close instead. rg? F)rrrxryr1rrdecimalargskwdsrrs rJr$r$tWCgX&$D 67 *  * * $4UPU * *$( * **rLcDddd| zz}}t||g|R||ddd|Srrrs rJr#r#}rrLparamcd|dS)Nz Providing z$ is only supported for numpy arrays.)rs rJr8r8s E E E EErLc8||jdS)Nr)rrOr`s rJr)r)s ::ag1 2 22rLcX|j}|tdS)zCReturn name for native namespace (without array_api_compat prefix)..)r removeprefixr)rCnames rJr/r/s- ;D    3 5 5888 9 99rLct|r ddl}|jSt|r ddl}|jSt |r|SdS)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)r*cupyxscipyr+jaxr-)rCrrs rJr0r0sZr{{ { bzz y|| 4rL)axiskeepdimsordrCrrrcd|t|n|}trtt|dr|j||||S|dkrt d||||z||dzStj ||||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_normrHsumconjrXnorm)r^rrrrCs rJr9r9s "z   rBH 2x M9((c(RR Raxx !66"''!**q.th6GGL Ly~~aSth~GGGrLcT|t|n|}||dS)N))rreshaper`s rJr6r6s. "z   rB ::a  rLc|t|n|}tt|j}||||c||<||<|||}|SN)rlistrangendim permute_dims)aaxis1axis2rCaxess rJ xp_swapaxesrsa!z   rB af  D#E{DKDKe 4  A HrL)force_floatingcfd|D}d|D}|r|dtr!jdkrd|D}j|S j|S#t$r|sg}|D]i}t j|r|n|}t|d|} |dr||jjg|tRcYSwxYw)a Returns the dtype that results from applying type promotion rules (see Array API Standard Type Promotion Rules) to the arguments. Augments standard `result_type` in a few ways: - There is a `force_floating` argument that ensures that the result type is floating point, even when all args are integer. - When a TypeError is raised (e.g. due to an unsupported promotion) and `force_floating=True`, we define a custom rule: use the result type of the default float and any other floats passed. See https://github.com/scipy/scipy/pull/22695/files#r1997905891 for rationale. - This function accepts array-like iterables, which are immediately converted to the namespace's arrays before result type calculation. Consequently, the result dtype may be different when an argument is `1.` vs `[1.]`. Typically, this function will be called shortly after `array_namespace` on a subset of the arguments passed to `array_namespace`. c~g|]9}t|stj|rt|dn|:ST)rNrCrrXiterabler".0argrCs rJ z"xp_result_type..` ' ' '2@1D1D TWHXHXXc" - - - - ' ' 'rLcg|]}||Srrrrs rJrz"xp_result_type..<<.sD333 '.c61&=&=&B&B333rLrOr) appendr, __version__ result_typer[rXrrZrrxp_default_dtype)rrCr args_not_none float_argsr arg_arrayrOs ` rJr;r;s. ' ' ' '!% ' ' 'D<.$rrL)rrCc>g|]}|t|dn|S)NT)rOrNrC)r")rrrOrCs rJrzxp_promote..)sE   ?BoXcdr : : : :SV   rLrrcg|]}||Srrrs rJrzxp_promote../rrLch|] }|j Sr)rrs rJ zxp_promote..2s 1 1 1Cci 1 1 1rLz/Array shapes are incompatible for broadcasting.NrNT) r;lentuplerXbroadcast_shapesrrHrr,rrOastype) rrrCrrshapesremessageoutrkwargsrOs ` @rJr<r< s$   ' ' ' '!% ' ' 'D DB G G GE        D 8d))Q,,tAwwE$KK7<> ! ! ! " 4rLarrayscJfd|D}|S)zbA replacement for `np.r_` as `xp.concat` does not accept python scalars or 0-D arrays. cdg|],}tj|d-S)r)rrC)xpx atleast_ndrZ)rrrCs rJrzconcat_1d..}s4 I I IQCN2::a==qR 8 8 8 I I IrL)concat)rCraryss` rJ concat_1drys/ J I I I& I I ID 99T??rLcd|jvS)z=Returns True if `xp` is an MArray namespace; False otherwise.marray)rr]s rJr'r's r{ ""rLc|t|n|}t|rVtj|rd}t |||||||jS|t|nOttj tj |j tj |S)Nz8`axis` must be an integer or None for use with `MArray`.r) rr'rXrNotImplementedErrorrcountrOr7intprodrZr)r^rrrCrs rJ_length_nonmaskedrs!z   rB}}M ;t   /PG%g.. .yy!$BBAGLLL,GAJJJ  17++BJt,<,<=>> ? ?ArLctr8tjtjd|Dfd|D}t |dkr|dn|S)Nc3$K|] }|jV dSrmaskrs rJ z_share_masks..s$.H.HCsx.H.H.H.H.H.HrLcHg|]}|jS)r")rZdata)rrr#rCs rJrz _share_masks..s+@@@C 38$ //@@@rLrr)r' functoolsreduceoperatoror_r)rCrr#s` @rJ _share_masksr+sn}}A .H.H4.H.H.HII@@@@@4@@@$ii1nn477$.rL)reprcteZdZUedzed<edzed<ejeZee ed<dZ dZ dS)_XPSphinxCapabilityNrdgpu)default_factorywarningsc|dS|sdS|jr>>#6>>>JurLc|||j}||j}|dd|dS)N20z )r8rdr/)r5rdr/s rJ__str__z_XPSphinxCapability.__str__sBll48$$ll48$$$$$C$$$$rL) r __module__ __qualname__bool__annotations__ dataclassesfieldrr1strr8r;rrLrJr.r.sx  +++DAAAHd3iAAA   %%%%%rLr.rc |rddiSt|}tddtddtddtddtdd|rgndgtdd|rdgngd} t|t|zD])\} } | | } | jd| _| jd| _*| D]@\} } |r%| |d hzvr| jd| _| jd| _,|r| |vr| jd| _A|D]'\} }| | } | j|(| S) N out_of_scopeT)rdr/zno JIT)rdr/r1zcomputes graph)rarray_api_strictcupytorch jax.numpy dask.arrayFr)rjr.rrdr/itemsr1r) skip_backendsxfail_backendscpu_onlynp_onlyrD exceptionsallow_dask_computejax_jitr1reason capabilitiesmodule_backendwarnings rJ_make_sphinx_capabilitiesrXs&%%ZJ%4888/DdCCC#$777$4888(Tt"2RR 444*d+=E&''2GGG  L-((4+?+??  v& ; "GK ; "GK'--//    vZ7)%;;;{&# {&#  & 22w{7NGK#))v&(((( rLc d|vrd|d}tj|Sd|d|dd|dd |d d |d d |dd |pdzdz}tj|S)NrDz2 **Array API Standard Support** `z` is not in-scope for support of Python Array API Standard compatible backends other than NumPy. See :ref:`dev-arrayapi` for more information. z* **Array API Standard Support** `aG` has experimental support for Python Array API Standard compatible backends in addition to NumPy. Please consider testing these features by setting an environment variable ``SCIPY_ARRAY_API=1`` and providing CuPy, PyTorch, JAX, or Dask arrays as array arguments. The following combinations of backend and device (or other capability) are supported. ==================== ==================== ==================== Library CPU GPU ==================== ==================== ==================== NumPy rz CuPy rFz PyTorch rGz JAX rHz Dask rIzK ==================== ==================== ==================== rqz1 See :ref:`dev-arrayapi` for more information.)textwrapdedent)fun_namerS extra_notenotes rJ_make_capabilities_noter_s%%     t$$$   (0  (/  (0  ( 4   ( 5!   &  ' & S' SD* ?4  rL) capabilities_tablerKrLrMrNrRrDrOr1rPrQr]c  tn|rd}t|||||||| | | tdi  fd} | S)a}Decorator for a function that states its support among various Array API compatible backends. This decorator has two effects: 1. It allows tagging tests with ``@make_xp_test_case`` or ``make_xp_pytest_param`` (see below) to automatically generate SKIP/XFAIL markers and perform additional backend-specific testing, such as extra validation for Dask and JAX; 2. It automatically adds a note to the function's docstring, containing a table matching what has been tested. See Also -------- make_xp_test_case make_xp_pytest_param array_api_extra.testing.lazy_xp_function NT) rKrLrMrNrDrRrOrPrQr1c<|<t|j}t|}|d|t |dddd} ||_n#t$rYnwxYw|S)NNotes rz ) r_rr rrBsplitlstrip__doc__AttributeError)fr^docrSr`r]sphinx_capabilitiess rJ decoratorz"xp_capabilities..decoratorHs!-1&qz3F SS!nn G D!!!#hhnnT1%%a(//66 AII    D  sB BBr)xp_capabilities_tabledictrX)r`rKrLrMrNrRrDrOr1rPrQr]rlrSrks` ` @@rJr:r:sH4F3M//1#%!-   L4CClCC rLr`c>|tn|} t|d|ifdS)Nr`c2tjd|S)Nc||Srr)rigs rJz5make_xp_test_case....saaddrL)r'r()funcmarkss rJrtz#make_xp_test_case..s ():):E4HHrL)rmr>)r`funcsrvs @rJr=r=[sA3E3M//1*V !% Orr)rur`rrrrvs rJr?r?sGZMMM :L M M ME 6< Ct C C5T] C C CCrLcr |tn|}ddl}g}|D] }|| d} d} dr0||jd|| dr0||jd||  d D]4\}}||j|| 5 d D]4\}}||j|| 5 fd dD}t |fi|"|S)aVariant of ``make_xp_test_case`` that returns a list of pytest marks, which can be used with the module-level `pytestmark = ...` variable:: pytestmark = make_xp_pytest_marks(f1, f2) def test(xp): ... In this example, the whole test module is dedicated to testing `f1` or `f2`, and the two functions have the same capabilities, so it's unnecessary to cherry-pick which test tests which function. The above is equivalent to:: pytestmark = [ pytest.mark.skip_xp_backends(...), pytest.mark.xfail_xp_backends(...), ...]), ] def test(xp): ... See Also -------- xp_capabilities make_xp_test_case make_xp_pytest_param array_api_extra.testing.lazy_xp_function NrrOrRrMT)rMrOrRrN)rNrOrRrK)rRrLc"i|] }|| Srr)rkrSs rJ z(make_xp_pytest_marks..s5CCC,q/CCCrL)rPrQ)rmrrrmarkskip_xp_backendsxfail_xp_backendsr) r`rwrrrvrurOrRmod_name lazy_kwargsrSs @rJr>r>s:4F3M//1MMM E..)$/ !,/ h'  # F LL55*V6EE F F F  " E LL55F6DD E E E!-_ = P P Hf LL55hv5NN O O O O ,-= > Q Q Hf LL66x6OO P P P PCCCC ACCC ------ LrLr)rdcudaNct|rdSt|rdSt|r |jjSt |r|jjx}dkrdn|St|rt|j SdS)Nrdrr/) r rrrrrplatformrxp_device_type_meta)rps rJrrsauQvax}AB x00!U::vvAQ'ag&&& 4rL)r)rr)FN) rrFFFrFTrN)orgr)r@r'rZcollections.abcr contextlibr contextvarsrtypesrtypingrrr r rrX numpy.typingnptscipy._lib.array_api_compatr r r rrrrrr7r.rr5rr,rr*rr-rr+rrrr(+scipy._lib.array_api_compat.common._helpersr"scipy._lib.array_api_extra.testingrscipy._lib._array_api_overriderrrscipy._lib._docscraper scipy._libr!r__all__r@r?rArKr>r"r4ndarrayrfrhr%r&rr~r2r1rr3r$r#rBr8r)r/r0rrfloatr9r6rr;r<r rrrr'rr+ dataclassr.rXr_r:r=r?r>rmrrrLrJrs%%%%%%%%%%%%""""""**********$$$$$$$LKKKKK??????.-----------   ys}, 9,,,J448 , !%",,,,,)*T1,Tk ,   ,,, ,,,,^26)))u)Z$.)%))))8&$&$2:&$&$&$&$R.8Z -F-FJz*FFF (: (),<"= ( ( ( ( (, L L L L L#'Dd%%%%%P((($9=$ $tRDKKKKK,.24 $$4BBBBB@34T%)td!# Q Q Q Q Q8<#dBQUJJJJJ.********FCFCFFFF3%3Z3D3333:*::::: J:+<048$)&'+/ HHHeHuSz)D0H!He H"D( H5: HHHH46:    d!2 e        */2A2A2A2A2Aj!&eA1A1A1A1A1H  U  T(9 U       *t#hy.Ae### A A A A///E"""%%%%%%%#"%0&(BD&* 3333l%!%!%!%!XR E$2 d#HHHHHV26/I/I/I/I/Id:>0D0D0D0D0Df5977777ve(; <rL