baloo.functions package

baloo.functions.raw module

baloo.functions.raw.sort(array, **kwargs)

baloo.functions.unary module

baloo.functions.unary.acos(array, weld_type)

baloo.functions.unary.asin(array, weld_type)

baloo.functions.unary.atan(array, weld_type)

baloo.functions.unary.cos(array, weld_type)

baloo.functions.unary.cosh(array, weld_type)

baloo.functions.unary.erf(array, weld_type)

baloo.functions.unary.exp(array, weld_type)

baloo.functions.unary.log(array, weld_type)

baloo.functions.unary.sin(array, weld_type)

baloo.functions.unary.sinh(array, weld_type)

baloo.functions.unary.sqrt(array, weld_type)

baloo.functions.unary.tan(array, weld_type)

baloo.functions.unary.tanh(array, weld_type)

baloo.functions.utils module

baloo.functions.utils.load_cudf(path_to_so)

Dynamically load a C UDF.

Parameters:

path_to_so : str

Absolute path to so.

baloo.functions.utils.raw(func, **func_args)

Decorator for eager functions checking input array and stripping away the weld_type.

Stripping the weld_type is required to keep the same code in Series.apply and because Numpy functions don’t (all) have kwargs. Passing weld_type to NumPy functions is unexpected and raises ValueError.

Parameters:

func : function

Function to execute eagerly over raw data.

func_args : kwargs

Arguments to pass to func, if any.

Returns:

function