sagemaker-core 1.0.11__py3-none-any.whl → 1.0.13__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",
@@ -2456,6 +2484,7 @@ SHAPE_DAG = {
             {"name": "SourceUri", "shape": "ModelPackageSourceUri", "type": "string"},
             {"name": "SecurityConfig", "shape": "ModelPackageSecurityConfig", "type": "structure"},
             {"name": "ModelCard", "shape": "ModelPackageModelCard", "type": "structure"},
+            {"name": "ModelLifeCycle", "shape": "ModelLifeCycle", "type": "structure"},
         ],
         "type": "structure",
     },
@@ -4969,6 +4998,7 @@ SHAPE_DAG = {
             {"name": "SourceUri", "shape": "ModelPackageSourceUri", "type": "string"},
             {"name": "SecurityConfig", "shape": "ModelPackageSecurityConfig", "type": "structure"},
             {"name": "ModelCard", "shape": "ModelPackageModelCard", "type": "structure"},
+            {"name": "ModelLifeCycle", "shape": "ModelLifeCycle", "type": "structure"},
         ],
         "type": "structure",
     },
@@ -10102,6 +10132,14 @@ SHAPE_DAG = {
         "member_type": "structure",
         "type": "list",
     },
+    "ModelLifeCycle": {
+        "members": [
+            {"name": "Stage", "shape": "EntityName", "type": "string"},
+            {"name": "StageStatus", "shape": "EntityName", "type": "string"},
+            {"name": "StageDescription", "shape": "StageDescription", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "ModelMetadataFilter": {
         "members": [
             {"name": "Name", "shape": "ModelMetadataFilterType", "type": "string"},
@@ -10194,6 +10232,7 @@ SHAPE_DAG = {
             {"name": "SourceUri", "shape": "ModelPackageSourceUri", "type": "string"},
             {"name": "SecurityConfig", "shape": "ModelPackageSecurityConfig", "type": "structure"},
             {"name": "ModelCard", "shape": "ModelPackageModelCard", "type": "structure"},
+            {"name": "ModelLifeCycle", "shape": "ModelLifeCycle", "type": "structure"},
             {"name": "Tags", "shape": "TagList", "type": "list"},
             {"name": "CustomerMetadataProperties", "shape": "CustomerMetadataMap", "type": "map"},
             {"name": "DriftCheckBaselines", "shape": "DriftCheckBaselines", "type": "structure"},
@@ -13555,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",
@@ -14337,6 +14377,8 @@ SHAPE_DAG = {
             },
             {"name": "SourceUri", "shape": "ModelPackageSourceUri", "type": "string"},
             {"name": "ModelCard", "shape": "ModelPackageModelCard", "type": "structure"},
+            {"name": "ModelLifeCycle", "shape": "ModelLifeCycle", "type": "structure"},
+            {"name": "ClientToken", "shape": "ClientToken", "type": "string"},
         ],
         "type": "structure",
     },

sagemaker_core/main/resources.py

@@ -18544,6 +18544,7 @@ class ModelPackage(Base):
         source_uri: The URI of the source for the model package.
         security_config: The KMS Key ID (KMSKeyId) used for encryption of model package information.
         model_card: The model card associated with the model package. Since ModelPackageModelCard is tied to a model package, it is a specific usage of a model card and its schema is simplified compared to the schema of ModelCard. The ModelPackageModelCard schema does not include model_package_details, and model_overview is composed of the model_creator and model_artifact properties. For more information about the model package model card schema, see Model package model card schema. For more information about the model card associated with the model package, see View the Details of a Model Version.
+        model_life_cycle:  A structure describing the current state of the model in its life cycle.
 
     """
 
@@ -18578,6 +18579,7 @@ class ModelPackage(Base):
     source_uri: Optional[str] = Unassigned()
     security_config: Optional[ModelPackageSecurityConfig] = Unassigned()
     model_card: Optional[ModelPackageModelCard] = Unassigned()
+    model_life_cycle: Optional[ModelLifeCycle] = Unassigned()
 
     def get_name(self) -> str:
         attributes = vars(self)
@@ -18675,6 +18677,7 @@ class ModelPackage(Base):
         source_uri: Optional[str] = Unassigned(),
         security_config: Optional[ModelPackageSecurityConfig] = Unassigned(),
         model_card: Optional[ModelPackageModelCard] = Unassigned(),
+        model_life_cycle: Optional[ModelLifeCycle] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional["ModelPackage"]:
@@ -18704,6 +18707,7 @@ class ModelPackage(Base):
             source_uri: The URI of the source for the model package. If you want to clone a model package, set it to the model package Amazon Resource Name (ARN). If you want to register a model, set it to the model ARN.
             security_config: The KMS Key ID (KMSKeyId) used for encryption of model package information.
             model_card: The model card associated with the model package. Since ModelPackageModelCard is tied to a model package, it is a specific usage of a model card and its schema is simplified compared to the schema of ModelCard. The ModelPackageModelCard schema does not include model_package_details, and model_overview is composed of the model_creator and model_artifact properties. For more information about the model package model card schema, see Model package model card schema. For more information about the model card associated with the model package, see View the Details of a Model Version.
+            model_life_cycle:  A structure describing the current state of the model in its life cycle.
             session: Boto3 session.
             region: Region name.
 
@@ -18755,6 +18759,7 @@ class ModelPackage(Base):
             "SourceUri": source_uri,
             "SecurityConfig": security_config,
             "ModelCard": model_card,
+            "ModelLifeCycle": model_life_cycle,
         }
 
         operation_input_args = Base.populate_chained_attributes(
@@ -18874,6 +18879,8 @@ class ModelPackage(Base):
         inference_specification: Optional[InferenceSpecification] = Unassigned(),
         source_uri: Optional[str] = Unassigned(),
         model_card: Optional[ModelPackageModelCard] = Unassigned(),
+        model_life_cycle: Optional[ModelLifeCycle] = Unassigned(),
+        client_token: Optional[str] = Unassigned(),
     ) -> Optional["ModelPackage"]:
         """
         Update a ModelPackage resource
@@ -18881,6 +18888,7 @@ class ModelPackage(Base):
         Parameters:
             customer_metadata_properties_to_remove: The metadata properties associated with the model package versions to remove.
             additional_inference_specifications_to_add: An array of additional Inference Specification objects to be added to the existing array additional Inference Specification. Total number of additional Inference Specifications can not exceed 15. Each additional Inference Specification specifies artifacts based on this model package that can be used on inference endpoints. Generally used with SageMaker Neo to store the compiled artifacts.
+            client_token:  A unique token that guarantees that the call to this API is idempotent.
 
         Returns:
             The ModelPackage resource.
@@ -18911,6 +18919,8 @@ class ModelPackage(Base):
             "InferenceSpecification": inference_specification,
             "SourceUri": source_uri,
             "ModelCard": model_card,
+            "ModelLifeCycle": model_life_cycle,
+            "ClientToken": client_token,
         }
         logger.debug(f"Input request: {operation_input_args}")
         # serialize the input request

sagemaker_core/main/shapes.py

@@ -2360,6 +2360,23 @@ 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 BatchDescribeModelPackageError(Base):
     """
     BatchDescribeModelPackageError
@@ -6292,6 +6309,23 @@ class ModelPackageModelCard(Base):
     model_card_status: Optional[str] = Unassigned()
 
 
+class ModelLifeCycle(Base):
+    """
+    ModelLifeCycle
+       A structure describing the current state of the model in its life cycle.
+
+    Attributes
+    ----------------------
+    stage:  The current stage in the model life cycle.
+    stage_status:  The current status of a stage in model life cycle.
+    stage_description:  Describes the stage related details.
+    """
+
+    stage: str
+    stage_status: str
+    stage_description: Optional[str] = Unassigned()
+
+
 class ModelQualityBaselineConfig(Base):
     """
     ModelQualityBaselineConfig
@@ -10896,6 +10930,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.
     """
 
@@ -10905,6 +10940,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()
 
 
@@ -11282,6 +11318,7 @@ class ModelPackage(Base):
     source_uri: The URI of the source for the model package.
     security_config
     model_card
+    model_life_cycle:  A structure describing the current state of the model in its life cycle.
     tags: A list of the tags associated with the model package. For more information, see Tagging Amazon Web Services resources in the Amazon Web Services General Reference Guide.
     customer_metadata_properties: The metadata properties for the model package.
     drift_check_baselines: Represents the drift check baselines that can be used when the model monitor is set using the model package.
@@ -11316,6 +11353,7 @@ class ModelPackage(Base):
     source_uri: Optional[str] = Unassigned()
     security_config: Optional[ModelPackageSecurityConfig] = Unassigned()
     model_card: Optional[ModelPackageModelCard] = Unassigned()
+    model_life_cycle: Optional[ModelLifeCycle] = Unassigned()
     tags: Optional[List[Tag]] = Unassigned()
     customer_metadata_properties: Optional[Dict[str, str]] = Unassigned()
     drift_check_baselines: Optional[DriftCheckBaselines] = Unassigned()

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.11
+Version: 1.0.13
 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.11.dist-info → sagemaker_core-1.0.13.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=db-1JPFeV7jlM0xnl_dGEEp_RvNpzrrbnZyk5LtYQC8,1321185
-sagemaker_core/main/shapes.py,sha256=Nfp0a9wTjJBP403505ck9rS6pMF8P9Kpq6PjShQOK1w,697388
+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/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=FyU_a0Jh3AVxTNVN2kjt0oHqqyHNjAgIy3G7kzqksm4,658657
+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.11.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.11.dist-info/METADATA,sha256=81rMU_96k8Ki06GRNPUpAE4pn-MyvHAj8ej--nix5jU,4878
-sagemaker_core-1.0.11.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-sagemaker_core-1.0.11.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.11.dist-info/RECORD,,
+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,,