a ;e@sJddlmZddlmZddlmZddlZddddddd d d ZdS) ) annotations)Array)_integer_dtypesNaxisrz Optional[int])xindicesrreturncCsX|dur|jdkrtd|jtvr,td|jdkr>tdttj|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 ValueErrorZdtyper TypeErrorr_newnptakeZ_array)rr rrW/var/www/sunrise/env/lib/python3.9/site-packages/numpy/array_api/_indexing_functions.pyrs  r) __future__rZ _array_objectrZ_dtypesrnumpyrrrrrrs