^jUdZddlZddlZddlmZddlmZddlmZddl Z ddl mZ ddl mZddlmcmcm ZddlmZmZddlmZd dd f Zd dd f Zej2j5d dZeezed <ej2j5ddZGddej@Z!eddde"de!fdZ#dddedefdZ$y)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) ModuleTypeAny)array_api_compat)is_array_api_obj is_jax_array) SparseABCArrayctSNrW/opt/ringagent/.cad-venv/lib/python3.12/site-packages/scipy/_lib/_array_api_override.pyr r sSr ArrayLikec0ttjzSr )r nptrrrrrrs&&rSCIPY_ARRAY_APIF SCIPY_DEVICEcpuceZdZdZdZdZdZy) _ArrayClsInforN)__name__ __module__ __qualname__skipnumpy array_likeunknownrrrrr s D EJGrrdclsreturncnt|ttzrtjSt|t r|s d}t |tjSt|tjjr tdt|tjr tdt|tjtjzrtjSt|tt zt"zt$zt'dzrtj(Stj*S)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.) issubclasslisttuplerr!r ValueErrorr npma MaskedArray TypeErrormatrixndarraygenericintfloatcomplexbooltyperr")r$ sparse_okmsgs r_validate_array_clsr9's#te|$'''#y!*CS/ !"""#ruu(()RSS#ryy!JKK#rzzBJJ./""" #sU{W,t3d4j@A!!!   rr7arraysctstSg}g}|D]>}tt||}|tj ur-|tj ury|jjdvr|j|i|jjdk(rt|r|j|td|jd|tjurt|r|j| tj|}|jjdvrtd|jd|j|A|stSt!j"g||S#t$r tdwxYw)aGet the array API compatible namespace for the arrays xs. Parameters ---------- *arrays : sequence of array_like Arrays used to infer the common namespace. sparse_ok : bool ``True`` if `scipy.sparse` arrays should be accepted where the namespace would otherwise be NumPy. Default: ``False``. 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 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. r: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_compatr9r6rrr dtypekindappendr r.r"rr+ asanyarrayrarray_namespace)r7r; numpy_arrays api_arraysarrayarr_infos rrDrDIsL LJ'&tE{iH })) )  ,, ,{{7*##E*!!S(\%-@!!%("9%++IS!STT.. .3CE3J   e $ 7 e,{{w."9%++IS!STT    &;'B   + + G\ GJ GG 7!677 7s ;E44F )F)%__doc__enumos functoolsrtypesrtypingrr r+ numpy.typingrscipy._externalr&scipy._external.array_api_compat.numpy _externalr? scipy._external.array_api_compatrr scipy._lib._sparser r renvirongetrstrr5__annotations__rEnumrr6r9rDrrrrZs  ,::K(&!jjnn->FtFzz~~ne4 DII 3!T!}!!B/4SHUSH SHr