diff --git a/dask_ml/_typing.py b/dask_ml/_typing.py index cda7c20b5..1c61560f8 100644 --- a/dask_ml/_typing.py +++ b/dask_ml/_typing.py @@ -16,6 +16,6 @@ FrameOrSeriesUnion = Union[DataFrame, Series, dd.Series, dd.DataFrame] SeriesType = Union[dd.Series, Series] DataFrameType = Union[DataFrame, dd.DataFrame] -Number = Union[int, float, np.float64, np.int64, np.int32] +Number = Union[int, np.float64, np.float64, np.int64, np.int32] Int = Union[int, np.int64, np.int32] NDArrayOrScalar = Union[np.ndarray, Sequence[Number], Number]