Apply import ordering in ruff check

benchmarks/db-benchmark/groupby-datafusion.py

@@ -15,21 +15,23 @@
 # specific language governing permissions and limitations
 # under the License.
 
-import os
 import gc
+import os
 import timeit
+
 import datafusion as df
+import pyarrow
 from datafusion import (
-    col,
-    functions as f,
     RuntimeEnvBuilder,
     SessionConfig,
     SessionContext,
+    col,
+)
+from datafusion import (
+    functions as f,
 )
-import pyarrow
 from pyarrow import csv as pacsv
 
-
 print("# groupby-datafusion.py", flush=True)
 
 exec(open("./_helpers/helpers.py").read())

benchmarks/db-benchmark/join-datafusion.py

@@ -15,15 +15,15 @@
 # specific language governing permissions and limitations
 # under the License.
 
-import os
 import gc
+import os
 import timeit
+
 import datafusion as df
-from datafusion import functions as f
 from datafusion import col
+from datafusion import functions as f
 from pyarrow import csv as pacsv
 
-
 print("# join-datafusion.py", flush=True)
 
 exec(open("./_helpers/helpers.py").read())

benchmarks/tpch/tpch.py

@@ -16,9 +16,10 @@
 # under the License.
 
 import argparse
-from datafusion import SessionContext
 import time
 
+from datafusion import SessionContext
+
 
 def bench(data_path, query_path):
     with open("results.csv", "w") as results:

dev/release/generate-changelog.py

@@ -16,11 +16,12 @@
 # limitations under the License.
 
 import argparse
-import sys
-from github import Github
 import os
 import re
 import subprocess
+import sys
+
+from github import Github
 
 
 def print_pulls(repo_name, title, pulls):

examples/export.py

@@ -17,7 +17,6 @@
 
 import datafusion
 
-
 # create a context
 ctx = datafusion.SessionContext()
 

examples/ffi-table-provider/python/tests/_test_table_provider.py

@@ -15,9 +15,9 @@
 # specific language governing permissions and limitations
 # under the License.
 
+import pyarrow as pa
 from datafusion import SessionContext
 from ffi_table_provider import MyTableProvider
-import pyarrow as pa
 
 
 def test_table_loading():

examples/import.py

@@ -16,10 +16,9 @@
 # under the License.
 
 import datafusion
-import pyarrow as pa
 import pandas as pd
 import polars as pl
-
+import pyarrow as pa
 
 # Create a context
 ctx = datafusion.SessionContext()

examples/python-udaf.py

@@ -15,11 +15,10 @@
 # specific language governing permissions and limitations
 # under the License.
 
+import datafusion
 import pyarrow
 import pyarrow.compute
-import datafusion
-from datafusion import udaf, Accumulator
-from datafusion import col
+from datafusion import Accumulator, col, udaf
 
 
 class MyAccumulator(Accumulator):

examples/python-udf-comparisons.py

@@ -15,11 +15,13 @@
 # specific language governing permissions and limitations
 # under the License.
 
-from datafusion import SessionContext, col, lit, udf, functions as F
 import os
+import time
+
 import pyarrow as pa
 import pyarrow.compute as pc
-import time
+from datafusion import SessionContext, col, lit, udf
+from datafusion import functions as F
 
 path = os.path.dirname(os.path.abspath(__file__))
 filepath = os.path.join(path, "./tpch/data/lineitem.parquet")

examples/python-udf.py

@@ -16,7 +16,8 @@
 # under the License.
 
 import pyarrow
-from datafusion import udf, SessionContext, functions as f
+from datafusion import SessionContext, udf
+from datafusion import functions as f
 
 
 def is_null(array: pyarrow.Array) -> pyarrow.Array:

examples/python-udwf.py

@@ -15,11 +15,12 @@
 # specific language governing permissions and limitations
 # under the License.
 
-import pyarrow as pa
 import datafusion
-from datafusion import udwf, functions as f, col, lit
-from datafusion.udf import WindowEvaluator
+import pyarrow as pa
+from datafusion import col, lit, udwf
+from datafusion import functions as f
 from datafusion.expr import WindowFrame
+from datafusion.udf import WindowEvaluator
 
 # This example creates five different examples of user defined window functions in order
 # to demonstrate the variety of ways a user may need to implement.

examples/query-pyarrow-data.py

@@ -16,9 +16,8 @@
 # under the License.
 
 import datafusion
-from datafusion import col
 import pyarrow
-
+from datafusion import col
 
 # create a context
 ctx = datafusion.SessionContext()

examples/sql-parquet-s3.py

@@ -16,6 +16,7 @@
 # under the License.
 
 import os
+
 import datafusion
 from datafusion.object_store import AmazonS3
 

