cohort_reduction dtype

Author: tomwhite

sgkit/stats/cohort_numba_fns.py

@@ -51,6 +51,7 @@ def func(x: ArrayLike, cohort: ArrayLike, n: int, axis: int = -1) -> ArrayLike:
             x,
             cohort,
             da.empty(n, dtype=np.int8),
+            dtype=x.dtype,  # TODO: is this right? Might need to be passed in.
             chunks=chunks,
             drop_axis=replaced,
             new_axis=replaced,