From f07d55b27671a4575e3b9b2fc7ca9ec897d4db9e Mon Sep 17 00:00:00 2001
From: Alex Rogozhnikov <arogozhnikov@users.noreply.github.com>
Date: Sun, 26 Feb 2023 07:56:47 +0000
Subject: add support for xp.take

---
 numpy/array_api/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'numpy/array_api/__init__.py')

diff --git a/numpy/array_api/__init__.py b/numpy/array_api/__init__.py
index 5e58ee0a8..e154b9952 100644
--- a/numpy/array_api/__init__.py
+++ b/numpy/array_api/__init__.py
@@ -333,6 +333,10 @@ __all__ += [
     "trunc",
 ]
 
+from ._indexing_functions import take
+
+__all__ += ["take"]
+
 # linalg is an extension in the array API spec, which is a sub-namespace. Only
 # a subset of functions in it are imported into the top-level namespace.
 from . import linalg
-- 
cgit v1.2.1