@@ -55,7 +55,7 @@ def __init__(
5555 groupby : pd .Categorical ,
5656 data : Array ,
5757 * ,
58- mask : NDArray [np .bool_ ] | None = None ,
58+ mask : NDArray [np .bool ] | None = None ,
5959 ) -> None :
6060 self .groupby = groupby
6161 if (missing := groupby .isna ()).any ():
@@ -188,7 +188,7 @@ def aggregate( # noqa: PLR0912
188188 func : AggType | Iterable [AggType ],
189189 * ,
190190 axis : Literal ["obs" , 0 , "var" , 1 ] | None = None ,
191- mask : NDArray [np .bool_ ] | str | None = None ,
191+ mask : NDArray [np .bool ] | str | None = None ,
192192 dof : int = 1 ,
193193 layer : str | None = None ,
194194 obsm : str | None = None ,
@@ -337,7 +337,7 @@ def _aggregate(
337337 by : pd .Categorical ,
338338 func : AggType | Iterable [AggType ],
339339 * ,
340- mask : NDArray [np .bool_ ] | None = None ,
340+ mask : NDArray [np .bool ] | None = None ,
341341 dof : int = 1 ,
342342) -> dict [AggType , np .ndarray | DaskArray ]:
343343 msg = f"Data type { type (data )} not supported for aggregation"
@@ -353,7 +353,7 @@ def aggregate_dask_mean_var(
353353 data : DaskArray ,
354354 by : pd .Categorical ,
355355 * ,
356- mask : NDArray [np .bool_ ] | None = None ,
356+ mask : NDArray [np .bool ] | None = None ,
357357 dof : int = 1 ,
358358) -> MeanVarDict :
359359 mean = aggregate_dask (data , by , "mean" , mask = mask , dof = dof )["mean" ]
@@ -374,7 +374,7 @@ def aggregate_dask(
374374 by : pd .Categorical ,
375375 func : AggType | Iterable [AggType ],
376376 * ,
377- mask : NDArray [np .bool_ ] | None = None ,
377+ mask : NDArray [np .bool ] | None = None ,
378378 dof : int = 1 ,
379379) -> dict [AggType , DaskArray ]:
380380 if not isinstance (data ._meta , CSBase | np .ndarray ):
@@ -465,7 +465,7 @@ def aggregate_array(
465465 by : pd .Categorical ,
466466 func : AggType | Iterable [AggType ],
467467 * ,
468- mask : NDArray [np .bool_ ] | None = None ,
468+ mask : NDArray [np .bool ] | None = None ,
469469 dof : int = 1 ,
470470) -> dict [AggType , np .ndarray ]:
471471 groupby = Aggregate (groupby = by , data = data , mask = mask )
@@ -549,7 +549,7 @@ def _combine_categories(
549549def sparse_indicator (
550550 categorical : pd .Categorical ,
551551 * ,
552- mask : NDArray [np .bool_ ] | None = None ,
552+ mask : NDArray [np .bool ] | None = None ,
553553 weight : NDArray [np .floating ] | None = None ,
554554) -> sparse .coo_matrix :
555555 if mask is not None and weight is None :
0 commit comments