sagemaker-core 1.0.23__tar.gz → 1.0.25__tar.gz

{sagemaker_core-1.0.23/src/sagemaker_core.egg-info → sagemaker_core-1.0.25}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: sagemaker-core
-Version: 1.0.23
+Version: 1.0.25
 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.25/VERSION

@@ -0,0 +1 @@
+1.0.25

{sagemaker_core-1.0.23 → sagemaker_core-1.0.25}/src/sagemaker_core/main/code_injection/shape_dag.py

@@ -1304,6 +1304,11 @@ SHAPE_DAG = {
         "member_type": "structure",
         "type": "list",
     },
+    "ClusterInstanceGroupsToDelete": {
+        "member_shape": "ClusterInstanceGroupName",
+        "member_type": "string",
+        "type": "list",
+    },
     "ClusterInstancePlacement": {
         "members": [
             {"name": "AvailabilityZone", "shape": "ClusterAvailabilityZone", "type": "string"},
@@ -4906,6 +4911,11 @@ SHAPE_DAG = {
                 "shape": "InferenceComponentStatus",
                 "type": "string",
             },
+            {
+                "name": "LastDeploymentConfig",
+                "shape": "InferenceComponentDeploymentConfig",
+                "type": "structure",
+            },
         ],
         "type": "structure",
     },
@@ -7665,6 +7675,13 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "InferenceComponentCapacitySize": {
+        "members": [
+            {"name": "Type", "shape": "InferenceComponentCapacitySizeType", "type": "string"},
+            {"name": "Value", "shape": "CapacitySizeValue", "type": "integer"},
+        ],
+        "type": "structure",
+    },
     "InferenceComponentComputeResourceRequirements": {
         "members": [
             {"name": "NumberOfCpuCoresRequired", "shape": "NumberOfCpuCores", "type": "float"},
@@ -7694,6 +7711,42 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "InferenceComponentDeploymentConfig": {
+        "members": [
+            {
+                "name": "RollingUpdatePolicy",
+                "shape": "InferenceComponentRollingUpdatePolicy",
+                "type": "structure",
+            },
+            {
+                "name": "AutoRollbackConfiguration",
+                "shape": "AutoRollbackConfig",
+                "type": "structure",
+            },
+        ],
+        "type": "structure",
+    },
+    "InferenceComponentRollingUpdatePolicy": {
+        "members": [
+            {
+                "name": "MaximumBatchSize",
+                "shape": "InferenceComponentCapacitySize",
+                "type": "structure",
+            },
+            {"name": "WaitIntervalInSeconds", "shape": "WaitIntervalInSeconds", "type": "integer"},
+            {
+                "name": "MaximumExecutionTimeoutInSeconds",
+                "shape": "MaximumExecutionTimeoutInSeconds",
+                "type": "integer",
+            },
+            {
+                "name": "RollbackMaximumBatchSize",
+                "shape": "InferenceComponentCapacitySize",
+                "type": "structure",
+            },
+        ],
+        "type": "structure",
+    },
     "InferenceComponentRuntimeConfig": {
         "members": [
             {"name": "CopyCount", "shape": "InferenceComponentCopyCount", "type": "integer"}
@@ -14737,6 +14790,11 @@ SHAPE_DAG = {
                 "type": "list",
             },
             {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
+            {
+                "name": "InstanceGroupsToDelete",
+                "shape": "ClusterInstanceGroupsToDelete",
+                "type": "list",
+            },
         ],
         "type": "structure",
     },
@@ -14994,6 +15052,11 @@ SHAPE_DAG = {
                 "shape": "InferenceComponentRuntimeConfig",
                 "type": "structure",
             },
+            {
+                "name": "DeploymentConfig",
+                "shape": "InferenceComponentDeploymentConfig",
+                "type": "structure",
+            },
         ],
         "type": "structure",
     },

{sagemaker_core-1.0.23 → sagemaker_core-1.0.25}/src/sagemaker_core/main/resources.py

@@ -3430,10 +3430,14 @@ class Cluster(Base):
         self,
         instance_groups: List[ClusterInstanceGroupSpecification],
         node_recovery: Optional[str] = Unassigned(),
+        instance_groups_to_delete: Optional[List[str]] = Unassigned(),
     ) -> Optional["Cluster"]:
         """
         Update a Cluster resource
 
+        Parameters:
+            instance_groups_to_delete: Specify the names of the instance groups to delete. Use a single , as the separator between multiple names.
+
         Returns:
             The Cluster resource.
 
@@ -3459,6 +3463,7 @@ class Cluster(Base):
             "ClusterName": self.cluster_name,
             "InstanceGroups": instance_groups,
             "NodeRecovery": node_recovery,
+            "InstanceGroupsToDelete": instance_groups_to_delete,
         }
         logger.debug(f"Input request: {operation_input_args}")
         # serialize the input request
@@ -3869,7 +3874,7 @@ class Cluster(Base):
         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.
+            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.   If you need to delete more than 99 instances, contact Support for assistance.
             session: Boto3 session.
             region: Region name.
 
@@ -14796,6 +14801,7 @@ class InferenceComponent(Base):
         specification: Details about the resources that are deployed with this inference component.
         runtime_config: Details about the runtime settings for the model that is deployed with the inference component.
         inference_component_status: The status of the inference component.
+        last_deployment_config: The deployment and rollback settings that you assigned to the inference component.
 
     """
 
