ray.data.aggregate.AggregateFn#

class ray.data.aggregate.AggregateFn(init: ~typing.Callable[[~ray.data.block.KeyType], ~ray.data.block.AggType], merge: ~typing.Callable[[~ray.data.block.AggType, ~ray.data.block.AggType], ~ray.data.block.AggType], accumulate_row: ~typing.Callable[[~ray.data.block.AggType, ~ray.data.block.T], ~ray.data.block.AggType] = None, accumulate_block: ~typing.Callable[[~ray.data.block.AggType, pyarrow.Table | pandas.DataFrame], ~ray.data.block.AggType] = None, finalize: ~typing.Callable[[~ray.data.block.AggType], ~ray.data.block.U] = <function AggregateFn.<lambda>>, name: str | None = None)[source]#

Methods

__init__

Defines an aggregate function in the accumulator style.