mypy-boto3-athena 1.26.100__py3-none-any.whl → 1.34.130__py3-none-any.whl

mypy_boto3_athena/__init__.py

@@ -29,6 +29,7 @@ Usage::
     list_tags_for_resource_paginator: ListTagsForResourcePaginator = client.get_paginator("list_tags_for_resource")
     ```
 """
+
 from .client import AthenaClient
 from .paginator import (
     GetQueryResultsPaginator,
@@ -42,7 +43,6 @@ from .paginator import (
 
 Client = AthenaClient
 
-
 __all__ = (
     "AthenaClient",
     "Client",

mypy_boto3_athena/__init__.pyi

@@ -29,6 +29,7 @@ Usage::
     list_tags_for_resource_paginator: ListTagsForResourcePaginator = client.get_paginator("list_tags_for_resource")
     ```
 """
+
 from .client import AthenaClient
 from .paginator import (
     GetQueryResultsPaginator,

mypy_boto3_athena/__main__.py

@@ -1,6 +1,7 @@
 """
 Main CLI entrypoint.
 """
+
 import sys
 
 
@@ -9,12 +10,13 @@ def print_info() -> None:
     Print package info to stdout.
     """
     print(
-        "Type annotations for boto3.Athena 1.26.100\nVersion:         1.26.100\nBuilder version:"
-        " 7.14.4\nDocs:           "
-        " https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena//\nBoto3 docs:     "
-        " https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena\nOther"
-        " services:  https://pypi.org/project/boto3-stubs/\nChangelog:      "
-        " https://github.com/youtype/mypy_boto3_builder/releases"
+        "Type annotations for boto3.Athena 1.34.130\n"
+        "Version:         1.34.130\n"
+        "Builder version: 7.24.0\n"
+        "Docs:            https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena//\n"
+        "Boto3 docs:      https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena\n"
+        "Other services:  https://pypi.org/project/boto3-stubs/\n"
+        "Changelog:       https://github.com/youtype/mypy_boto3_builder/releases"
     )
 
 
@@ -22,7 +24,7 @@ def print_version() -> None:
     """
     Print package version to stdout.
     """
-    print("1.26.100")
+    print("1.34.130")
 
 
 def main() -> None:

mypy_boto3_athena/client.py

