arrow-udf-python 0.4.2

Python runtime for Arrow UDFs.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

# // this piece of code is included in the documentation
#
# use arrow_udf_python::{Runtime, CallMode};
# use arrow_schema::{DataType, Field, Schema};
# use arrow_array::{ArrayRef, BooleanArray, Int32Array, RecordBatch};
# use std::sync::Arc;
#
# let mut runtime = Runtime::new().unwrap();
# runtime.add_aggregate(
#     "sum",
#     DataType::Int32,
#     DataType::Int32,
#     CallMode::ReturnNullOnNullInput,
#     r#"
# def create_state():
#     return 0
#
# def accumulate(state, value):
#     return state + value
#
# def retract(state, value):
#     return state - value
#
# def merge(state1, state2):
#     return state1 + state2
# "#,
# )
# .unwrap();
// suppose we have created an aggregate function `sum`
// see the example in `add_aggregate`