Apply import ordering in ruff check (#1001)

Author: timsaucer

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: