sagemaker-core 1.0.12__tar.gz → 1.0.14__tar.gz

{sagemaker_core-1.0.12/src/sagemaker_core.egg-info → sagemaker_core-1.0.14}/PKG-INFO

@@ -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.14/VERSION

@@ -0,0 +1 @@
+1.0.14

{sagemaker_core-1.0.12 → sagemaker_core-1.0.14}/src/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-1.0.12 → sagemaker_core-1.0.14}/src/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-1.0.12 → sagemaker_core-1.0.14}/src/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 → sagemaker_core-1.0.14/src/sagemaker_core.egg-info}/PKG-INFO

@@ -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/VERSION

@@ -1 +0,0 @@
-1.0.12