Ng@ddlmZddlmZddlmZddlZdddd ZdS)) annotations)Array)_integer_dtypesNaxisxrindicesr Optional[int]returnc||jdkrtd|jtvrt d|jdkrtdt jtj|j |j |S)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)r r rs _/var/www/html/ai-engine/env/lib/python3.11/site-packages/numpy/array_api/_indexing_functions.pyrrs  |! ?@@@}O++EFFF|q@AAA :bgahTBBB C CC)r rr rrr r r) __future__r _array_objectr_dtypesrnumpyrrrrrsz"""""" $$$$$$?C D D D D D D D Dr