fix types

Author: dcherian

flox/aggregations.py

@@ -162,8 +162,8 @@ def __init__(
         self,
         name: str,
         *,
-        numpy: str | None = None,
-        chunk: str | FuncTuple | None,
+        numpy: partial | str | None = None,
+        chunk: partial | str | FuncTuple | None,
         combine: str | FuncTuple | None,
         preprocess: Callable | None = None,
         finalize: Callable | None = None,

flox/multiarray.py

@@ -1,8 +1,9 @@
+from collections.abc import Callable
 from typing import Self
 
 import numpy as np
 
-MULTIARRAY_HANDLED_FUNCTIONS = {}
+MULTIARRAY_HANDLED_FUNCTIONS: dict[Callable, Callable] = {}
 
 
 class MultiArray: