o e@s@ddlmZddlmZddlmZddlZdddd dZdS)) annotations)Array)_integer_dtypesNaxisxrindicesr Optional[int]returncCsX|dur |jdkr td|jtvrtd|jdkrtdttj|j |j |dS)zx Array API compatible wrapper for :py:func:`np.take `. See its docstring for more information. Nrz$axis must be specified when ndim > 1z+Only integer dtypes are allowed in indexingz%Only 1-dim indices array is supportedr) ndim ValueErrordtyper TypeErrorr_newnptake_array)rr rrV/var/www/html/venv/lib/python3.10/site-packages/numpy/array_api/_indexing_functions.pyrs  r)rrr rrr r r) __future__r _array_objectr_dtypesrnumpyrrrrrrs