examples/sql-to-pandas.py

@@ -17,7 +17,6 @@
 
 from datafusion import SessionContext
 
-
 # Create a DataFusion context
 ctx = SessionContext()
 

examples/sql-using-python-udaf.py

@@ -15,8 +15,8 @@
 # specific language governing permissions and limitations
 # under the License.
 
-from datafusion import udaf, SessionContext, Accumulator
 import pyarrow as pa
+from datafusion import Accumulator, SessionContext, udaf
 
 
 # Define a user-defined aggregation function (UDAF)

examples/sql-using-python-udf.py

@@ -15,8 +15,8 @@
 # specific language governing permissions and limitations
 # under the License.
 
-from datafusion import udf, SessionContext
 import pyarrow as pa
+from datafusion import SessionContext, udf
 
 
 # Define a user-defined function (UDF)

examples/tpch/_tests.py

@@ -15,10 +15,12 @@
 # specific language governing permissions and limitations
 # under the License.
 
-import pytest
 from importlib import import_module
+
 import pyarrow as pa
-from datafusion import DataFrame, col, lit, functions as F
+import pytest
+from datafusion import DataFrame, col, lit
+from datafusion import functions as F
 from util import get_answer_file
 
 

examples/tpch/convert_data_to_parquet.py

@@ -23,8 +23,9 @@
 """
 
 import os
-import pyarrow
+
 import datafusion
+import pyarrow
 
 ctx = datafusion.SessionContext()
 

examples/tpch/q01_pricing_summary_report.py

@@ -30,7 +30,8 @@
 """
 
 import pyarrow as pa
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 ctx = SessionContext()

examples/tpch/q02_minimum_cost_supplier.py

@@ -30,7 +30,8 @@
 """
 
 import datafusion
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 # This is the part we're looking for. Values selected here differ from the spec in order to run

examples/tpch/q03_shipping_priority.py

@@ -27,7 +27,8 @@
 as part of their TPC Benchmark H Specification revision 2.18.0.
 """
 
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 SEGMENT_OF_INTEREST = "BUILDING"

examples/tpch/q04_order_priority_checking.py

@@ -27,8 +27,10 @@
 """
 
 from datetime import datetime
+
 import pyarrow as pa
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 # Ideally we could put 3 months into the interval. See note below.

examples/tpch/q05_local_supplier_volume.py

@@ -30,11 +30,12 @@
 """
 
 from datetime import datetime
+
 import pyarrow as pa
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
-
 DATE_OF_INTEREST = "1994-01-01"
 INTERVAL_DAYS = 365
 REGION_OF_INTEREST = "ASIA"

examples/tpch/q06_forecasting_revenue_change.py

@@ -30,8 +30,10 @@
 """
 
 from datetime import datetime
+
 import pyarrow as pa
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 # Variables from the example query

examples/tpch/q07_volume_shipping.py

@@ -29,8 +29,10 @@
 """
 
 from datetime import datetime
+
 import pyarrow as pa
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 # Variables of interest to query over

examples/tpch/q08_market_share.py

@@ -28,8 +28,10 @@
 """
 
 from datetime import datetime
+
 import pyarrow as pa
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 supplier_nation = lit("BRAZIL")

examples/tpch/q09_product_type_profit_measure.py

@@ -30,7 +30,8 @@
 """
 
 import pyarrow as pa
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 part_color = lit("green")

examples/tpch/q10_returned_item_reporting.py

@@ -30,8 +30,10 @@
 """
 
 from datetime import datetime
+
 import pyarrow as pa
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 DATE_START_OF_QUARTER = "1993-10-01"

examples/tpch/q11_important_stock_identification.py

@@ -27,7 +27,8 @@
 as part of their TPC Benchmark H Specification revision 2.18.0.
 """
 
-from datafusion import SessionContext, WindowFrame, col, lit, functions as F
+from datafusion import SessionContext, WindowFrame, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 NATION = "GERMANY"

examples/tpch/q12_ship_mode_order_priority.py

@@ -30,8 +30,10 @@
 """
 
 from datetime import datetime
+
 import pyarrow as pa
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 SHIP_MODE_1 = "MAIL"

examples/tpch/q13_customer_distribution.py

@@ -28,7 +28,8 @@
 as part of their TPC Benchmark H Specification revision 2.18.0.
 """
 
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 WORD_1 = "special"

examples/tpch/q14_promotion_effect.py

@@ -27,8 +27,10 @@
 """
 
 from datetime import datetime
+
 import pyarrow as pa
-from datafusion import SessionContext, col, lit, functions as F
+from datafusion import SessionContext, col, lit
+from datafusion import functions as F
 from util import get_data_path
 
 DATE = "1995-09-01"