Skip to content

Commit f7f518c

Browse files
clean up functions_window invocation
1 parent 6be0016 commit f7f518c

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/functions.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
// under the License.
1717

1818
use datafusion::functions_aggregate::all_default_aggregate_functions;
19-
use datafusion::logical_expr::window_function;
2019
use datafusion::logical_expr::ExprFunctionExt;
2120
use datafusion::logical_expr::WindowFrame;
2221
use pyo3::{prelude::*, wrap_pyfunction};
@@ -33,6 +32,7 @@ use datafusion::common::{Column, ScalarValue, TableReference};
3332
use datafusion::execution::FunctionRegistry;
3433
use datafusion::functions;
3534
use datafusion::functions_aggregate;
35+
use datafusion::functions_window;
3636
use datafusion::logical_expr::expr::Alias;
3737
use datafusion::logical_expr::sqlparser::ast::NullTreatment as DFNullTreatment;
3838
use datafusion::logical_expr::{
@@ -757,7 +757,7 @@ pub fn lead(
757757
partition_by: Option<Vec<PyExpr>>,
758758
order_by: Option<Vec<PySortExpr>>,
759759
) -> PyResult<PyExpr> {
760-
let window_fn = datafusion::functions_window::expr_fn::lead(arg.expr, Some(shift_offset), default_value);
760+
let window_fn = functions_window::expr_fn::lead(arg.expr, Some(shift_offset), default_value);
761761

762762
add_builder_fns_to_window(window_fn, partition_by, None, order_by, None)
763763
}
@@ -771,7 +771,7 @@ pub fn lag(
771771
partition_by: Option<Vec<PyExpr>>,
772772
order_by: Option<Vec<PySortExpr>>,
773773
) -> PyResult<PyExpr> {
774-
let window_fn = datafusion::functions_window::expr_fn::lag(arg.expr, Some(shift_offset), default_value);
774+
let window_fn = functions_window::expr_fn::lag(arg.expr, Some(shift_offset), default_value);
775775

776776
add_builder_fns_to_window(window_fn, partition_by, None, order_by, None)
777777
}
@@ -782,7 +782,7 @@ pub fn row_number(
782782
partition_by: Option<Vec<PyExpr>>,
783783
order_by: Option<Vec<PySortExpr>>,
784784
) -> PyResult<PyExpr> {
785-
let window_fn = datafusion::functions_window::expr_fn::row_number();
785+
let window_fn = functions_window::expr_fn::row_number();
786786

787787
add_builder_fns_to_window(window_fn, partition_by, None, order_by, None)
788788
}
@@ -793,7 +793,7 @@ pub fn rank(
793793
partition_by: Option<Vec<PyExpr>>,
794794
order_by: Option<Vec<PySortExpr>>,
795795
) -> PyResult<PyExpr> {
796-
let window_fn = datafusion::functions_window::expr_fn::rank();
796+
let window_fn = functions_window::expr_fn::rank();
797797

798798
add_builder_fns_to_window(window_fn, partition_by, None, order_by, None)
799799
}
@@ -804,7 +804,7 @@ pub fn dense_rank(
804804
partition_by: Option<Vec<PyExpr>>,
805805
order_by: Option<Vec<PySortExpr>>,
806806
) -> PyResult<PyExpr> {
807-
let window_fn = datafusion::functions_window::expr_fn::dense_rank();
807+
let window_fn = functions_window::expr_fn::dense_rank();
808808

809809
add_builder_fns_to_window(window_fn, partition_by, None, order_by, None)
810810
}
@@ -815,7 +815,7 @@ pub fn percent_rank(
815815
partition_by: Option<Vec<PyExpr>>,
816816
order_by: Option<Vec<PySortExpr>>,
817817
) -> PyResult<PyExpr> {
818-
let window_fn = datafusion::functions_window::expr_fn::percent_rank();
818+
let window_fn = functions_window::expr_fn::percent_rank();
819819

820820
add_builder_fns_to_window(window_fn, partition_by, None, order_by, None)
821821
}
@@ -826,7 +826,7 @@ pub fn cume_dist(
826826
partition_by: Option<Vec<PyExpr>>,
827827
order_by: Option<Vec<PySortExpr>>,
828828
) -> PyResult<PyExpr> {
829-
let window_fn = datafusion::functions_window::expr_fn::cume_dist();
829+
let window_fn = functions_window::expr_fn::cume_dist();
830830

831831
add_builder_fns_to_window(window_fn, partition_by, None, order_by, None)
832832
}
@@ -838,7 +838,7 @@ pub fn ntile(
838838
partition_by: Option<Vec<PyExpr>>,
839839
order_by: Option<Vec<PySortExpr>>,
840840
) -> PyResult<PyExpr> {
841-
let window_fn = datafusion::functions_window::expr_fn::ntile(arg.into());
841+
let window_fn = functions_window::expr_fn::ntile(arg.into());
842842

843843
add_builder_fns_to_window(window_fn, partition_by, None, order_by, None)
844844
}

0 commit comments

Comments
 (0)