fixes

pydata · Oct 14, 2023 · 8fe9828 · 8fe9828
1 parent d126abc
commit 8fe9828
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/xarray/namedarray/core.py b/xarray/namedarray/core.py
@@ -538,7 +538,8 @@ def _nonzero(self: T_NamedArrayInteger) -> tuple[T_NamedArrayInteger, ...]:
         nonzeros = np.nonzero(cast(NDArray[np.integer[Any]], self.data))
         _attrs = self.attrs
         return tuple(
-            _new(self, (dim,), nz, _attrs) for nz, dim in zip(nonzeros, self.dims)
+            (T_NamedArrayInteger, _new(self, (dim,), nz, _attrs))
+            for nz, dim in zip(nonzeros, self.dims)
         )
 
     def _as_sparse(

diff --git a/xarray/tests/test_namedarray.py b/xarray/tests/test_namedarray.py
@@ -6,7 +6,7 @@
 import pytest
 
 import xarray as xr
-from xarray.namedarray._typing import _arrayfunction_or_api
+from xarray.namedarray._typing import _arrayfunction_or_api, _ShapeType_co
 from xarray.namedarray.core import NamedArray, from_array
 
 if TYPE_CHECKING:
@@ -20,7 +20,6 @@
         _DType_co,
         _ScalarType,
         _Shape,
-        _ShapeType_co,
         duckarray,
     )