@@ -13,6 +13,7 @@ Usage::
     client: AthenaClient = session.client("athena")
     ```
 """
+
 import sys
 from typing import Any, Dict, Mapping, Sequence, Type, overload
 
@@ -39,15 +40,18 @@ from .type_defs import (
     BatchGetPreparedStatementOutputTypeDef,
     BatchGetQueryExecutionOutputTypeDef,
     CalculationConfigurationTypeDef,
+    CapacityAssignmentUnionTypeDef,
     CreateNamedQueryOutputTypeDef,
     CreateNotebookOutputTypeDef,
     CreatePresignedNotebookUrlResponseTypeDef,
-    EngineConfigurationTypeDef,
+    EngineConfigurationUnionTypeDef,
     ExportNotebookOutputTypeDef,
     FilterDefinitionTypeDef,
     GetCalculationExecutionCodeResponseTypeDef,
     GetCalculationExecutionResponseTypeDef,
     GetCalculationExecutionStatusResponseTypeDef,
+    GetCapacityAssignmentConfigurationOutputTypeDef,
+    GetCapacityReservationOutputTypeDef,
     GetDatabaseOutputTypeDef,
     GetDataCatalogOutputTypeDef,
     GetNamedQueryOutputTypeDef,
@@ -63,6 +67,7 @@ from .type_defs import (
     ImportNotebookOutputTypeDef,
     ListApplicationDPUSizesOutputTypeDef,
     ListCalculationExecutionsResponseTypeDef,
+    ListCapacityReservationsOutputTypeDef,
     ListDatabasesOutputTypeDef,
     ListDataCatalogsOutputTypeDef,
     ListEngineVersionsOutputTypeDef,
@@ -89,16 +94,15 @@ from .type_defs import (
     WorkGroupConfigurationUpdatesTypeDef,
 )
 
-if sys.version_info >= (3, 9):
+if sys.version_info >= (3, 12):
     from typing import Literal
 else:
     from typing_extensions import Literal
 
-
 __all__ = ("AthenaClient",)
 
 
-class BotocoreClientError(BaseException):
+class BotocoreClientError(Exception):
     MSG_TEMPLATE: str
 
     def __init__(self, error_response: Mapping[str, Any], operation_name: str) -> None:
@@ -137,8 +141,9 @@ class AthenaClient(BaseClient):
         self, *, NamedQueryIds: Sequence[str]
     ) -> BatchGetNamedQueryOutputTypeDef:
         """
-        Returns the details of a single named query or a list of up to 50 queries, which
-        you provide as an array of query ID strings.
+        Returns the details of a single named query or a list of up to 50 queries,
+        which you provide as an array of query ID
+        strings.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.batch_get_named_query)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#batch_get_named_query)
@@ -149,7 +154,8 @@ class AthenaClient(BaseClient):
     ) -> BatchGetPreparedStatementOutputTypeDef:
         """
         Returns the details of a single prepared statement or a list of up to 256
-        prepared statements for the array of prepared statement names that you provide.
+        prepared statements for the array of prepared statement names that you
+        provide.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.batch_get_prepared_statement)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#batch_get_prepared_statement)
@@ -160,7 +166,8 @@ class AthenaClient(BaseClient):
     ) -> BatchGetQueryExecutionOutputTypeDef:
         """
         Returns the details of a single query execution or a list of up to 50 query
-        executions, which you provide as an array of query execution ID strings.
+        executions, which you provide as an array of query execution ID
+        strings.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.batch_get_query_execution)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#batch_get_query_execution)
@@ -174,6 +181,14 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#can_paginate)
         """
 
+    def cancel_capacity_reservation(self, *, Name: str) -> Dict[str, Any]:
+        """
+        Cancels the capacity reservation with the specified name.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.cancel_capacity_reservation)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#cancel_capacity_reservation)
+        """
+
     def close(self) -> None:
         """
         Closes underlying endpoint connections.
@@ -182,6 +197,18 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#close)
         """
 
+    def create_capacity_reservation(
+        self, *, TargetDpus: int, Name: str, Tags: Sequence[TagTypeDef] = ...
+    ) -> Dict[str, Any]:
+        """
+        Creates a capacity reservation with the specified name and number of requested
+        data processing
+        units.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.create_capacity_reservation)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#create_capacity_reservation)
+        """
+
     def create_data_catalog(
         self,
         *,
@@ -189,7 +216,7 @@ class AthenaClient(BaseClient):
         Type: DataCatalogTypeType,
         Description: str = ...,
         Parameters: Mapping[str, str] = ...,
-        Tags: Sequence[TagTypeDef] = ...
+        Tags: Sequence[TagTypeDef] = ...,
     ) -> Dict[str, Any]:
         """
         Creates (registers) a data catalog with the specified name and properties.
@@ -206,7 +233,7 @@ class AthenaClient(BaseClient):
         QueryString: str,
         Description: str = ...,
         ClientRequestToken: str = ...,
-        WorkGroup: str = ...
+        WorkGroup: str = ...,
     ) -> CreateNamedQueryOutputTypeDef:
         """
         Creates a named query in the specified workgroup.
@@ -251,7 +278,7 @@ class AthenaClient(BaseClient):
         Name: str,
         Configuration: WorkGroupConfigurationTypeDef = ...,
         Description: str = ...,
-        Tags: Sequence[TagTypeDef] = ...
+        Tags: Sequence[TagTypeDef] = ...,
     ) -> Dict[str, Any]:
         """
         Creates a workgroup with the specified name.
@@ -260,6 +287,14 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#create_work_group)
         """
 
+    def delete_capacity_reservation(self, *, Name: str) -> Dict[str, Any]:
+        """
+        Deletes a cancelled capacity reservation.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.delete_capacity_reservation)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#delete_capacity_reservation)
+        """
+
     def delete_data_catalog(self, *, Name: str) -> Dict[str, Any]:
         """
         Deletes a data catalog.
@@ -271,7 +306,8 @@ class AthenaClient(BaseClient):
     def delete_named_query(self, *, NamedQueryId: str) -> Dict[str, Any]:
         """
         Deletes the named query if you have access to the workgroup in which the query
-        was saved.
+        was
+        saved.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.delete_named_query)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#delete_named_query)
@@ -356,7 +392,26 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_calculation_execution_status)
         """
 
-    def get_data_catalog(self, *, Name: str) -> GetDataCatalogOutputTypeDef:
+    def get_capacity_assignment_configuration(
+        self, *, CapacityReservationName: str
+    ) -> GetCapacityAssignmentConfigurationOutputTypeDef:
+        """
+        Gets the capacity assignment configuration for a capacity reservation, if one
+        exists.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_capacity_assignment_configuration)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_capacity_assignment_configuration)
+        """
+
+    def get_capacity_reservation(self, *, Name: str) -> GetCapacityReservationOutputTypeDef:
+        """
+        Returns information about the capacity reservation with the specified name.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_capacity_reservation)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_capacity_reservation)
+        """
+
+    def get_data_catalog(self, *, Name: str, WorkGroup: str = ...) -> GetDataCatalogOutputTypeDef:
         """
         Returns the specified data catalog.
 
@@ -364,7 +419,9 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_data_catalog)
         """
 
-    def get_database(self, *, CatalogName: str, DatabaseName: str) -> GetDatabaseOutputTypeDef:
+    def get_database(
+        self, *, CatalogName: str, DatabaseName: str, WorkGroup: str = ...
+    ) -> GetDatabaseOutputTypeDef:
         """
         Returns a database object for the specified database and data catalog.
 
@@ -402,7 +459,8 @@ class AthenaClient(BaseClient):
     def get_query_execution(self, *, QueryExecutionId: str) -> GetQueryExecutionOutputTypeDef:
         """
         Returns information about a single execution of a query if you have access to
-        the workgroup in which the query ran.
+        the workgroup in which the query
+        ran.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_query_execution)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_query_execution)
@@ -413,7 +471,8 @@ class AthenaClient(BaseClient):
     ) -> GetQueryResultsOutputTypeDef:
         """
         Streams the results of a single query execution specified by `QueryExecutionId`
-        from the Athena query results location in Amazon S3.
+        from the Athena query results location in Amazon
+        S3.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_query_results)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_query_results)
@@ -424,7 +483,8 @@ class AthenaClient(BaseClient):
     ) -> GetQueryRuntimeStatisticsOutputTypeDef:
         """
         Returns query execution runtime statistics related to a single execution of a
-        query if you have access to the workgroup in which the query ran.
+        query if you have access to the workgroup in which the query
+        ran.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_query_runtime_statistics)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_query_runtime_statistics)
@@ -433,7 +493,8 @@ class AthenaClient(BaseClient):
     def get_session(self, *, SessionId: str) -> GetSessionResponseTypeDef:
         """
         Gets the full details of a previously created session, including the session
-        status and configuration.
+        status and
+        configuration.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_session)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_session)
@@ -448,7 +509,7 @@ class AthenaClient(BaseClient):
         """
 
     def get_table_metadata(
-        self, *, CatalogName: str, DatabaseName: str, TableName: str
+        self, *, CatalogName: str, DatabaseName: str, TableName: str, WorkGroup: str = ...
     ) -> GetTableMetadataOutputTypeDef:
         """
         Returns table metadata for the specified catalog, database, and table.
@@ -470,9 +531,10 @@ class AthenaClient(BaseClient):
         *,
         WorkGroup: str,
         Name: str,
-        Payload: str,
         Type: Literal["IPYNB"],
-        ClientRequestToken: str = ...
+        Payload: str = ...,
+        NotebookS3LocationUri: str = ...,
+        ClientRequestToken: str = ...,
     ) -> ImportNotebookOutputTypeDef:
         """
         Imports a single `ipynb` file to a Spark enabled workgroup.
@@ -486,7 +548,8 @@ class AthenaClient(BaseClient):
     ) -> ListApplicationDPUSizesOutputTypeDef:
         """
         Returns the supported DPU sizes for the supported application runtimes (for
-        example, `Athena notebook version 1`).
+        example, `Athena notebook version
+        1`).
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_application_dpu_sizes)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_application_dpu_sizes)
@@ -498,7 +561,7 @@ class AthenaClient(BaseClient):
         SessionId: str,
         StateFilter: CalculationExecutionStateType = ...,
         MaxResults: int = ...,
-        NextToken: str = ...
+        NextToken: str = ...,
     ) -> ListCalculationExecutionsResponseTypeDef:
         """
         Lists the calculations that have been submitted to a session in descending
@@ -508,8 +571,18 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_calculation_executions)
         """
 
-    def list_data_catalogs(
+    def list_capacity_reservations(
         self, *, NextToken: str = ..., MaxResults: int = ...
+    ) -> ListCapacityReservationsOutputTypeDef:
+        """
+        Lists the capacity reservations for the current account.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_capacity_reservations)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_capacity_reservations)
+        """
+
+    def list_data_catalogs(
+        self, *, NextToken: str = ..., MaxResults: int = ..., WorkGroup: str = ...
     ) -> ListDataCatalogsOutputTypeDef:
         """
         Lists the data catalogs in the current Amazon Web Services account.
@@ -519,7 +592,7 @@ class AthenaClient(BaseClient):
         """
 
     def list_databases(
-        self, *, CatalogName: str, NextToken: str = ..., MaxResults: int = ...
+        self, *, CatalogName: str, NextToken: str = ..., MaxResults: int = ..., WorkGroup: str = ...
     ) -> ListDatabasesOutputTypeDef:
         """
         Lists the databases in the specified data catalog.
@@ -533,7 +606,8 @@ class AthenaClient(BaseClient):
     ) -> ListEngineVersionsOutputTypeDef:
         """
         Returns a list of engine versions that are available to choose from, including
-        the Auto option.
+        the Auto
+        option.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_engine_versions)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_engine_versions)
@@ -545,7 +619,7 @@ class AthenaClient(BaseClient):
         SessionId: str,
         ExecutorStateFilter: ExecutorStateType = ...,
         MaxResults: int = ...,
-        NextToken: str = ...
+        NextToken: str = ...,
     ) -> ListExecutorsResponseTypeDef:
         """
         Lists, in descending order, the executors that joined a session.
@@ -571,7 +645,7 @@ class AthenaClient(BaseClient):
         WorkGroup: str,
         Filters: FilterDefinitionTypeDef = ...,
         NextToken: str = ...,
-        MaxResults: int = ...
+        MaxResults: int = ...,
     ) -> ListNotebookMetadataOutputTypeDef:
         """
         Displays the notebook files for the specified workgroup in paginated format.
@@ -585,7 +659,8 @@ class AthenaClient(BaseClient):
     ) -> ListNotebookSessionsResponseTypeDef:
         """
         Lists, in descending order, the sessions that have been created in a notebook
-        that are in an active state like `CREATING`, `CREATED`, `IDLE` or `BUSY`.
+        that are in an active state like `CREATING`, `CREATED`, `IDLE` or
+        `BUSY`.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_notebook_sessions)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_notebook_sessions)
@@ -606,7 +681,8 @@ class AthenaClient(BaseClient):
     ) -> ListQueryExecutionsOutputTypeDef:
         """
         Provides a list of available query execution IDs for the queries in the
-        specified workgroup.
+        specified
+        workgroup.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_query_executions)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_query_executions)
@@ -618,11 +694,12 @@ class AthenaClient(BaseClient):
         WorkGroup: str,
         StateFilter: SessionStateType = ...,
         MaxResults: int = ...,
-        NextToken: str = ...
+        NextToken: str = ...,
     ) -> ListSessionsResponseTypeDef:
         """
         Lists the sessions in a workgroup that are in an active state like `CREATING`,
-        `CREATED`, `IDLE`, or `BUSY`.
+        `CREATED`, `IDLE`, or
+        `BUSY`.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_sessions)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_sessions)
@@ -635,7 +712,8 @@ class AthenaClient(BaseClient):
         DatabaseName: str,
         Expression: str = ...,
         NextToken: str = ...,
-        MaxResults: int = ...
+        MaxResults: int = ...,
+        WorkGroup: str = ...,
     ) -> ListTableMetadataOutputTypeDef:
         """
         Lists the metadata for the tables in the specified data catalog database.
@@ -648,7 +726,7 @@ class AthenaClient(BaseClient):
         self, *, ResourceARN: str, NextToken: str = ..., MaxResults: int = ...
     ) -> ListTagsForResourceOutputTypeDef:
         """
-        Lists the tags associated with an Athena workgroup or data catalog resource.
+        Lists the tags associated with an Athena resource.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_tags_for_resource)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_tags_for_resource)
@@ -664,6 +742,20 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_work_groups)
         """
 
+    def put_capacity_assignment_configuration(
+        self,
+        *,
+        CapacityReservationName: str,
+        CapacityAssignments: Sequence[CapacityAssignmentUnionTypeDef],
+    ) -> Dict[str, Any]:
+        """
+        Puts a new capacity assignment configuration for a specified capacity
+        reservation.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.put_capacity_assignment_configuration)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#put_capacity_assignment_configuration)
+        """
+
     def start_calculation_execution(
         self,
         *,
@@ -671,7 +763,7 @@ class AthenaClient(BaseClient):
         Description: str = ...,
         CalculationConfiguration: CalculationConfigurationTypeDef = ...,
         CodeBlock: str = ...,
-        ClientRequestToken: str = ...
+        ClientRequestToken: str = ...,
     ) -> StartCalculationExecutionResponseTypeDef:
         """
         Submits calculations for execution within a session.
@@ -689,7 +781,7 @@ class AthenaClient(BaseClient):
         ResultConfiguration: ResultConfigurationTypeDef = ...,
         WorkGroup: str = ...,
         ExecutionParameters: Sequence[str] = ...,
-        ResultReuseConfiguration: ResultReuseConfigurationTypeDef = ...
+        ResultReuseConfiguration: ResultReuseConfigurationTypeDef = ...,
     ) -> StartQueryExecutionOutputTypeDef:
         """
         Runs the SQL query statements contained in the `Query`.
@@ -702,11 +794,11 @@ class AthenaClient(BaseClient):
         self,
         *,
         WorkGroup: str,
-        EngineConfiguration: EngineConfigurationTypeDef,
+        EngineConfiguration: EngineConfigurationUnionTypeDef,
         Description: str = ...,
         NotebookVersion: str = ...,
         SessionIdleTimeoutInMinutes: int = ...,
-        ClientRequestToken: str = ...
+        ClientRequestToken: str = ...,
     ) -> StartSessionResponseTypeDef:
         """
         Creates a session for running calculations within a workgroup.
@@ -751,19 +843,29 @@ class AthenaClient(BaseClient):
 
     def untag_resource(self, *, ResourceARN: str, TagKeys: Sequence[str]) -> Dict[str, Any]:
         """
-        Removes one or more tags from a data catalog or workgroup resource.
+        Removes one or more tags from an Athena resource.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.untag_resource)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#untag_resource)
         """
 
+    def update_capacity_reservation(self, *, TargetDpus: int, Name: str) -> Dict[str, Any]:
+        """
+        Updates the number of requested data processing units for the capacity
+        reservation with the specified
+        name.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.update_capacity_reservation)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#update_capacity_reservation)
+        """
+
     def update_data_catalog(
         self,
         *,
         Name: str,
         Type: DataCatalogTypeType,
         Description: str = ...,
-        Parameters: Mapping[str, str] = ...
+        Parameters: Mapping[str, str] = ...,
     ) -> Dict[str, Any]:
         """
         Updates the data catalog that has the specified name.
@@ -789,7 +891,7 @@ class AthenaClient(BaseClient):
         Payload: str,
         Type: Literal["IPYNB"],
         SessionId: str = ...,
-        ClientRequestToken: str = ...
+        ClientRequestToken: str = ...,
     ) -> Dict[str, Any]:
         """
         Updates the contents of a Spark notebook.
@@ -824,7 +926,7 @@ class AthenaClient(BaseClient):
         WorkGroup: str,
         Description: str = ...,
         ConfigurationUpdates: WorkGroupConfigurationUpdatesTypeDef = ...,
-        State: WorkGroupStateType = ...
+        State: WorkGroupStateType = ...,
     ) -> Dict[str, Any]:
         """
         Updates the workgroup with the specified name.