Update daskmanager.py

pydata · Apr 14, 2024 · c2fb035 · c2fb035
1 parent 5c4ecab
commit c2fb035
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/xarray/namedarray/daskmanager.py b/xarray/namedarray/daskmanager.py
@@ -90,10 +90,12 @@ def from_array(
 
     def compute(
         self, *data: chunkedduckarray[Any, _DType] | Any, **kwargs: Any
-    ) -> tuple[np.ndarray[Any, np.dtype[np.generic]], ...]:
+    ) -> tuple[duckarray[Any, _DType], ...]:
         from dask.base import compute
 
-        return compute(*data, **kwargs)  # type: ignore[no-untyped-call, no-any-return]
+        out: tuple[np.ndarray[Any, np.dtype[np.generic]], ...]
+        out = compute(*data, **kwargs)  # type: ignore[no-untyped-call]
+        return out
 
     @property
     def array_api(self) -> ModuleType: