fix in namedarray

pydata · Dec 12, 2023 · 48caa38 · 48caa38
1 parent 6231ae2
commit 48caa38
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/xarray/namedarray/core.py b/xarray/namedarray/core.py
@@ -642,6 +642,14 @@ def _dask_finalize(
         data = array_func(results, *args, **kwargs)
         return type(self)(self._dims, data, attrs=self._attrs)
 
+    @overload
+    def get_axis_num(self, dim: Iterable[Hashable]) -> tuple[int, ...]:
+        ...
+
+    @overload
+    def get_axis_num(self, dim: Hashable) -> int:
+        ...
+
     def get_axis_num(self, dim: Hashable | Iterable[Hashable]) -> int | tuple[int, ...]:
         """Return axis number(s) corresponding to dimension(s) in this array.