Skip to content

Commit 29bca8f

Browse files
committed
refactor: convert to ScalarUDF
1 parent 50ff1a4 commit 29bca8f

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

query/tests/integration.rs

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ use datafusion_udf_wasm_query::{
2020
use tokio::runtime::Handle;
2121

2222
mod integration_tests;
23-
use datafusion_expr::async_udf::AsyncScalarUDF;
2423

2524
use crate::integration_tests::python::test_utils::python_component;
2625

@@ -33,8 +32,7 @@ impl UdfQueryInvocator {
3332
parsed_query: ParsedQuery,
3433
) -> DataFusionResult<DataFrame> {
3534
for udf in parsed_query.udfs {
36-
let scalar_udf = AsyncScalarUDF::new(udf);
37-
ctx.register_udf(scalar_udf.into());
35+
ctx.register_udf(udf.as_async_udf().into());
3836
}
3937

4038
ctx.sql(&parsed_query.sql).await

0 commit comments

Comments
 (0)