1j V d dl mZ ddlmZ ddlmZ d dlZdddd ddZdddd ddZ dS ) ) annotations )Array)_real_numeric_dtypesNFT)axis
descendingstablexr r intr boolr
returnc | j t vrt d |rdnd}|s t j | j || }nkt j t j t j | j | || | }|dk r|n | j |z }| j | dz
}||z
}t j
| S )z~
Array API compatible wrapper for :py:func:`np.argsort `.
See its docstring for more information.
z/Only real numeric dtypes are allowed in argsortr
quicksortr kindr r r ) dtyper TypeErrornpargsort_arrayflipndimshaper _new)r r r r
r resnormalised_axismax_is v/builddir/build/BUILD/cloudlinux-venv-1.0.11/venv/lib64/python3.11/site-packages/numpy/array_api/_sorting_functions.pyr r
s w***IJJJ .88;D
j4888
g Jrwqxd333$T J J J
#'!))$$$ (1,ck :c?? c | j t vrt d |rdnd}t j | j || }|rt j || }t j | S )zx
Array API compatible wrapper for :py:func:`np.sort `.
See its docstring for more information.
z,Only real numeric dtypes are allowed in sortr
r r r ) r r r r sortr r r r )r r r r
r r s r r# r# ' sq w***FGGG .88;D
'!(D
1
1
1C &gc%%% :c??r! )
r r r r r r
r
r
r r )
__future__r
_array_objectr _dtypesr numpyr r r# r! r r) s " " " " " " ) ) ) ) ) )
!#uT < !#uT r!