-
Notifications
You must be signed in to change notification settings - Fork 11
Description
Arrow and polars default to a NaN is NULL style behavior, while cudf doesn't for aggregations. In the normal reductions (min/max -- maybe also sum/mean, but not sure), we currently account for this inside the task.
Since arrow does this, we should also account for this inside group-by aggregations.
In practice a nans_to_nulls function may be useful, but so long we mix arrow and libcudf it seems likely easier to solve this at the lower rather than higher level with pre-processing.
Similar behavior is that in some places NaN handling in either polars or cudf or arrow might be that NaNs are considered equal to NULLs or not.
Aligning these (similar to for reductions/groupbys) is probably impossible without nans_to_nulls (or work-arounds in other places).