fix typing

Author: LeonLuttenberger

awswrangler/_utils.py

@@ -856,7 +856,7 @@ def split_pandas_frame(df: pd.DataFrame, splits: int) -> list[pd.DataFrame]:
     total = len(df)
     each_section, extras = divmod(total, splits)
     section_sizes = [0] + extras * [each_section + 1] + (splits - extras) * [each_section]
-    div_points = _nx.array(section_sizes, dtype=_nx.intp).cumsum()  # type: ignore[attr-defined]
+    div_points = _nx.array(section_sizes, dtype=_nx.intp).cumsum()
 
     sub_dfs = []
     for i in range(splits):

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

@@ -36,7 +36,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 +993,7 @@ def generate_create_query(
 
     """
 
-    def parse_columns(columns_description: Sequence["ColumnTypeDef"]) -> str:
+    def parse_columns(columns_description: Sequence["ColumnOutputTypeDef"]) -> str:
         columns_str: list[str] = []
         for column in columns_description:
             column_str = f"  `{column['Name']}` {column['Type']}"

awswrangler/catalog/_get.py

@@ -921,7 +921,7 @@ def get_columns_parameters(
     """
     client_glue = _utils.client("glue", session=boto3_session)
     response = client_glue.get_table(**_catalog_id(catalog_id=catalog_id, DatabaseName=database, Name=table))
-    parameters = {}
+    parameters: dict[str, Mapping[str, str] | None] = {}
     for c in response["Table"]["StorageDescriptor"]["Columns"]:
         parameters[c["Name"]] = c.get("Parameters")
     if "PartitionKeys" in response["Table"]:

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"):