From 0ac7de9a670517a46ee82670d3a790dafbb6071c Mon Sep 17 00:00:00 2001 From: Aaron Meurer Date: Tue, 9 Mar 2021 17:01:41 -0700 Subject: Implement __array_namespace__ on the array API ndarray object --- numpy/_array_api/_array_object.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'numpy/_array_api') diff --git a/numpy/_array_api/_array_object.py b/numpy/_array_api/_array_object.py index 64ce740f0..84cbf3527 100644 --- a/numpy/_array_api/_array_object.py +++ b/numpy/_array_api/_array_object.py @@ -87,6 +87,12 @@ class ndarray: res = x1._array.__and__(asarray(x2)._array) return x1.__class__._new(res) + def __array_namespace__(self, /, *, api_version=None): + if api_version is not None: + raise ValueError("Unrecognized array API version") + from numpy import _array_api + return _array_api + def __bool__(x: array, /) -> bool: """ Performs the operation __bool__. -- cgit v1.2.1