fix typing

xarray-contrib · Oct 21, 2024 · bd889ec · bd889ec
1 parent e4ef7b8
commit bd889ec
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/flox/aggregations.py b/flox/aggregations.py
@@ -149,9 +149,9 @@ def size(self) -> int:
 class Aggregation:
     def __init__(
         self,
-        name,
+        name: str,
         *,
-        numpy: str | FuncTuple | None = None,
+        numpy: str | None = None,
         chunk: str | FuncTuple | None,
         combine: str | FuncTuple | None,
         preprocess: Callable | None = None,
@@ -217,7 +217,7 @@ def __init__(
         self.preprocess = preprocess
         # Use "chunk_reduce" or "chunk_argreduce"
         self.reduction_type = reduction_type
-        self.numpy: FuncTuple = (numpy,) if numpy else (self.name,)
+        self.numpy: FuncTuple = (numpy,) if numpy is not None else (self.name,)
         # initialize blockwise reduction
         self.chunk: OptionalFuncTuple = _atleast_1d(chunk)
         # how to aggregate results after first round of reduction