From 6b95e44dd9611590e5c29bfda1cbcb2606a7af2e Mon Sep 17 00:00:00 2001 From: Aditya Goel Date: Mon, 1 Jul 2024 18:45:46 +0100 Subject: [PATCH 1/2] Update __array_namespace__ --- ndonnx/_array.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/ndonnx/_array.py b/ndonnx/_array.py index 7d415f3..2067261 100644 --- a/ndonnx/_array.py +++ b/ndonnx/_array.py @@ -329,8 +329,14 @@ def spox_var(self) -> Var: return self._core().var @staticmethod - def __array_namespace__(): - return ndx + def __array_namespace__(*, api_version: str | None = None): + if api_version is None: + api_version = "2023.12" + + if api_version == "2022.12": + return ndx + else: + raise ValueError(f"Unsupported API version {api_version}") def __float__(self) -> float: eager_value = self.to_numpy() From 8502d8f1c0728587687bff5f1eef9d3056f76645 Mon Sep 17 00:00:00 2001 From: Aditya Goel Date: Mon, 1 Jul 2024 18:49:25 +0100 Subject: [PATCH 2/2] Update version --- CHANGELOG.rst | 4 ++++ ndonnx/_array.py | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/CHANGELOG.rst b/CHANGELOG.rst index 7deff86..68765db 100644 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -13,6 +13,10 @@ Changelog - ndonnx now exports type annotations. +**Bug fixes** + +- ``__array_namespace__`` now accepts the optional ``api_version`` argument to specify the version of the Array API to use. + 0.4.0 (2024-05-16) ------------------ diff --git a/ndonnx/_array.py b/ndonnx/_array.py index 2067261..cc0e727 100644 --- a/ndonnx/_array.py +++ b/ndonnx/_array.py @@ -333,7 +333,7 @@ def __array_namespace__(*, api_version: str | None = None): if api_version is None: api_version = "2023.12" - if api_version == "2022.12": + if api_version == "2023.12": return ndx else: raise ValueError(f"Unsupported API version {api_version}")