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

sagemaker_core/main/code_injection/shape_dag.py

@@ -749,6 +749,33 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "BatchDeleteClusterNodesError": {
+        "members": [
+            {"name": "Code", "shape": "BatchDeleteClusterNodesErrorCode", "type": "string"},
+            {"name": "Message", "shape": "String", "type": "string"},
+            {"name": "NodeId", "shape": "ClusterNodeId", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "BatchDeleteClusterNodesErrorList": {
+        "member_shape": "BatchDeleteClusterNodesError",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "BatchDeleteClusterNodesRequest": {
+        "members": [
+            {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
+            {"name": "NodeIds", "shape": "ClusterNodeIds", "type": "list"},
+        ],
+        "type": "structure",
+    },
+    "BatchDeleteClusterNodesResponse": {
+        "members": [
+            {"name": "Failed", "shape": "BatchDeleteClusterNodesErrorList", "type": "list"},
+            {"name": "Successful", "shape": "ClusterNodeIds", "type": "list"},
+        ],
+        "type": "structure",
+    },
     "BatchDescribeModelPackageError": {
         "members": [
             {"name": "ErrorCode", "shape": "String", "type": "string"},
@@ -1322,6 +1349,7 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "ClusterNodeIds": {"member_shape": "ClusterNodeId", "member_type": "string", "type": "list"},
     "ClusterNodeSummaries": {
         "member_shape": "ClusterNodeSummary",
         "member_type": "structure",
@@ -13566,6 +13594,7 @@ SHAPE_DAG = {
             {"name": "TrainingEndTime", "shape": "Timestamp", "type": "timestamp"},
             {"name": "LastModifiedTime", "shape": "Timestamp", "type": "timestamp"},
             {"name": "TrainingJobStatus", "shape": "TrainingJobStatus", "type": "string"},
+            {"name": "SecondaryStatus", "shape": "SecondaryStatus", "type": "string"},
             {"name": "WarmPoolStatus", "shape": "WarmPoolStatus", "type": "structure"},
         ],
         "type": "structure",

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

@@ -2360,6 +2360,37 @@ class BatchDataCaptureConfig(Base):
     generate_inference_id: Optional[bool] = Unassigned()
 
 
+class BatchDeleteClusterNodesError(Base):
+    """
+    BatchDeleteClusterNodesError
+      Represents an error encountered when deleting a node from a SageMaker HyperPod cluster.
+
+    Attributes
+    ----------------------
+    code: The error code associated with the error encountered when deleting a node. The code provides information about the specific issue encountered, such as the node not being found, the node's status being invalid for deletion, or the node ID being in use by another process.
+    message: A message describing the error encountered when deleting a node.
+    node_id: The ID of the node that encountered an error during the deletion process.
+    """
+
+    code: str
+    message: str
+    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
@@ -10913,6 +10944,7 @@ class TrainingJobSummary(Base):
     training_end_time: A timestamp that shows when the training job ended. This field is set only if the training job has one of the terminal statuses (Completed, Failed, or Stopped).
     last_modified_time:  Timestamp when the training job was last modified.
     training_job_status: The status of the training job.
+    secondary_status: The secondary status of the training job.
     warm_pool_status: The status of the warm pool associated with the training job.
     """
 
@@ -10922,6 +10954,7 @@ class TrainingJobSummary(Base):
     training_job_status: str
     training_end_time: Optional[datetime.datetime] = Unassigned()
     last_modified_time: Optional[datetime.datetime] = Unassigned()
+    secondary_status: Optional[str] = Unassigned()
     warm_pool_status: Optional[WarmPoolStatus] = Unassigned()
 
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.12
+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.12.dist-info → sagemaker_core-1.0.14.dist-info}/RECORD

@@ -7,15 +7,15 @@ 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=bhMPIF2D-8ZpP-m2HdebHj7EQcqzq9Iz0LuNSUiR8dE,698010
+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
 sagemaker_core/main/code_injection/base.py,sha256=11_Jif0nOzfbLGlXaacKf-wcizzfS64U0OSZGoVffFU,1733
 sagemaker_core/main/code_injection/codec.py,sha256=2DjmeD2uND307UqDefvVEpE0rZ8yfFU3Bi3TvQCQveI,7658
 sagemaker_core/main/code_injection/constants.py,sha256=2ICExGge8vAWx7lSTW0JGh-bH1korkvpOpDu5M63eI4,980
-sagemaker_core/main/code_injection/shape_dag.py,sha256=Kmv27Znp4AKYoocm7urZAW2OFJboTJd2jA5FigiFM8w,659419
+sagemaker_core/main/code_injection/shape_dag.py,sha256=Akn-5sH0T8ujXuiXEOJtq5agJuRk-tKoFP04qwFx-NM,660663
 sagemaker_core/resources/__init__.py,sha256=EAYTFMN-nPjnPjjBbhIUeaL67FLKNPd7qbcbl9VIrws,31
 sagemaker_core/shapes/__init__.py,sha256=RnbIu9eTxKt-DNsOFJabrWIgrrtS9_SdAozP9JBl_ic,28
 sagemaker_core/tools/__init__.py,sha256=xX79JImxCVzrWMnjgntLCve2G5I-R4pRar5s20kT9Rs,56
@@ -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.12.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.12.dist-info/METADATA,sha256=D3PfM1phfSdsCn28CHTR9Flm4Yo8Ano6d6XnHYuVJbs,4878
-sagemaker_core-1.0.12.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-sagemaker_core-1.0.12.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.12.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.12.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