PyAthena 3.29.3__tar.gz → 3.30.0__tar.gz

{pyathena-3.29.3 → pyathena-3.30.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyAthena
-Version: 3.29.3
+Version: 3.30.0
 Summary: Python DB API 2.0 (PEP 249) client for Amazon Athena
 Project-URL: homepage, https://github.com/pyathena-dev/PyAthena/
 Project-URL: repository, https://github.com/pyathena-dev/PyAthena/
@@ -92,12 +92,13 @@ $ pip install PyAthena
 
 Extra packages:
 
-| Package    | Install command                      | Version  |
-|------------|--------------------------------------|----------|
-| SQLAlchemy | `pip install PyAthena[SQLAlchemy]`   | >=1.0.0  |
-| Pandas     | `pip install PyAthena[Pandas]`       | >=1.3.0  |
-| Arrow      | `pip install PyAthena[Arrow]`        | >=10.0.0 |
-| Polars     | `pip install PyAthena[Polars]`       | >=1.0.0  |
+| Package       | Install command                         | Version  |
+|---------------|-----------------------------------------|----------|
+| SQLAlchemy    | `pip install PyAthena[SQLAlchemy]`      | >=1.0.0  |
+| AioSQLAlchemy | `pip install PyAthena[AioSQLAlchemy]`   | >=2.0.0  |
+| Pandas        | `pip install PyAthena[Pandas]`          | >=1.3.0  |
+| Arrow         | `pip install PyAthena[Arrow]`           | >=10.0.0 |
+| Polars        | `pip install PyAthena[Polars]`          | >=1.0.0  |
 
 ## Usage
 

{pyathena-3.29.3 → pyathena-3.30.0}/README.md

@@ -33,12 +33,13 @@ $ pip install PyAthena
 
 Extra packages:
 
-| Package    | Install command                      | Version  |
-|------------|--------------------------------------|----------|
-| SQLAlchemy | `pip install PyAthena[SQLAlchemy]`   | >=1.0.0  |
-| Pandas     | `pip install PyAthena[Pandas]`       | >=1.3.0  |
-| Arrow      | `pip install PyAthena[Arrow]`        | >=10.0.0 |
-| Polars     | `pip install PyAthena[Polars]`       | >=1.0.0  |
+| Package       | Install command                         | Version  |
+|---------------|-----------------------------------------|----------|
+| SQLAlchemy    | `pip install PyAthena[SQLAlchemy]`      | >=1.0.0  |
+| AioSQLAlchemy | `pip install PyAthena[AioSQLAlchemy]`   | >=2.0.0  |
+| Pandas        | `pip install PyAthena[Pandas]`          | >=1.3.0  |
+| Arrow         | `pip install PyAthena[Arrow]`           | >=10.0.0 |
+| Polars        | `pip install PyAthena[Polars]`          | >=1.0.0  |
 
 ## Usage
 

{pyathena-3.29.3 → pyathena-3.30.0}/pyathena/__init__.py

@@ -1,10 +1,9 @@
-# -*- coding: utf-8 -*-
 from __future__ import annotations
 
 import datetime
-from typing import TYPE_CHECKING, Any, FrozenSet, Type, overload
+from typing import TYPE_CHECKING, Any, overload
 
-from pyathena.error import *  # noqa
+from pyathena.error import *  # noqa: F403
 
 if TYPE_CHECKING:
     from pyathena.aio.connection import AioConnection
@@ -28,7 +27,7 @@ threadsafety: int = 2
 paramstyle: str = "pyformat"
 
 
-class DBAPITypeObject(FrozenSet[str]):
+class DBAPITypeObject(frozenset[str]):
     """Type Objects and Constructors
 
     https://www.python.org/dev/peps/pep-0249/#type-objects-and-constructors
@@ -60,22 +59,22 @@ TIME: DBAPITypeObject = DBAPITypeObject(("time", "time with time zone"))
 DATETIME: DBAPITypeObject = DBAPITypeObject(("timestamp", "timestamp with time zone"))
 JSON: DBAPITypeObject = DBAPITypeObject(("json",))
 
-Date: Type[datetime.date] = datetime.date
-Time: Type[datetime.time] = datetime.time
-Timestamp: Type[datetime.datetime] = datetime.datetime
+Date: type[datetime.date] = datetime.date
+Time: type[datetime.time] = datetime.time
+Timestamp: type[datetime.datetime] = datetime.datetime
 
 
 @overload
-def connect(*args, cursor_class: None = ..., **kwargs) -> "Connection[Cursor]": ...
+def connect(*args, cursor_class: None = ..., **kwargs) -> Connection[Cursor]: ...
 
 
 @overload
 def connect(
-    *args, cursor_class: Type[ConnectionCursor], **kwargs
-) -> "Connection[ConnectionCursor]": ...
+    *args, cursor_class: type[ConnectionCursor], **kwargs
+) -> Connection[ConnectionCursor]: ...
 
 
-def connect(*args, **kwargs) -> "Connection[Any]":
+def connect(*args, **kwargs) -> Connection[Any]:
     """Create a new database connection to Amazon Athena.
 
     This function provides the main entry point for establishing connections
@@ -131,7 +130,7 @@ def connect(*args, **kwargs) -> "Connection[Any]":
     return Connection(*args, **kwargs)
 
 
-async def aio_connect(*args, **kwargs) -> "AioConnection":
+async def aio_connect(*args, **kwargs) -> AioConnection:
     """Create a new async database connection to Amazon Athena.
 
     This is the async counterpart of :func:`connect`. It returns an

{pyathena-3.29.3 → pyathena-3.30.0}/pyathena/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '3.29.3'
-__version_tuple__ = version_tuple = (3, 29, 3)
+__version__ = version = '3.30.0'
+__version_tuple__ = version_tuple = (3, 30, 0)
 
 __commit_id__ = commit_id = None

{pyathena-3.29.3 → pyathena-3.30.0}/pyathena/aio/arrow/cursor.py

@@ -1,9 +1,8 @@
-# -*- coding: utf-8 -*-
 from __future__ import annotations
 
 import asyncio
 import logging
-from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union, cast
+from typing import TYPE_CHECKING, Any, cast
 
 from pyathena.aio.common import WithAsyncFetch
 from pyathena.arrow.converter import (
@@ -19,7 +18,7 @@ if TYPE_CHECKING:
     import polars as pl
     from pyarrow import Table
 
-_logger = logging.getLogger(__name__)  # type: ignore
+_logger = logging.getLogger(__name__)
 
 
 class AioArrowCursor(WithAsyncFetch):
@@ -37,19 +36,19 @@ class AioArrowCursor(WithAsyncFetch):
 
     def __init__(
         self,
-        s3_staging_dir: Optional[str] = None,
-        schema_name: Optional[str] = None,
-        catalog_name: Optional[str] = None,
-        work_group: Optional[str] = None,
+        s3_staging_dir: str | None = None,
+        schema_name: str | None = None,
+        catalog_name: str | None = None,
+        work_group: str | None = None,
         poll_interval: float = 1,
-        encryption_option: Optional[str] = None,
-        kms_key: Optional[str] = None,
+        encryption_option: str | None = None,
+        kms_key: str | None = None,
         kill_on_interrupt: bool = True,
         unload: bool = False,
         result_reuse_enable: bool = False,
         result_reuse_minutes: int = CursorIterator.DEFAULT_RESULT_REUSE_MINUTES,
-        connect_timeout: Optional[float] = None,
-        request_timeout: Optional[float] = None,
+        connect_timeout: float | None = None,
+        request_timeout: float | None = None,
         **kwargs,
     ) -> None:
         super().__init__(
@@ -68,12 +67,12 @@ class AioArrowCursor(WithAsyncFetch):
         self._unload = unload
         self._connect_timeout = connect_timeout
         self._request_timeout = request_timeout
-        self._result_set: Optional[AthenaArrowResultSet] = None
+        self._result_set: AthenaArrowResultSet | None = None
 
     @staticmethod
     def get_default_converter(
         unload: bool = False,
-    ) -> Union[DefaultArrowTypeConverter, DefaultArrowUnloadTypeConverter, Any]:
+    ) -> DefaultArrowTypeConverter | DefaultArrowUnloadTypeConverter | Any:
         if unload:
             return DefaultArrowUnloadTypeConverter()
         return DefaultArrowTypeConverter()
@@ -81,16 +80,16 @@ class AioArrowCursor(WithAsyncFetch):
     async def execute(  # type: ignore[override]
         self,
         operation: str,
-        parameters: Optional[Union[Dict[str, Any], List[str]]] = None,
-        work_group: Optional[str] = None,
-        s3_staging_dir: Optional[str] = None,
-        cache_size: Optional[int] = 0,
-        cache_expiration_time: Optional[int] = 0,
-        result_reuse_enable: Optional[bool] = None,
-        result_reuse_minutes: Optional[int] = None,
-        paramstyle: Optional[str] = None,
+        parameters: dict[str, Any] | list[str] | None = None,
+        work_group: str | None = None,
+        s3_staging_dir: str | None = None,
+        cache_size: int | None = 0,
+        cache_expiration_time: int | None = 0,
+        result_reuse_enable: bool | None = None,
+        result_reuse_minutes: int | None = None,
+        paramstyle: str | None = None,
         **kwargs,
-    ) -> "AioArrowCursor":
+    ) -> AioArrowCursor:
         """Execute a SQL query asynchronously and return results as Arrow Tables.
 
         Args:
@@ -143,7 +142,7 @@ class AioArrowCursor(WithAsyncFetch):
 
     async def fetchone(  # type: ignore[override]
         self,
-    ) -> Optional[Union[Tuple[Optional[Any], ...], Dict[Any, Optional[Any]]]]:
+    ) -> tuple[Any | None, ...] | dict[Any, Any | None] | None:
         """Fetch the next row of the result set.
 
         Wraps the synchronous fetch in ``asyncio.to_thread`` to avoid
@@ -161,8 +160,8 @@ class AioArrowCursor(WithAsyncFetch):
         return await asyncio.to_thread(result_set.fetchone)
 
     async def fetchmany(  # type: ignore[override]
-        self, size: Optional[int] = None
-    ) -> List[Union[Tuple[Optional[Any], ...], Dict[Any, Optional[Any]]]]:
+        self, size: int | None = None
+    ) -> list[tuple[Any | None, ...] | dict[Any, Any | None]]:
         """Fetch multiple rows from the result set.
 
         Wraps the synchronous fetch in ``asyncio.to_thread`` to avoid
