Fix: rearrange imports instead of suppressing lint

Author: Awesome06

examples/datafusion-ffi-example/run_demo.py

@@ -8,6 +8,8 @@
 
 import sys
 
+from datafusion import LogicalPlan, SessionConfig, SessionContext, udf
+
 try:
     from datafusion_ffi_example import (
         IsNullUDF,
@@ -19,8 +21,6 @@
 except ImportError:
     sys.exit("build the extension first:\n  uv run maturin develop\nSee README.md.")
 
-from datafusion import LogicalPlan, SessionConfig, SessionContext, udf  # noqa: I001, E402
-
 
 print("1. table provider")
 ctx = SessionContext()

examples/datafusion-ffi-query-planner-example/run_demo.py

@@ -8,6 +8,8 @@
 
 import sys
 
+from datafusion import SessionConfig, SessionContext
+
 try:
     from datafusion_ffi_query_planner_example import (
         MyPlannerConfig,
@@ -16,8 +18,6 @@
 except ImportError:
     sys.exit("build the extension first:\n  uv run maturin develop\nSee README.md.")
 
-from datafusion import SessionConfig, SessionContext  # noqa: I001, E402
-
 
 print("1. logical plan")
 config = SessionConfig().with_extension(MyPlannerConfig(max_rows=5))