chore: clean up code formatting

Author: Prathamesh9284

pyproject.toml

@@ -173,20 +173,20 @@ ignore-words-list = ["ans", "IST"]
 
 [dependency-groups]
 dev = [
-    "maturin>=1.8.1",
-    "numpy>1.25.0;python_version<'3.14'",
-    "numpy>=2.3.2;python_version>='3.14'",
-    "pyarrow>=19.0.0",
-    "pre-commit>=4.3.0",
-    "pyyaml>=6.0.3",
-    "pytest>=7.4.4",
-    "pytest-asyncio>=0.23.3",
-    "ruff>=0.9.1",
-    "toml>=0.10.2",
-    "pygithub==2.5.0",
-    "codespell==2.4.1",
-    "protobuf>=6.33.5",
-    "substrait>=0.27.0",
+  "maturin>=1.8.1",
+  "numpy>1.25.0;python_version<'3.14'",
+  "numpy>=2.3.2;python_version>='3.14'",
+  "pyarrow>=19.0.0",
+  "pre-commit>=4.3.0",
+  "pyyaml>=6.0.3",
+  "pytest>=7.4.4",
+  "pytest-asyncio>=0.23.3",
+  "ruff>=0.9.1",
+  "toml>=0.10.2",
+  "pygithub==2.5.0",
+  "codespell==2.4.1",
+  "protobuf>=6.33.5",
+  "substrait>=0.27.0",
 ]
 docs = [
   "sphinx>=7.1.2",
@@ -198,4 +198,4 @@ docs = [
   "pickleshare>=0.7.5",
   "sphinx-autoapi>=3.4.0",
   "setuptools>=75.3.0",
-]
+]

python/datafusion/substrait.py

@@ -210,4 +210,4 @@ def from_substrait_plan(ctx: SessionContext, plan: Plan) -> LogicalPlan:
 
 @deprecated("Use `Consumer` instead.")
 class consumer(Consumer):  # noqa: N801
-    """Use `Consumer` instead."""
+    """Use `Consumer` instead."""

python/tests/test_substrait.py

@@ -96,4 +96,4 @@ def test_plan_json_stability_and_validator_compatibility(ctx):
     json_format.Parse(json_str, proto_plan)
     actual_binary_plan = proto_plan.SerializeToString()
 
-    assert expected_binary_plan == actual_binary_plan
+    assert expected_binary_plan == actual_binary_plan

src/errors.rs

@@ -22,10 +22,8 @@ use std::fmt::Debug;
 use datafusion::arrow::error::ArrowError;
 use datafusion::error::DataFusionError as InnerDataFusionError;
 use prost::EncodeError;
-use prost::EncodeError;
 use pyo3::PyErr;
 use pyo3::exceptions::{PyException, PyValueError};
-use pyo3::exceptions::{PyException, PyValueError};
 
 pub type PyDataFusionResult<T> = std::result::Result<T, PyDataFusionError>;
 

src/substrait.rs

@@ -170,4 +170,4 @@ pub fn init_module(m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_class::<PySubstraitProducer>()?;
     m.add_class::<PySubstraitSerializer>()?;
     Ok(())
-}
+}