sagemaker-core 1.0.13__py3-none-any.whl → 1.0.14__py3-none-any.whl

sagemaker_core/main/resources.py

@@ -3855,6 +3855,56 @@ class Cluster(Base):
         response = client.update_cluster_software(**operation_input_args)
         logger.debug(f"Response: {response}")
 
+    @Base.add_validate_call
+    def batch_delete_nodes(
+        self,
+        node_ids: List[str],
+        session: Optional[Session] = None,
+        region: Optional[str] = None,
+    ) -> Optional[BatchDeleteClusterNodesResponse]:
+        """
+        Deletes specific nodes within a SageMaker HyperPod cluster.
+
+        Parameters:
+            node_ids: A list of node IDs to be deleted from the specified cluster.  For SageMaker HyperPod clusters using the Slurm workload manager, you cannot remove instances that are configured as Slurm controller nodes.
+            session: Boto3 session.
+            region: Region name.
+
+        Returns:
+            BatchDeleteClusterNodesResponse
+
+        Raises:
+            botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
+                The error message and error code can be parsed from the exception as follows:
+                ```
+                try:
+                    # AWS service call here
+                except botocore.exceptions.ClientError as e:
+                    error_message = e.response['Error']['Message']
+                    error_code = e.response['Error']['Code']
+                ```
+            ResourceNotFound: Resource being access is not found.
+        """
+
+        operation_input_args = {
+            "ClusterName": self.cluster_name,
+            "NodeIds": node_ids,
+        }
+        # serialize the input request
+        operation_input_args = serialize(operation_input_args)
+        logger.debug(f"Serialized input request: {operation_input_args}")
+
+        client = Base.get_sagemaker_client(
+            session=session, region_name=region, service_name="sagemaker"
+        )
+
+        logger.debug(f"Calling batch_delete_cluster_nodes API")
+        response = client.batch_delete_cluster_nodes(**operation_input_args)
+        logger.debug(f"Response: {response}")
+
+        transformed_response = transform(response, "BatchDeleteClusterNodesResponse")
+        return BatchDeleteClusterNodesResponse(**transformed_response)
+
 
 class CodeRepository(Base):
     """

sagemaker_core/main/shapes.py

@@ -2377,6 +2377,20 @@ class BatchDeleteClusterNodesError(Base):
     node_id: str
 
 
+class BatchDeleteClusterNodesResponse(Base):
+    """
+    BatchDeleteClusterNodesResponse
+
+    Attributes
+    ----------------------
+    failed: A list of errors encountered when deleting the specified nodes.
+    successful: A list of node IDs that were successfully deleted from the specified cluster.
+    """
+
+    failed: Optional[List[BatchDeleteClusterNodesError]] = Unassigned()
+    successful: Optional[List[str]] = Unassigned()
+
+
 class BatchDescribeModelPackageError(Base):
     """
     BatchDescribeModelPackageError

{sagemaker_core-1.0.13.dist-info → sagemaker_core-1.0.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.13
+Version: 1.0.14
 Summary: An python package for sagemaker core functionalities
 Author-email: AWS <sagemaker-interests@amazon.com>
 Project-URL: Repository, https://github.com/aws/sagemaker-core.git

{sagemaker_core-1.0.13.dist-info → sagemaker_core-1.0.14.dist-info}/RECORD

@@ -7,8 +7,8 @@ sagemaker_core/main/config_schema.py,sha256=TeGoTobT4yotEVyfguLF0IdKYlOymsDZ45yS
 sagemaker_core/main/exceptions.py,sha256=87DUlrmHxaWoiYNlpNY9ixxFMPRk_dIGPsA2e_xdVwQ,5602
 sagemaker_core/main/intelligent_defaults_helper.py,sha256=5SDM6UavZtp-k5LhqRL7GRIDgzFB5UsC_p7YuiSPK9A,8334
 sagemaker_core/main/logs.py,sha256=yfEH7uP91nbE1lefymOlBr81ziBzsDSIOF2Qyd54FJE,6241
-sagemaker_core/main/resources.py,sha256=b6oYPs-7aQxs_R5Ql_G3TGO71gQuykobmD5HpVMdlDY,1321873
-sagemaker_core/main/shapes.py,sha256=sG_Pa36B5cHb-TR-smy6xS51kujnvu-BsJt9ByZlVKE,698850
+sagemaker_core/main/resources.py,sha256=idwGeRbSSO7Acwtd0y_zOkIxOrMlJvwxx-gL0k7GlS8,1323921
+sagemaker_core/main/shapes.py,sha256=Vge1sJb41xqpOFdCnzl8JNdIOwPflgr0Ioy2IKhP01g,699286
 sagemaker_core/main/user_agent.py,sha256=4sZybDXkzRoZnOnVDQ8p8zFTfiRJdsH7amDWInVQ4xU,2708
 sagemaker_core/main/utils.py,sha256=LCFDM6oxf6_e1i-_Dgtkm3ehl7YfoEpJ2kTTFTL6iOU,18471
 sagemaker_core/main/code_injection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -28,8 +28,8 @@ sagemaker_core/tools/resources_extractor.py,sha256=hN61ehZbPnhFW-2FIVDi7NsEz4rLv
 sagemaker_core/tools/shapes_codegen.py,sha256=_ve959bwH8usZ6dPlpXxi2on9t0hLpcmhRWnaWHCWMQ,11745
 sagemaker_core/tools/shapes_extractor.py,sha256=4KjgDmhlPM4G1f1NeYbORKlXs1s7Q_sm_NK31S_ROQ0,11950
 sagemaker_core/tools/templates.py,sha256=yX2RQKeClgYwKS5Qu_mDpnWJIBCuj0yELrdm95aiTpk,23262
-sagemaker_core-1.0.13.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.13.dist-info/METADATA,sha256=3e90CliozddoYVt0JUQgADZQ2I38nOlQl-x222rmWuY,4878
-sagemaker_core-1.0.13.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-sagemaker_core-1.0.13.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.13.dist-info/RECORD,,
+sagemaker_core-1.0.14.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-1.0.14.dist-info/METADATA,sha256=626z-hexHS0uj3CGQtc_f02Obevywo2OdAUISjfogz8,4878
+sagemaker_core-1.0.14.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+sagemaker_core-1.0.14.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
+sagemaker_core-1.0.14.dist-info/RECORD,,

{sagemaker_core-1.0.13.dist-info → sagemaker_core-1.0.14.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.3.0)
+Generator: setuptools (75.5.0)
 Root-Is-Purelib: true
 Tag: py3-none-any