Indicate that we support both 2021.12 and 2022.12 versions of the Arr…

…ay API
cubed-dev · Jun 21, 2024 · a26fb58 · a26fb58
1 parent e9de1ae
commit a26fb58
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cubed/array_api/array_object.py b/cubed/array_api/array_object.py
@@ -361,7 +361,7 @@ def __abs__(self, /):
         return elemwise(nxp.abs, self, dtype=dtype)
 
     def __array_namespace__(self, /, *, api_version=None):
-        if api_version is not None and not api_version.startswith("2022."):
+        if api_version is not None and api_version not in ("2021.12", "2022.12"):
             raise ValueError(f"Unrecognized array API version: {api_version!r}")
         import cubed.array_api as array_api