fix: mypy

Author: jaidisido

awswrangler/athena/_statements.py

@@ -97,7 +97,9 @@ def create_prepared_statement(
 
 
 @apply_configs
-def list_prepared_statements(workgroup: str = "primary", boto3_session: boto3.Session | None = None) -> list[str]:
+def list_prepared_statements(
+    workgroup: str = "primary", boto3_session: boto3.Session | None = None
+) -> list[dict[str, Any]]:
     """
     List the prepared statements in the specified workgroup.
 

awswrangler/athena/_utils.py

@@ -16,7 +16,6 @@
     Dict,
     Generator,
     NamedTuple,
-    Sequence,
     TypedDict,
     Union,
     cast,
@@ -36,7 +35,7 @@
 from ._cache import _cache_manager, _LocalMetadataCacheManager
 
 if TYPE_CHECKING:
-    from mypy_boto3_glue.type_defs import ColumnTypeDef
+    from mypy_boto3_glue.type_defs import ColumnOutputTypeDef
 
 _QUERY_FINAL_STATES: list[str] = ["FAILED", "SUCCEEDED", "CANCELLED"]
 _QUERY_WAIT_POLLING_DELAY: float = 1.0  # SECONDS
@@ -993,7 +992,7 @@ def generate_create_query(
 
     """
 
-    def parse_columns(columns_description: Sequence["ColumnTypeDef"]) -> str:
+    def parse_columns(columns_description: list["ColumnOutputTypeDef"]) -> str:
         columns_str: list[str] = []
         for column in columns_description:
             column_str = f"  `{column['Name']}` {column['Type']}"

awswrangler/catalog/_get.py

@@ -5,7 +5,7 @@
 import base64
 import itertools
 import logging
-from typing import TYPE_CHECKING, Any, Dict, Iterator, Mapping, cast
+from typing import TYPE_CHECKING, Any, Dict, Iterator, cast
 
 import boto3
 import botocore.exceptions
@@ -893,7 +893,7 @@ def get_columns_parameters(
     table: str,
     catalog_id: str | None = None,
     boto3_session: boto3.Session | None = None,
-) -> dict[str, Mapping[str, str] | None]:
+) -> dict[str, dict[str, str] | None]:
     """Get all columns parameters.
 
     Parameters

awswrangler/dynamodb/_utils.py

@@ -23,7 +23,7 @@
         ExecuteStatementOutputTypeDef,
         KeySchemaElementTypeDef,
         TableAttributeValueTypeDef,
-        WriteRequestTypeDef,
+        WriteRequestOutputTypeDef,
     )
 
 
@@ -214,7 +214,7 @@ def __init__(
     ):
         self._table_name = table_name
         self._client = client
-        self._items_buffer: list["WriteRequestTypeDef"] = []
+        self._items_buffer: list["WriteRequestOutputTypeDef"] = []
         self._flush_amount = flush_amount
         self._overwrite_by_pkeys = overwrite_by_pkeys
 
@@ -240,14 +240,16 @@ def delete_item(self, key: dict[str, "AttributeValueTypeDef"]) -> None:
         """
         self._add_request_and_process({"DeleteRequest": {"Key": key}})
 
-    def _add_request_and_process(self, request: "WriteRequestTypeDef") -> None:
+    def _add_request_and_process(self, request: "WriteRequestOutputTypeDef") -> None:
         if self._overwrite_by_pkeys:
             self._remove_dup_pkeys_request_if_any(request, self._overwrite_by_pkeys)
 
         self._items_buffer.append(request)
         self._flush_if_needed()
 
-    def _remove_dup_pkeys_request_if_any(self, request: "WriteRequestTypeDef", overwrite_by_pkeys: list[str]) -> None:
+    def _remove_dup_pkeys_request_if_any(
+        self, request: "WriteRequestOutputTypeDef", overwrite_by_pkeys: list[str]
+    ) -> None:
         pkey_values_new = self._extract_pkey_values(request, overwrite_by_pkeys)
         for item in self._items_buffer:
             if self._extract_pkey_values(item, overwrite_by_pkeys) == pkey_values_new:
@@ -257,7 +259,9 @@ def _remove_dup_pkeys_request_if_any(self, request: "WriteRequestTypeDef", overw
                     item,
                 )
 
-    def _extract_pkey_values(self, request: "WriteRequestTypeDef", overwrite_by_pkeys: list[str]) -> list[Any] | None:
+    def _extract_pkey_values(
+        self, request: "WriteRequestOutputTypeDef", overwrite_by_pkeys: list[str]
+    ) -> list[Any] | None:
         if request.get("PutRequest"):
             return [request["PutRequest"]["Item"][key] for key in overwrite_by_pkeys]
         if request.get("DeleteRequest"):