@@ -184,7 +183,7 @@ class AioArrowCursor(WithAsyncFetch):
 
     async def fetchall(  # type: ignore[override]
         self,
-    ) -> List[Union[Tuple[Optional[Any], ...], Dict[Any, Optional[Any]]]]:
+    ) -> list[tuple[Any | None, ...] | dict[Any, Any | None]]:
         """Fetch all remaining rows from the result set.
 
         Wraps the synchronous fetch in ``asyncio.to_thread`` to avoid
@@ -207,7 +206,7 @@ class AioArrowCursor(WithAsyncFetch):
             raise StopAsyncIteration
         return row
 
-    def as_arrow(self) -> "Table":
+    def as_arrow(self) -> Table:
         """Return query results as an Apache Arrow Table.
 
         Returns:
@@ -218,7 +217,7 @@ class AioArrowCursor(WithAsyncFetch):
         result_set = cast(AthenaArrowResultSet, self.result_set)
         return result_set.as_arrow()
 
-    def as_polars(self) -> "pl.DataFrame":
+    def as_polars(self) -> pl.DataFrame:
         """Return query results as a Polars DataFrame.
 
         Returns:

{pyathena-3.29.3 → pyathena-3.30.0}/pyathena/aio/common.py

@@ -1,11 +1,10 @@
-# -*- coding: utf-8 -*-
 from __future__ import annotations
 
 import asyncio
 import logging
 import sys
 from datetime import datetime, timedelta, timezone
-from typing import Any, Dict, List, Optional, Tuple, Union, cast
+from typing import Any, cast
 
 from pyathena.aio.util import async_retry_api_call
 from pyathena.common import BaseCursor, CursorIterator
@@ -13,7 +12,7 @@ from pyathena.error import DatabaseError, OperationalError, ProgrammingError
 from pyathena.model import AthenaDatabase, AthenaQueryExecution, AthenaTableMetadata
 from pyathena.result_set import AthenaResultSet, WithResultSet
 
-_logger = logging.getLogger(__name__)  # type: ignore
+_logger = logging.getLogger(__name__)
 
 
 class AioBaseCursor(BaseCursor):
@@ -27,14 +26,14 @@ class AioBaseCursor(BaseCursor):
     async def _execute(  # type: ignore[override]
         self,
         operation: str,
-        parameters: Optional[Union[Dict[str, Any], List[str]]] = None,
-        work_group: Optional[str] = None,
-        s3_staging_dir: Optional[str] = None,
-        cache_size: Optional[int] = 0,
-        cache_expiration_time: Optional[int] = 0,
-        result_reuse_enable: Optional[bool] = None,
-        result_reuse_minutes: Optional[int] = None,
-        paramstyle: Optional[str] = None,
+        parameters: dict[str, Any] | list[str] | None = None,
+        work_group: str | None = None,
+        s3_staging_dir: str | None = None,
+        cache_size: int | None = 0,
+        cache_expiration_time: int | None = 0,
+        result_reuse_enable: bool | None = None,
+        result_reuse_minutes: int | None = None,
+        paramstyle: str | None = None,
     ) -> str:
         query, execution_parameters = self._prepare_query(operation, parameters, paramstyle)
 
@@ -118,8 +117,8 @@ class AioBaseCursor(BaseCursor):
             raise OperationalError(*e.args) from e
 
     async def _batch_get_query_execution(  # type: ignore[override]
-        self, query_ids: List[str]
-    ) -> List[AthenaQueryExecution]:
+        self, query_ids: list[str]
+    ) -> list[AthenaQueryExecution]:
         try:
             response = await async_retry_api_call(
                 self.connection._client.batch_get_query_execution,
@@ -138,10 +137,10 @@ class AioBaseCursor(BaseCursor):
 
     async def _list_query_executions(  # type: ignore[override]
         self,
-        work_group: Optional[str] = None,
-        next_token: Optional[str] = None,
-        max_results: Optional[int] = None,
-    ) -> Tuple[Optional[str], List[AthenaQueryExecution]]:
+        work_group: str | None = None,
+        next_token: str | None = None,
+        max_results: int | None = None,
+    ) -> tuple[str | None, list[AthenaQueryExecution]]:
         request = self._build_list_query_executions_request(
             work_group=work_group, next_token=next_token, max_results=max_results
         )
@@ -165,10 +164,10 @@ class AioBaseCursor(BaseCursor):
     async def _find_previous_query_id(  # type: ignore[override]
         self,
         query: str,
-        work_group: Optional[str],
+        work_group: str | None,
         cache_size: int = 0,
         cache_expiration_time: int = 0,
-    ) -> Optional[str]:
+    ) -> str | None:
         query_id = None
         if cache_size == 0 and cache_expiration_time > 0:
             cache_size = sys.maxsize
@@ -191,7 +190,7 @@ class AioBaseCursor(BaseCursor):
                         if e.state == AthenaQueryExecution.STATE_SUCCEEDED
                         and e.statement_type == AthenaQueryExecution.STATEMENT_TYPE_DML
                     ),
-                    key=lambda e: e.completion_date_time,  # type: ignore
+                    key=lambda e: e.completion_date_time,  # type: ignore[arg-type, return-value]
                     reverse=True,
                 ):
                     if (
@@ -213,10 +212,10 @@ class AioBaseCursor(BaseCursor):
 
     async def _list_databases(  # type: ignore[override]
         self,
-        catalog_name: Optional[str],
-        next_token: Optional[str] = None,
-        max_results: Optional[int] = None,
-    ) -> Tuple[Optional[str], List[AthenaDatabase]]:
+        catalog_name: str | None,
+        next_token: str | None = None,
+        max_results: int | None = None,
+    ) -> tuple[str | None, list[AthenaDatabase]]:
         request = self._build_list_databases_request(
             catalog_name=catalog_name,
             next_token=next_token,
@@ -239,10 +238,10 @@ class AioBaseCursor(BaseCursor):
 
     async def list_databases(  # type: ignore[override]
         self,
-        catalog_name: Optional[str],
-        max_results: Optional[int] = None,
-    ) -> List[AthenaDatabase]:
-        databases: List[AthenaDatabase] = []
+        catalog_name: str | None,
+        max_results: int | None = None,
+    ) -> list[AthenaDatabase]:
+        databases: list[AthenaDatabase] = []
         next_token = None
         while True:
             next_token, response = await self._list_databases(
@@ -258,8 +257,8 @@ class AioBaseCursor(BaseCursor):
     async def _get_table_metadata(  # type: ignore[override]
         self,
         table_name: str,
-        catalog_name: Optional[str] = None,
-        schema_name: Optional[str] = None,
+        catalog_name: str | None = None,
+        schema_name: str | None = None,
         logging_: bool = True,
     ) -> AthenaTableMetadata:
         request = self._build_get_table_metadata_request(
@@ -284,8 +283,8 @@ class AioBaseCursor(BaseCursor):
     async def get_table_metadata(  # type: ignore[override]
         self,
         table_name: str,
-        catalog_name: Optional[str] = None,
-        schema_name: Optional[str] = None,
+        catalog_name: str | None = None,
+        schema_name: str | None = None,
         logging_: bool = True,
     ) -> AthenaTableMetadata:
         return await self._get_table_metadata(
@@ -297,12 +296,12 @@ class AioBaseCursor(BaseCursor):
 
     async def _list_table_metadata(  # type: ignore[override]
         self,
-        catalog_name: Optional[str] = None,
-        schema_name: Optional[str] = None,
-        expression: Optional[str] = None,
-        next_token: Optional[str] = None,
-        max_results: Optional[int] = None,
-    ) -> Tuple[Optional[str], List[AthenaTableMetadata]]:
+        catalog_name: str | None = None,
+        schema_name: str | None = None,
+        expression: str | None = None,
+        next_token: str | None = None,
+        max_results: int | None = None,
+    ) -> tuple[str | None, list[AthenaTableMetadata]]:
         request = self._build_list_table_metadata_request(
             catalog_name=catalog_name,
             schema_name=schema_name,
@@ -328,12 +327,12 @@ class AioBaseCursor(BaseCursor):
 
     async def list_table_metadata(  # type: ignore[override]
         self,
-        catalog_name: Optional[str] = None,
-        schema_name: Optional[str] = None,
-        expression: Optional[str] = None,
-        max_results: Optional[int] = None,
-    ) -> List[AthenaTableMetadata]:
-        metadata: List[AthenaTableMetadata] = []
+        catalog_name: str | None = None,
+        schema_name: str | None = None,
+        expression: str | None = None,
+        max_results: int | None = None,
+    ) -> list[AthenaTableMetadata]:
+        metadata: list[AthenaTableMetadata] = []
         next_token = None
         while True:
             next_token, response = await self._list_table_metadata(
@@ -363,8 +362,8 @@ class WithAsyncFetch(AioBaseCursor, CursorIterator, WithResultSet):
 
     def __init__(self, **kwargs) -> None:
         super().__init__(**kwargs)
-        self._query_id: Optional[str] = None
-        self._result_set: Optional[AthenaResultSet] = None
+        self._query_id: str | None = None
+        self._result_set: AthenaResultSet | None = None
 
     @property
     def arraysize(self) -> int:
@@ -376,8 +375,8 @@ class WithAsyncFetch(AioBaseCursor, CursorIterator, WithResultSet):
             raise ProgrammingError("arraysize must be a positive integer value.")
         self._arraysize = value
 
-    @property  # type: ignore
-    def result_set(self) -> Optional[AthenaResultSet]:
+    @property  # type: ignore[override]
+    def result_set(self) -> AthenaResultSet | None:
         return self._result_set
 
     @result_set.setter
@@ -385,7 +384,7 @@ class WithAsyncFetch(AioBaseCursor, CursorIterator, WithResultSet):
         self._result_set = val
 
     @property
-    def query_id(self) -> Optional[str]:
+    def query_id(self) -> str | None:
         return self._query_id
 
     @query_id.setter
@@ -393,7 +392,7 @@ class WithAsyncFetch(AioBaseCursor, CursorIterator, WithResultSet):
         self._query_id = val
 
     @property
-    def rownumber(self) -> Optional[int]:
+    def rownumber(self) -> int | None:
         return self.result_set.rownumber if self.result_set else None
 
     @property
@@ -408,7 +407,7 @@ class WithAsyncFetch(AioBaseCursor, CursorIterator, WithResultSet):
     async def executemany(  # type: ignore[override]
         self,
         operation: str,
-        seq_of_parameters: List[Optional[Union[Dict[str, Any], List[str]]]],
+        seq_of_parameters: list[dict[str, Any] | list[str] | None],
         **kwargs,
     ) -> None:
         """Execute a SQL query multiple times with different parameters.
@@ -435,7 +434,7 @@ class WithAsyncFetch(AioBaseCursor, CursorIterator, WithResultSet):
 
     def fetchone(
         self,
-    ) -> Optional[Union[Tuple[Optional[Any], ...], Dict[Any, Optional[Any]]]]:
+    ) -> tuple[Any | None, ...] | dict[Any, Any | None] | None:
         """Fetch the next row of the result set.
 
         Returns:
@@ -450,8 +449,8 @@ class WithAsyncFetch(AioBaseCursor, CursorIterator, WithResultSet):
         return result_set.fetchone()
 
     def fetchmany(
-        self, size: Optional[int] = None
-    ) -> List[Union[Tuple[Optional[Any], ...], Dict[Any, Optional[Any]]]]:
+        self, size: int | None = None
+    ) -> list[tuple[Any | None, ...] | dict[Any, Any | None]]:
         """Fetch multiple rows from the result set.
 
         Args:
@@ -470,7 +469,7 @@ class WithAsyncFetch(AioBaseCursor, CursorIterator, WithResultSet):
 
     def fetchall(
         self,
-    ) -> List[Union[Tuple[Optional[Any], ...], Dict[Any, Optional[Any]]]]:
+    ) -> list[tuple[Any | None, ...] | dict[Any, Any | None]]:
         """Fetch all remaining rows from the result set.
 
         Returns:

{pyathena-3.29.3 → pyathena-3.30.0}/pyathena/aio/connection.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 from __future__ import annotations
 
 import asyncio
@@ -33,7 +32,7 @@ class AioConnection(Connection[AioCursor]):
     async def create(
         cls,
         **kwargs: Any,
-    ) -> "AioConnection":
+    ) -> AioConnection:
         """Async factory for creating an ``AioConnection``.
 
         Runs the (potentially blocking) ``__init__`` in a thread so that
