add array_namespace

Author: mtar

heat/array_api/_array_object.py

@@ -319,6 +319,7 @@ def __and__(self: Array, other: Union[int, bool, Array], /) -> Array:
         res = self._array.__and__(other._array)
         return self.__class__._new(res)
 
+    # TODO: Ready to Remove
     def __array_namespace__(self: Array, /, *, api_version: Optional[str] = None) -> Any:
         """
         Returns an object that has all the array API functions on it.

heat/core/dndarray.py

@@ -10,7 +10,7 @@
 from inspect import stack
 from mpi4py import MPI
 from pathlib import Path
-from typing import Union, TypeVar
+from typing import Any, Union, TypeVar
 
 warnings.simplefilter("always", ResourceWarning)
 
@@ -509,6 +509,24 @@ def __array_function__(self, func, types, args, kwargs):
             return NotImplemented
         return ht_func(*args, **kwargs)
 
+    def __array_namespace__(self, *, api_version: str | None = None) -> Any:
+        """
+        Returns an object that has all the array API functions on it.
+
+        Parameters
+        ----------
+        api_version : Optional[str]
+            string representing the version of the array API specification to
+            be returned, in ``'YYYY.MM'`` form. If it is ``None`` (default), it
+            returns the namespace corresponding to latest version of the
+            array API specification.
+        """
+        if api_version is not None and api_version != "2025.12":
+            raise ValueError(f"Unrecognized array API version: {api_version}")
+        import heat
+
+        return heat
+
     def astype(self, dtype, copy=True) -> DNDarray:
         """
         Returns a casted version of this array.