Update _typing.py

pydata · Sep 2, 2024 · 5fccd5f · 5fccd5f
1 parent d05791c
commit 5fccd5f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/xarray/namedarray/_typing.py b/xarray/namedarray/_typing.py
@@ -260,7 +260,9 @@ def __getitem__(
 
     def __array_namespace__(self) -> ModuleType: ...
 
-    def to_device(self, device: _Device, /, stream: None = None) -> Self: ...
+    def to_device(
+        self, device: _Device, /, stream: None = None
+    ) -> _arrayapi[_ShapeType_co, _DType_co]: ...
 
     @property
     def device(self) -> _Device: ...