Enabled ruff rule UP006

Author: Spaarsh

pyproject.toml

@@ -90,7 +90,7 @@ ignore = [
     # "C400",
     "TRY003",
     # "B904",
-    "UP006",
+    # "UP006",
     "RUF012",
     "FBT003",
     "C416",

python/datafusion/dataframe.py

@@ -26,10 +26,8 @@
     TYPE_CHECKING,
     Any,
     Iterable,
-    List,
     Literal,
     Optional,
-    Type,
     Union,
     overload,
 )
@@ -75,7 +73,7 @@ class Compression(Enum):
     LZ4_RAW = "lz4_raw"
 
     @classmethod
-    def from_str(cls: Type[Compression], value: str) -> Compression:
+    def from_str(cls: type[Compression], value: str) -> Compression:
         """Convert a string to a Compression enum value.
 
         Args:
@@ -132,7 +130,7 @@ def into_view(self) -> pa.Table:
         """Convert DataFrame as a ViewTable which can be used in register_table."""
         return self.df.into_view()
 
-    def __getitem__(self, key: str | List[str]) -> DataFrame:
+    def __getitem__(self, key: str | list[str]) -> DataFrame:
         """Return a new :py:class`DataFrame` with the specified column or columns.
 
         Args:

python/datafusion/expr.py

@@ -22,7 +22,7 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Optional, Type
+from typing import TYPE_CHECKING, Any, Optional
 
 import pyarrow as pa
 
@@ -446,9 +446,7 @@ def fill_null(self, value: Any | Expr | None = None) -> Expr:
         bool: pa.bool_(),
     }
 
-    def cast(
-        self, to: pa.DataType[Any] | Type[float] | Type[int] | Type[str] | Type[bool]
-    ) -> Expr:
+    def cast(self, to: pa.DataType[Any] | type[float | int | str | bool]) -> Expr:
         """Cast to a new data type."""
         if not isinstance(to, pa.DataType):
             try: