mypy-boto3-athena 1.26.20__py3-none-any.whl → 1.28.36__py3-none-any.whl

mypy_boto3_athena/__main__.py

@@ -9,8 +9,8 @@ def print_info() -> None:
     Print package info to stdout.
     """
     print(
-        "Type annotations for boto3.Athena 1.26.20\nVersion:         1.26.20\nBuilder version:"
-        " 7.11.11\nDocs:           "
+        "Type annotations for boto3.Athena 1.28.36\nVersion:         1.28.36\nBuilder version:"
+        " 7.18.0\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:      "
@@ -22,7 +22,7 @@ def print_version() -> None:
     """
     Print package version to stdout.
     """
-    print("1.26.20")
+    print("1.28.36")
 
 
 def main() -> None:

mypy_boto3_athena/client.py

@@ -39,6 +39,7 @@ from .type_defs import (
     BatchGetPreparedStatementOutputTypeDef,
     BatchGetQueryExecutionOutputTypeDef,
     CalculationConfigurationTypeDef,
+    CapacityAssignmentTypeDef,
     CreateNamedQueryOutputTypeDef,
     CreateNotebookOutputTypeDef,
     CreatePresignedNotebookUrlResponseTypeDef,
@@ -48,6 +49,8 @@ from .type_defs import (
     GetCalculationExecutionCodeResponseTypeDef,
     GetCalculationExecutionResponseTypeDef,
     GetCalculationExecutionStatusResponseTypeDef,
+    GetCapacityAssignmentConfigurationOutputTypeDef,
+    GetCapacityReservationOutputTypeDef,
     GetDatabaseOutputTypeDef,
     GetDataCatalogOutputTypeDef,
     GetNamedQueryOutputTypeDef,
@@ -63,6 +66,7 @@ from .type_defs import (
     ImportNotebookOutputTypeDef,
     ListApplicationDPUSizesOutputTypeDef,
     ListCalculationExecutionsResponseTypeDef,
+    ListCapacityReservationsOutputTypeDef,
     ListDatabasesOutputTypeDef,
     ListDataCatalogsOutputTypeDef,
     ListEngineVersionsOutputTypeDef,
@@ -89,7 +93,7 @@ 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
@@ -174,6 +178,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 +194,17 @@ 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,
         *,
@@ -260,6 +283,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.
@@ -340,8 +371,7 @@ class AthenaClient(BaseClient):
         self, *, CalculationExecutionId: str
     ) -> GetCalculationExecutionCodeResponseTypeDef:
         """
-        Retrieves a pre-signed URL to a copy of the code that was executed for the
-        calculation.
+        Retrieves the unencrypted code that was executed for the calculation.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_calculation_execution_code)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_calculation_execution_code)
@@ -357,6 +387,25 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_calculation_execution_status)
         """
 
+    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) -> GetDataCatalogOutputTypeDef:
         """
         Returns the specified data catalog.
@@ -487,7 +536,7 @@ class AthenaClient(BaseClient):
     ) -> ListApplicationDPUSizesOutputTypeDef:
         """
         Returns the supported DPU sizes for the supported application runtimes (for
-        example, `Jupyter 1.0` ).
+        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)
@@ -509,6 +558,16 @@ class AthenaClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_calculation_executions)
         """
 
+    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 = ...
     ) -> ListDataCatalogsOutputTypeDef:
@@ -549,7 +608,7 @@ class AthenaClient(BaseClient):
         NextToken: str = ...
     ) -> ListExecutorsResponseTypeDef:
         """
-        Lists, in descending order, the executors that have been submitted to a session.
+        Lists, in descending order, the executors that joined a session.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_executors)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_executors)
@@ -586,7 +645,7 @@ 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)
@@ -622,8 +681,8 @@ class AthenaClient(BaseClient):
         NextToken: str = ...
     ) -> ListSessionsResponseTypeDef:
         """
-        Lists the sessions in a workgroup that are in an active state like `CREATING` ,
-        `CREATED` , `IDLE` , or `BUSY`.
+        Lists the sessions in a workgroup 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_sessions)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_sessions)
@@ -649,7 +708,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)
@@ -665,6 +724,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[CapacityAssignmentTypeDef]
+    ) -> 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,
         *,
@@ -752,12 +825,21 @@ 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,
         *,
@@ -787,8 +869,8 @@ class AthenaClient(BaseClient):
         self,
         *,
         NotebookId: str,
-        Payload: str = ...,
-        Type: Literal["IPYNB"] = ...,
+        Payload: str,
+        Type: Literal["IPYNB"],
         SessionId: str = ...,
         ClientRequestToken: str = ...
     ) -> Dict[str, Any]:

mypy_boto3_athena/client.pyi

@@ -39,6 +39,7 @@ from .type_defs import (
     BatchGetPreparedStatementOutputTypeDef,
     BatchGetQueryExecutionOutputTypeDef,
     CalculationConfigurationTypeDef,
+    CapacityAssignmentTypeDef,
     CreateNamedQueryOutputTypeDef,
     CreateNotebookOutputTypeDef,
     CreatePresignedNotebookUrlResponseTypeDef,
@@ -48,6 +49,8 @@ from .type_defs import (
     GetCalculationExecutionCodeResponseTypeDef,
     GetCalculationExecutionResponseTypeDef,
     GetCalculationExecutionStatusResponseTypeDef,
+    GetCapacityAssignmentConfigurationOutputTypeDef,
+    GetCapacityReservationOutputTypeDef,
     GetDatabaseOutputTypeDef,
     GetDataCatalogOutputTypeDef,
     GetNamedQueryOutputTypeDef,
@@ -63,6 +66,7 @@ from .type_defs import (
     ImportNotebookOutputTypeDef,
     ListApplicationDPUSizesOutputTypeDef,
     ListCalculationExecutionsResponseTypeDef,
+    ListCapacityReservationsOutputTypeDef,
     ListDatabasesOutputTypeDef,
     ListDataCatalogsOutputTypeDef,
     ListEngineVersionsOutputTypeDef,
@@ -89,7 +93,7 @@ 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
@@ -165,6 +169,13 @@ class AthenaClient(BaseClient):
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.can_paginate)
         [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.
@@ -172,6 +183,16 @@ class AthenaClient(BaseClient):
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.close)
         [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,
         *,
@@ -244,6 +265,13 @@ class AthenaClient(BaseClient):
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.create_work_group)
         [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.
@@ -316,8 +344,7 @@ class AthenaClient(BaseClient):
         self, *, CalculationExecutionId: str
     ) -> GetCalculationExecutionCodeResponseTypeDef:
         """
-        Retrieves a pre-signed URL to a copy of the code that was executed for the
-        calculation.
+        Retrieves the unencrypted code that was executed for the calculation.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_calculation_execution_code)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_calculation_execution_code)
@@ -331,6 +358,23 @@ class AthenaClient(BaseClient):
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.get_calculation_execution_status)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#get_calculation_execution_status)
         """
+    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) -> GetDataCatalogOutputTypeDef:
         """
         Returns the specified data catalog.
@@ -448,7 +492,7 @@ class AthenaClient(BaseClient):
     ) -> ListApplicationDPUSizesOutputTypeDef:
         """
         Returns the supported DPU sizes for the supported application runtimes (for
-        example, `Jupyter 1.0` ).
+        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)
@@ -468,6 +512,15 @@ class AthenaClient(BaseClient):
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_calculation_executions)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_calculation_executions)
         """
+    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 = ...
     ) -> ListDataCatalogsOutputTypeDef:
@@ -505,7 +558,7 @@ class AthenaClient(BaseClient):
         NextToken: str = ...
     ) -> ListExecutorsResponseTypeDef:
         """
-        Lists, in descending order, the executors that have been submitted to a session.
+        Lists, in descending order, the executors that joined a session.
 
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_executors)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_executors)
@@ -539,7 +592,7 @@ 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)
@@ -572,8 +625,8 @@ class AthenaClient(BaseClient):
         NextToken: str = ...
     ) -> ListSessionsResponseTypeDef:
         """
-        Lists the sessions in a workgroup that are in an active state like `CREATING` ,
-        `CREATED` , `IDLE` , or `BUSY`.
+        Lists the sessions in a workgroup 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_sessions)
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_athena/client/#list_sessions)
@@ -597,7 +650,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)
@@ -611,6 +664,19 @@ class AthenaClient(BaseClient):
         [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/athena.html#Athena.Client.list_work_groups)
         [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[CapacityAssignmentTypeDef]
+    ) -> 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,
         *,
@@ -691,11 +757,19 @@ 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,
         *,
@@ -723,8 +797,8 @@ class AthenaClient(BaseClient):
         self,
         *,
         NotebookId: str,
-        Payload: str = ...,
-        Type: Literal["IPYNB"] = ...,
+        Payload: str,
+        Type: Literal["IPYNB"],
         SessionId: str = ...,
         ClientRequestToken: str = ...
     ) -> Dict[str, Any]:

mypy_boto3_athena/literals.py

@@ -13,7 +13,7 @@ Usage::
 """
 import sys
 
-if sys.version_info >= (3, 9):
+if sys.version_info >= (3, 12):
     from typing import Literal
 else:
     from typing_extensions import Literal
