o 1ήc@sLddlmZddlmZddlZdddddddZdddddddZdS)) annotations)ArrayNFT)axis descendingstablexrrintrboolrreturncCs|rdnd}|stj|j||d}n)tjtjtj|j|d||d|d}|dkr+|n|j|}|j|d}||}t|S)z~ Array API compatible wrapper for :py:func:`np.argsort `. See its docstring for more information. r quicksortrkindrrr)npargsort_arrayflipndimshaper_new)r rrrrresnormalised_axismax_irI/tmp/pip-target-vg8gfxp4/lib/python/numpy/array_api/_sorting_functions.pyr s  rcCs:|rdnd}tj|j||d}|rtj||d}t|S)zx Array API compatible wrapper for :py:func:`np.sort `. See its docstring for more information. rr rr)rsortrrrr)r rrrrrrrrr$s  r) r rrr rr rr r r) __future__r _array_objectrnumpyrrrrrrrs