Update core.py

pydata · Oct 14, 2023 · f5bc328 · f5bc328
1 parent 117b850
commit f5bc328
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xarray/namedarray/core.py b/xarray/namedarray/core.py
@@ -535,10 +535,10 @@ def _nonzero(self: T_NamedArrayInteger) -> tuple[T_NamedArrayInteger, ...]:
         # TODO: we should replace dask's native nonzero
         # after https://github.com/dask/dask/issues/1076 is implemented.
         # TODO: cast to ndarray and back to T_DuckArray is a workaround
-        nonzeros = np.nonzero(cast(NDArray[np.integer[Any]], self.data))
+        nonzeros = np.nonzero(cast("NDArray[np.integer[Any]]", self.data))
         _attrs = self.attrs
         return tuple(
-            cast(T_NamedArrayInteger, _new(self, (dim,), nz, _attrs))
+            cast("T_NamedArrayInteger", _new(self, (dim,), nz, _attrs))
             for nz, dim in zip(nonzeros, self.dims)
         )