@@ -14810,6 +14816,7 @@ class InferenceComponent(Base):
     creation_time: Optional[datetime.datetime] = Unassigned()
     last_modified_time: Optional[datetime.datetime] = Unassigned()
     inference_component_status: Optional[str] = Unassigned()
+    last_deployment_config: Optional[InferenceComponentDeploymentConfig] = Unassigned()
 
     def get_name(self) -> str:
         attributes = vars(self)
@@ -14994,10 +15001,14 @@ class InferenceComponent(Base):
         self,
         specification: Optional[InferenceComponentSpecification] = Unassigned(),
         runtime_config: Optional[InferenceComponentRuntimeConfig] = Unassigned(),
+        deployment_config: Optional[InferenceComponentDeploymentConfig] = Unassigned(),
     ) -> Optional["InferenceComponent"]:
         """
         Update a InferenceComponent resource
 
+        Parameters:
+            deployment_config: The deployment configuration for the inference component. The configuration contains the desired deployment strategy and rollback settings.
+
         Returns:
             The InferenceComponent resource.
 
@@ -15021,6 +15032,7 @@ class InferenceComponent(Base):
             "InferenceComponentName": self.inference_component_name,
             "Specification": specification,
             "RuntimeConfig": runtime_config,
+            "DeploymentConfig": deployment_config,
         }
         logger.debug(f"Input request: {operation_input_args}")
         # serialize the input request

{sagemaker_core-1.0.23 → sagemaker_core-1.0.25}/src/sagemaker_core/main/shapes.py

@@ -8404,6 +8404,55 @@ class InferenceComponentRuntimeConfigSummary(Base):
     current_copy_count: Optional[int] = Unassigned()
 
 
+class InferenceComponentCapacitySize(Base):
+    """
+    InferenceComponentCapacitySize
+      Specifies the type and size of the endpoint capacity to activate for a rolling deployment or a rollback strategy. You can specify your batches as either of the following:   A count of inference component copies    The overall percentage or your fleet    For a rollback strategy, if you don't specify the fields in this object, or if you set the Value parameter to 100%, then SageMaker AI uses a blue/green rollback strategy and rolls all traffic back to the blue fleet.
+
+    Attributes
+    ----------------------
+    type: Specifies the endpoint capacity type.  COPY_COUNT  The endpoint activates based on the number of inference component copies.  CAPACITY_PERCENT  The endpoint activates based on the specified percentage of capacity.
+    value: Defines the capacity size, either as a number of inference component copies or a capacity percentage.
+    """
+
+    type: str
+    value: int
+
+
+class InferenceComponentRollingUpdatePolicy(Base):
+    """
+    InferenceComponentRollingUpdatePolicy
+      Specifies a rolling deployment strategy for updating a SageMaker AI inference component.
+
+    Attributes
+    ----------------------
+    maximum_batch_size: The batch size for each rolling step in the deployment process. For each step, SageMaker AI provisions capacity on the new endpoint fleet, routes traffic to that fleet, and terminates capacity on the old endpoint fleet. The value must be between 5% to 50% of the copy count of the inference component.
+    wait_interval_in_seconds: The length of the baking period, during which SageMaker AI monitors alarms for each batch on the new fleet.
+    maximum_execution_timeout_in_seconds: The time limit for the total deployment. Exceeding this limit causes a timeout.
+    rollback_maximum_batch_size: The batch size for a rollback to the old endpoint fleet. If this field is absent, the value is set to the default, which is 100% of the total capacity. When the default is used, SageMaker AI provisions the entire capacity of the old fleet at once during rollback.
+    """
+
+    maximum_batch_size: InferenceComponentCapacitySize
+    wait_interval_in_seconds: int
+    maximum_execution_timeout_in_seconds: Optional[int] = Unassigned()
+    rollback_maximum_batch_size: Optional[InferenceComponentCapacitySize] = Unassigned()
+
+
+class InferenceComponentDeploymentConfig(Base):
+    """
+    InferenceComponentDeploymentConfig
+      The deployment configuration for an endpoint that hosts inference components. The configuration includes the desired deployment strategy and rollback settings.
+
+    Attributes
+    ----------------------
+    rolling_update_policy: Specifies a rolling deployment strategy for updating a SageMaker AI endpoint.
+    auto_rollback_configuration
+    """
+
+    rolling_update_policy: InferenceComponentRollingUpdatePolicy
+    auto_rollback_configuration: Optional[AutoRollbackConfig] = Unassigned()
+
+
 class EndpointMetadata(Base):
     """
     EndpointMetadata