@@ -47,7 +46,7 @@ class AioConnection(Connection[AioCursor]):
         """
         return await asyncio.to_thread(cls, **kwargs)
 
-    async def __aenter__(self) -> "AioConnection":
+    async def __aenter__(self) -> AioConnection:
         return self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb) -> None:

{pyathena-3.29.3 → pyathena-3.30.0}/pyathena/aio/cursor.py

@@ -1,8 +1,7 @@
-# -*- coding: utf-8 -*-
 from __future__ import annotations
 
 import logging
-from typing import Any, Dict, List, Optional, Union, cast
+from typing import Any, cast
 
 from pyathena.aio.common import WithAsyncFetch
 from pyathena.aio.result_set import AthenaAioDictResultSet, AthenaAioResultSet
@@ -10,7 +9,7 @@ from pyathena.common import CursorIterator
 from pyathena.error import OperationalError, ProgrammingError
 from pyathena.model import AthenaQueryExecution
 
-_logger = logging.getLogger(__name__)  # type: ignore
+_logger = logging.getLogger(__name__)
 
 
 class AioCursor(WithAsyncFetch):
@@ -29,13 +28,13 @@ class AioCursor(WithAsyncFetch):
 
     def __init__(
         self,
-        s3_staging_dir: Optional[str] = None,
-        schema_name: Optional[str] = None,
-        catalog_name: Optional[str] = None,
-        work_group: Optional[str] = None,
+        s3_staging_dir: str | None = None,
+        schema_name: str | None = None,
+        catalog_name: str | None = None,
+        work_group: str | None = None,
         poll_interval: float = 1,
-        encryption_option: Optional[str] = None,
-        kms_key: Optional[str] = None,
+        encryption_option: str | None = None,
+        kms_key: str | None = None,
         kill_on_interrupt: bool = True,
         result_reuse_enable: bool = False,
         result_reuse_minutes: int = CursorIterator.DEFAULT_RESULT_REUSE_MINUTES,
@@ -54,7 +53,7 @@ class AioCursor(WithAsyncFetch):
             result_reuse_minutes=result_reuse_minutes,
             **kwargs,
         )
-        self._result_set: Optional[AthenaAioResultSet] = None
+        self._result_set: AthenaAioResultSet | None = None
         self._result_set_class = AthenaAioResultSet
 
     @property
@@ -72,16 +71,17 @@ class AioCursor(WithAsyncFetch):
     async def execute(  # type: ignore[override]
         self,
         operation: str,
-        parameters: Optional[Union[Dict[str, Any], List[str]]] = None,
-        work_group: Optional[str] = None,
-        s3_staging_dir: Optional[str] = None,
+        parameters: dict[str, Any] | list[str] | None = None,
+        work_group: str | None = None,
+        s3_staging_dir: str | None = None,
         cache_size: int = 0,
         cache_expiration_time: int = 0,
-        result_reuse_enable: Optional[bool] = None,
-        result_reuse_minutes: Optional[int] = None,
-        paramstyle: Optional[str] = None,
+        result_reuse_enable: bool | None = None,
+        result_reuse_minutes: int | None = None,
+        paramstyle: str | None = None,
+        result_set_type_hints: dict[str | int, str] | None = None,
         **kwargs,
-    ) -> "AioCursor":
+    ) -> AioCursor:
         """Execute a SQL query asynchronously.
 
         Args:
