fix Ruff errors

Author: kosiew

python/datafusion/context.py

@@ -21,15 +21,14 @@
 
 import warnings
 from typing import TYPE_CHECKING, Any, Protocol
-from collections.abc import Sequence
-
-import pyarrow as pa
 
 try:
     from warnings import deprecated  # Python 3.13+
 except ImportError:
     from typing_extensions import deprecated  # Python 3.12
 
+import pyarrow as pa
+
 from datafusion.catalog import Catalog, CatalogProvider, Table
 from datafusion.dataframe import DataFrame
 from datafusion.expr import SortKey, sort_list_to_raw_sort_list
@@ -44,9 +43,10 @@
 
 if TYPE_CHECKING:
     import pathlib
+    from collections.abc import Sequence
 
     import pandas as pd
-    import polars as pl
+    import polars as pl  # type: ignore[import]
 
     from datafusion.plan import ExecutionPlan, LogicalPlan
 

python/datafusion/expr.py

@@ -23,7 +23,6 @@
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any, ClassVar, Iterable, Optional
-from collections.abc import Sequence
 
 import pyarrow as pa
 
@@ -38,6 +37,9 @@
 from ._internal import functions as functions_internal
 
 if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+    from datafusion.common import DataTypeMap, RexType
     from datafusion.plan import LogicalPlan