Skip to content

Commit 6a89977

Browse files
committed
Refactor imports in udf.rs for improved organization and clarity
1 parent 5fb08d6 commit 6a89977

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

src/udf.rs

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -22,23 +22,20 @@ use datafusion_ffi::udf::{FFI_ScalarUDF, ForeignScalarUDF};
2222
use pyo3::types::PyCapsule;
2323
use pyo3::{prelude::*, types::PyTuple};
2424

25+
use crate::errors::to_datafusion_err;
26+
use crate::errors::{py_datafusion_err, PyDataFusionResult};
27+
use crate::expr::PyExpr;
28+
use crate::utils::{parse_volatility, validate_pycapsule};
2529
use datafusion::arrow::array::{make_array, Array, ArrayData, ArrayRef};
2630
use datafusion::arrow::datatypes::{DataType, Field};
2731
use datafusion::arrow::pyarrow::FromPyArrow;
2832
use datafusion::arrow::pyarrow::{PyArrowType, ToPyArrow};
2933
use datafusion::error::DataFusionError;
30-
use datafusion::logical_expr::function::ScalarFunctionImplementation;
31-
use datafusion::logical_expr::ptr_eq::PtrEq;
3234
use datafusion::logical_expr::{
33-
ColumnarValue, ReturnFieldArgs, ScalarFunctionArgs, ScalarUDF, ScalarUDFImpl, Signature,
34-
Volatility,
35+
function::ScalarFunctionImplementation, ptr_eq::PtrEq, ColumnarValue, ReturnFieldArgs,
36+
ScalarFunctionArgs, ScalarUDF, ScalarUDFImpl, Signature, Volatility,
3537
};
3638

37-
use crate::errors::to_datafusion_err;
38-
use crate::errors::{py_datafusion_err, PyDataFusionResult};
39-
use crate::expr::PyExpr;
40-
use crate::utils::{parse_volatility, validate_pycapsule};
41-
4239
/// Create a Rust callable function from a python function that expects pyarrow arrays
4340
fn pyarrow_function_to_rust(
4441
func: PyObject,

0 commit comments

Comments
 (0)