@@ -21,6 +21,8 @@ else:
 
 __all__ = (
     "CalculationExecutionStateType",
+    "CapacityAllocationStatusType",
+    "CapacityReservationStatusType",
     "ColumnNullableType",
     "DataCatalogTypeType",
     "EncryptionOptionType",
@@ -50,6 +52,10 @@ __all__ = (
 CalculationExecutionStateType = Literal[
     "CANCELED", "CANCELING", "COMPLETED", "CREATED", "CREATING", "FAILED", "QUEUED", "RUNNING"
 ]
+CapacityAllocationStatusType = Literal["FAILED", "PENDING", "SUCCEEDED"]
+CapacityReservationStatusType = Literal[
+    "ACTIVE", "CANCELLED", "CANCELLING", "FAILED", "PENDING", "UPDATE_PENDING"
+]
 ColumnNullableType = Literal["NOT_NULL", "NULLABLE", "UNKNOWN"]
 DataCatalogTypeType = Literal["GLUE", "HIVE", "LAMBDA"]
 EncryptionOptionType = Literal["CSE_KMS", "SSE_KMS", "SSE_S3"]
@@ -88,6 +94,7 @@ ServiceName = Literal[
     "apigatewayv2",
     "appconfig",
     "appconfigdata",
+    "appfabric",
     "appflow",
     "appintegrations",
     "application-autoscaling",
@@ -116,6 +123,7 @@ ServiceName = Literal[
     "chime-sdk-meetings",
     "chime-sdk-messaging",
     "chime-sdk-voice",
+    "cleanrooms",
     "cloud9",
     "cloudcontrol",
     "clouddirectory",
@@ -126,12 +134,15 @@ ServiceName = Literal[
     "cloudsearch",
     "cloudsearchdomain",
     "cloudtrail",
+    "cloudtrail-data",
     "cloudwatch",
     "codeartifact",
     "codebuild",
+    "codecatalyst",
     "codecommit",
     "codedeploy",
     "codeguru-reviewer",
+    "codeguru-security",
     "codeguruprofiler",
     "codepipeline",
     "codestar",
@@ -187,6 +198,7 @@ ServiceName = Literal[
     "emr",
     "emr-containers",
     "emr-serverless",
+    "entityresolution",
     "es",
     "events",
     "evidently",
@@ -218,6 +230,7 @@ ServiceName = Literal[
     "importexport",
     "inspector",
     "inspector2",
+    "internetmonitor",
     "iot",
     "iot-data",
     "iot-jobs-data",
@@ -236,15 +249,18 @@ ServiceName = Literal[
     "iottwinmaker",
     "iotwireless",
     "ivs",
+    "ivs-realtime",
     "ivschat",
     "kafka",
     "kafkaconnect",
     "kendra",
+    "kendra-ranking",
     "keyspaces",
     "kinesis",
     "kinesis-video-archived-media",
     "kinesis-video-media",
     "kinesis-video-signaling",
+    "kinesis-video-webrtc-storage",
     "kinesisanalytics",
     "kinesisanalyticsv2",
     "kinesisvideo",
@@ -256,6 +272,7 @@ ServiceName = Literal[
     "lexv2-models",
     "lexv2-runtime",
     "license-manager",
+    "license-manager-linux-subscriptions",
     "license-manager-user-subscriptions",
     "lightsail",
     "location",
@@ -268,6 +285,7 @@ ServiceName = Literal[
     "macie",
     "macie2",
     "managedblockchain",
+    "managedblockchain-query",
     "marketplace-catalog",
     "marketplace-entitlement",
     "marketplacecommerceanalytics",
@@ -276,9 +294,11 @@ ServiceName = Literal[
     "medialive",
     "mediapackage",
     "mediapackage-vod",
+    "mediapackagev2",
     "mediastore",
     "mediastore-data",
     "mediatailor",
+    "medical-imaging",
     "memorydb",
     "meteringmarketplace",
     "mgh",
@@ -302,8 +322,11 @@ ServiceName = Literal[
     "opsworks",
     "opsworkscm",
     "organizations",
+    "osis",
     "outposts",
     "panorama",
+    "payment-cryptography",
+    "payment-cryptography-data",
     "personalize",
     "personalize-events",
     "personalize-runtime",
@@ -312,6 +335,7 @@ ServiceName = Literal[
     "pinpoint-email",
     "pinpoint-sms-voice",
     "pinpoint-sms-voice-v2",
+    "pipes",
     "polly",
     "pricing",
     "privatenetworks",
@@ -348,6 +372,7 @@ ServiceName = Literal[
     "sagemaker-edge",
     "sagemaker-featurestore-runtime",
     "sagemaker-geospatial",
+    "sagemaker-metrics",
     "sagemaker-runtime",
     "savingsplans",
     "scheduler",
@@ -389,10 +414,13 @@ ServiceName = Literal[
     "textract",
     "timestream-query",
     "timestream-write",
+    "tnb",
     "transcribe",
     "transfer",
     "translate",
+    "verifiedpermissions",
     "voice-id",
+    "vpc-lattice",
     "waf",
     "waf-regional",
     "wafv2",
@@ -434,16 +462,22 @@ RegionName = Literal[
     "ap-northeast-2",
     "ap-northeast-3",
     "ap-south-1",
+    "ap-south-2",
     "ap-southeast-1",
     "ap-southeast-2",
     "ap-southeast-3",
+    "ap-southeast-4",
     "ca-central-1",
     "eu-central-1",
+    "eu-central-2",
     "eu-north-1",
     "eu-south-1",
+    "eu-south-2",
     "eu-west-1",
     "eu-west-2",
     "eu-west-3",
+    "il-central-1",
+    "me-central-1",
     "me-south-1",
     "sa-east-1",
     "us-east-1",