@@ -94,6 +94,9 @@ class AioCursor(WithAsyncFetch):
             result_reuse_enable: Enable result reuse (optional).
             result_reuse_minutes: Result reuse duration in minutes (optional).
             paramstyle: Parameter style to use (optional).
+            result_set_type_hints: Optional dictionary mapping column names to
+                Athena DDL type signatures for precise type conversion within
+                complex types.
             **kwargs: Additional execution parameters.
 
         Returns:
@@ -120,6 +123,7 @@ class AioCursor(WithAsyncFetch):
                 query_execution,
                 self.arraysize,
                 self._retry_config,
+                result_set_type_hints=result_set_type_hints,
             )
         else:
             raise OperationalError(query_execution.state_change_reason)
@@ -127,7 +131,7 @@ class AioCursor(WithAsyncFetch):
 
     async def fetchone(  # type: ignore[override]
         self,
-    ) -> Optional[Union[Any, Dict[Any, Optional[Any]]]]:
+    ) -> Any | dict[Any, Any | None] | None:
         """Fetch the next row of a query result set.
 
         Returns:
@@ -143,8 +147,8 @@ class AioCursor(WithAsyncFetch):
         return await result_set.fetchone()
 
     async def fetchmany(  # type: ignore[override]
-        self, size: Optional[int] = None
-    ) -> List[Union[Any, Dict[Any, Optional[Any]]]]:
+        self, size: int | None = None
+    ) -> list[Any | dict[Any, Any | None]]:
         """Fetch multiple rows from a query result set.
 
         Args:
@@ -164,7 +168,7 @@ class AioCursor(WithAsyncFetch):
 
     async def fetchall(  # type: ignore[override]
         self,
-    ) -> List[Union[Any, Dict[Any, Optional[Any]]]]:
+    ) -> list[Any | dict[Any, Any | None]]:
         """Fetch all remaining rows from a query result set.
 
         Returns: