/jwUdZddlZddlZddlmZddlmZddlmZm Z ddl Z ddl mZ ddlmZddlmcmcm ZddlmZmZddlmZeZe ed <ee j4zZe ed <ej6j9d d Zeezed <ej6j9d dZ GddejBZ"edde#de"fdZ$dedefdZ%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) ModuleType)Any TypeAlias)array_api_compat)is_array_api_obj is_jax_array) SparseABCArray ArrayLikeSCIPY_ARRAY_APIF SCIPY_DEVICEcpuceZdZdZdZdZdZy) _ArrayClsInforN)__name__ __module__ __qualname__skipnumpy array_likeunknowna/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/scipy/_lib/_array_api_override.pyrr s D EJGrrdclsreturnc 8t|ttfrtjSt|t r d}t |t|tjjr tdt|tjr tdt|tjtjfrtjSt|tt t"t$t'dfrtj(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.) issubclasslisttuplerrr ValueErrornpma MaskedArray TypeErrormatrixndarraygenericrintfloatcomplexbooltyperr)r msgs r_validate_array_clsr4's#u~&'''#y!&o#ruu(()RSS#ryy!JKK# BJJ/0""" #UGT4:>?!!!   rarraysctstSg}g}|D]<}tt|}|tj ur+|tj ury|jjdvr|j|g|jjdk(rt|r|j|td|jd|tjurt|r|j| tj|}|jjdvrtd|jd|j|?|stSt!j"g||S#t$r tdwxYw)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{3 })) )  ,, ,{{7*##E*!!S(\%-@!!%("9%++IS!STT.. .3CE3J   e $ 7 e,{{w."9%++IS!STT    &;'B   + + G\ GJ GG 7!677 7s 9E22F)&__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  !'55F(ys}}, 9,!jjnn->FtFzz~~ne4 DII 3!T!m!!:QHUQHzQHr