def create_add_one_struct_udf(result_formatter):
return elementwise(
input_type=[dt.double],
output_type=dt.Struct(['col1', 'col2'], [dt.double, dt.double]),
)(_format_struct_udf_return_type(add_one_struct, result_formatter))
def create_add_one_udf(result_formatter):
return elementwise(input_type=[dt.double],
output_type=dt.double)(_format_udf_return_type(
add_one, result_formatter))