!j UdZddlZddlZddlmZddlmZddlmZm Z ddl Z ddl mZ ddlmZddlmcmcm ZddlmZmZddlmZeZe ed <ee jzZe ed <ejd d Zeezed <ejd dZ Gddej!Z"edde#de"fdZ$dedefdZ%dS)z| Override functions from array_api_compat, for use by array-api-extra and internally. See also _array_api_compat_vendor.py N) lru_cache) ModuleType)Any TypeAlias)array_api_compat)is_array_api_obj is_jax_array) SparseABCArray ArrayLikeSCIPY_ARRAY_APIF SCIPY_DEVICEcpuceZdZdZdZdZdZdS) _ArrayClsInforN)__name__ __module__ __qualname__skipnumpy array_likeunknownLC:\Projects\DSCons\.venv\Lib\site-packages\scipy\_lib\_array_api_override.pyrr s" D EJGGGrrdclsreturnc Lt|ttfr tjSt|t rd}t |t|tjj rtdt|tj rtdt|tj tj fr tjSt|tt t"t$t'dfr tjStjS)NzSparse 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.) issubclasslisttuplerrr ValueErrornpma MaskedArray TypeErrormatrixndarraygenericrintfloatcomplexbooltyperr)r msgs r_validate_array_clsr4's#u~&&(''#y!!&oo#ru())TRSSS#ry!!LJKKK# BJ/00#"" #UGT4::>??"!!   rarraysctstSg}g}|D]N}tt|}|tjur.|tjurq|jjdvr| |`|jjdkr%t|r| |td|jd|tj ur%t|r| | tj|}n#t$rtdwxYw|jjdvrtd|jd| |P|stSt!jg||RS)alGet 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 ----- Wrapper around `array_api_compat.array_namespace`. 1. Check for the global switch `SCIPY_ARRAY_API`. If disabled, just return array_api_compat.numpy namespace and skip all compliance checks. 2. Check for known-bad array classes. 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 - `scipy.sparse` arrays 3. Coerce array-likes to NumPy arrays and check their dtype. Note that non-scalar array-likes can't be mixed with non-NumPy Array API objects; e.g. - `array_namespace([1, 2])` returns NumPy namespace; - `array_namespace(np.asarray([1, 2], [3, 4])` returns NumPy namespace; - `array_namespace(cp.asarray([1, 2], [3, 4])` raises an error. iufcbVzAn argument has dtype `z3`; only boolean and numerical dtypes are supported.zCAn argument is neither array API compatible nor coercible by NumPy.)r np_compatr4r2rrrdtypekindappendr r*rrr' asanyarrayrarray_namespace)r5 numpy_arrays api_arraysarrayarr_infos rr>r>EsH LJ''&tE{{33 }) ) )  , , ,{7**##E****!S((\%-@-@(!!%((((!S%+!S!S!STTT. . .3CE3J3J .   e $ $ $ $ 7 e,, 7 7 7!6777 7{w..!S%+!S!S!STTT    & & & &   + G\ GJ G G GGs 6D  D%)&__doc__enumos functoolsrtypesrtypingrrrr' numpy.typingnpt scipy._libr!scipy._lib.array_api_compat.numpy_libr9scipy._lib.array_api_compatrr scipy._lib._sparser r __annotations__r environgetr strr1rEnumrr2r4r>rrrrUs  !!!!!!!!''''''555555555555FFFFFFFF((((((ys}, 9,,,!jnn->FFtFFFz~~ne44 DI 3!T!m!!!!:QHUQHzQHQHQHQHQHQHr