Source code for xpark.dataset.aggregate

from __future__ import annotations

import ray.data.aggregate

from xpark.dataset.utils import wrap_ray_class, wrap_ray_doc



[docs]
class AggregateFnV2(ray.data.aggregate.AggregateFnV2):
    __doc__ = wrap_ray_doc(ray.data.aggregate.AggregateFnV2.__doc__)



Count = wrap_ray_class(ray.data.aggregate.Count)
Sum = wrap_ray_class(ray.data.aggregate.Sum)
Mean = wrap_ray_class(ray.data.aggregate.Mean)
Max = wrap_ray_class(ray.data.aggregate.Max)
Min = wrap_ray_class(ray.data.aggregate.Min)
Std = wrap_ray_class(ray.data.aggregate.Std)
AbsMax = wrap_ray_class(ray.data.aggregate.AbsMax)
Quantile = wrap_ray_class(ray.data.aggregate.Quantile)
Unique = wrap_ray_class(ray.data.aggregate.Unique)
MissingValuePercentage = wrap_ray_class(ray.data.aggregate.MissingValuePercentage)
ZeroPercentage = wrap_ray_class(ray.data.aggregate.ZeroPercentage)
ApproximateQuantile = wrap_ray_class(ray.data.aggregate.ApproximateQuantile)
ValueCounter = wrap_ray_class(ray.data.aggregate.ValueCounter)
ApproximateTopK = wrap_ray_class(ray.data.aggregate.ApproximateTopK)