typing

pydata · Sep 25, 2024 · 2cec2c7 · 2cec2c7
1 parent 0f26d51
commit 2cec2c7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/xarray/namedarray/_array_api/_utils.py b/xarray/namedarray/_array_api/_utils.py
@@ -497,7 +497,7 @@ def _atleast1d_dims(dims: _Dims) -> _Dims:
     return (_new_unique_dim_name(dims),) if len(dims) < 1 else dims
 
 
-def _reduce_dims(dims: _Dims, *, axis: _AxisLike | None, keepdims: False) -> _Dims:
+def _reduce_dims(dims: _Dims, *, axis: _AxisLike | None, keepdims: bool) -> _Dims:
     """
     Reduce dims according to axis.
 
@@ -524,7 +524,8 @@ def _reduce_dims(dims: _Dims, *, axis: _AxisLike | None, keepdims: False) -> _Di
     else:
         _axis = _normalize_axis_tuple(axis, ndim)
 
-    key = [slice(None)] * ndim
+    k: _IndexKeys = (slice(None),) * ndim
+    key = list(k)
     for i, v in enumerate(_axis):
         key[v] = 0