Update core.py

pydata · Aug 19, 2024 · c32abe5 · c32abe5
1 parent c1163ad
commit c32abe5
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/xarray/namedarray/core.py b/xarray/namedarray/core.py
@@ -59,6 +59,7 @@
         _AttrsLike,
         _AxisLike,
         _Chunks,
+        _Device,
         _Dim,
         _Dims,
         _DimsLike,
@@ -752,6 +753,20 @@ def __getitem__(self, key: _IndexKeyLike | NamedArray):
         else:
             raise NotImplementedError("{k=} is not supported")
 
+    @property
+    def device(self) -> _Device:
+        """
+        Device of the array’s elements.
+
+        See Also
+        --------
+        ndarray.device
+        """
+        if isinstance(self._data, _arrayapi):
+            return self._data.device
+        else:
+            raise NotImplementedError("self._data missing device")
+
     @property
     def dtype(self) -> _DType_co:
         """