@@ -9638,7 +9687,7 @@ class EndpointSummary(Base):
 class Experiment(Base):
     """
     Experiment
-      The properties of an experiment as returned by the Search API.
+      The properties of an experiment as returned by the Search API. For information about experiments, see the CreateExperiment API.
 
     Attributes
     ----------------------
@@ -11690,11 +11739,11 @@ class ModelDashboardModel(Base):
 class ModelPackage(Base):
     """
     ModelPackage
-      A versioned model that can be deployed for SageMaker inference.
+      A container for your trained model that can be deployed for SageMaker inference. This can include inference code, artifacts, and metadata. The model package type can be one of the following.   Versioned model: A part of a model package group in Model Registry.   Unversioned model: Not part of a model package group and used in Amazon Web Services Marketplace.   For more information, see  CreateModelPackage .
 
     Attributes
     ----------------------
-    model_package_name: The name of the model.
+    model_package_name: The name of the model package. The name can be as follows:   For a versioned model, the name is automatically generated by SageMaker Model Registry and follows the format 'ModelPackageGroupName/ModelPackageVersion'.   For an unversioned model, you must provide the name.
     model_package_group_name: The model group to which the model belongs.
     model_package_version: The version number of a versioned model.
     model_package_arn: The Amazon Resource Name (ARN) of the model package.
@@ -11765,7 +11814,7 @@ class ModelPackage(Base):
 class ModelPackageGroup(Base):
     """
     ModelPackageGroup
-      A group of versioned models in the model registry.
+      A group of versioned models in the Model Registry.
 
     Attributes
     ----------------------

{sagemaker_core-1.0.23 → sagemaker_core-1.0.25/src/sagemaker_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: sagemaker-core
-Version: 1.0.23
+Version: 1.0.25
 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.23/VERSION

@@ -1 +0,0 @@
-1.0.23