apply cargo fmt (#12)

Author: jimexist

src/context.rs

@@ -60,10 +60,7 @@ impl PyExecutionContext {
         Ok(PyDataFrame::new(df))
     }
 
-    fn create_dataframe(
-        &mut self,
-        partitions: Vec<Vec<RecordBatch>>,
-    ) -> PyResult<PyDataFrame> {
+    fn create_dataframe(&mut self, partitions: Vec<Vec<RecordBatch>>) -> PyResult<PyDataFrame> {
         let table = MemTable::try_new(partitions[0][0].schema(), partitions)
             .map_err(DataFusionError::from)?;
 

src/functions.rs

@@ -19,9 +19,7 @@ use pyo3::{prelude::*, wrap_pyfunction};
 
 use datafusion::logical_plan;
 
-use datafusion::physical_plan::{
-    aggregates::AggregateFunction, functions::BuiltinScalarFunction,
-};
+use datafusion::physical_plan::{aggregates::AggregateFunction, functions::BuiltinScalarFunction};
 
 use crate::errors;
 use crate::expression::PyExpr;
@@ -88,11 +86,7 @@ fn concat_ws(sep: String, args: Vec<PyExpr>) -> PyResult<PyExpr> {
 
 /// Creates a new Sort expression
 #[pyfunction]
-fn order_by(
-    expr: PyExpr,
-    asc: Option<bool>,
-    nulls_first: Option<bool>,
-) -> PyResult<PyExpr> {
+fn order_by(expr: PyExpr, asc: Option<bool>, nulls_first: Option<bool>) -> PyResult<PyExpr> {
     Ok(PyExpr {
         expr: datafusion::logical_plan::Expr::Sort {
             expr: Box::new(expr.expr),
@@ -106,10 +100,7 @@ fn order_by(
 #[pyfunction]
 fn alias(expr: PyExpr, name: &str) -> PyResult<PyExpr> {
     Ok(PyExpr {
-        expr: datafusion::logical_plan::Expr::Alias(
-            Box::new(expr.expr),
-            String::from(name),
-        ),
+        expr: datafusion::logical_plan::Expr::Alias(Box::new(expr.expr), String::from(name)),
     })
 }
 
@@ -244,7 +235,11 @@ scalar_function!(sha384, SHA384);
 scalar_function!(sha512, SHA512);
 scalar_function!(signum, Signum);
 scalar_function!(sin, Sin);
-scalar_function!(split_part, SplitPart, "Splits string at occurrences of delimiter and returns the n'th field (counting from one).");
+scalar_function!(
+    split_part,
+    SplitPart,
+    "Splits string at occurrences of delimiter and returns the n'th field (counting from one)."
+);
 scalar_function!(sqrt, Sqrt);
 scalar_function!(
     starts_with,

src/udf.rs

@@ -24,9 +24,7 @@ use datafusion::arrow::datatypes::DataType;
 use datafusion::arrow::pyarrow::PyArrowConvert;
 use datafusion::error::DataFusionError;
 use datafusion::logical_plan;
-use datafusion::physical_plan::functions::{
-    make_scalar_function, ScalarFunctionImplementation,
-};
+use datafusion::physical_plan::functions::{make_scalar_function, ScalarFunctionImplementation};
 use datafusion::physical_plan::udf::ScalarUDF;
 
 use crate::expression::PyExpr;