aws-sdk-sagemaker 1.327.0 → 1.328.0

data/lib/aws-sdk-sagemaker/client_api.rb

@@ -320,6 +320,7 @@ module Aws::SageMaker
     ClusterAutoScalingStatus = Shapes::StringShape.new(name: 'ClusterAutoScalingStatus')
     ClusterAvailabilityZone = Shapes::StringShape.new(name: 'ClusterAvailabilityZone')
     ClusterAvailabilityZoneId = Shapes::StringShape.new(name: 'ClusterAvailabilityZoneId')
+    ClusterConfigMode = Shapes::StringShape.new(name: 'ClusterConfigMode')
     ClusterEbsVolumeConfig = Shapes::StructureShape.new(name: 'ClusterEbsVolumeConfig')
     ClusterEbsVolumeSizeInGB = Shapes::IntegerShape.new(name: 'ClusterEbsVolumeSizeInGB')
     ClusterEventDetail = Shapes::StructureShape.new(name: 'ClusterEventDetail')
@@ -334,6 +335,7 @@ module Aws::SageMaker
     ClusterInstanceGroupSpecification = Shapes::StructureShape.new(name: 'ClusterInstanceGroupSpecification')
     ClusterInstanceGroupSpecifications = Shapes::ListShape.new(name: 'ClusterInstanceGroupSpecifications')
     ClusterInstanceGroupsToDelete = Shapes::ListShape.new(name: 'ClusterInstanceGroupsToDelete')
+    ClusterInstanceMemoryAllocationPercentage = Shapes::IntegerShape.new(name: 'ClusterInstanceMemoryAllocationPercentage')
     ClusterInstancePlacement = Shapes::StructureShape.new(name: 'ClusterInstancePlacement')
     ClusterInstanceStatus = Shapes::StringShape.new(name: 'ClusterInstanceStatus')
     ClusterInstanceStatusDetails = Shapes::StructureShape.new(name: 'ClusterInstanceStatusDetails')
@@ -374,6 +376,7 @@ module Aws::SageMaker
     ClusterSummaries = Shapes::ListShape.new(name: 'ClusterSummaries')
     ClusterSummary = Shapes::StructureShape.new(name: 'ClusterSummary')
     ClusterThreadsPerCore = Shapes::IntegerShape.new(name: 'ClusterThreadsPerCore')
+    ClusterTieredStorageConfig = Shapes::StructureShape.new(name: 'ClusterTieredStorageConfig')
     CodeEditorAppImageConfig = Shapes::StructureShape.new(name: 'CodeEditorAppImageConfig')
     CodeEditorAppSettings = Shapes::StructureShape.new(name: 'CodeEditorAppSettings')
     CodeRepositories = Shapes::ListShape.new(name: 'CodeRepositories')
@@ -1199,6 +1202,7 @@ module Aws::SageMaker
     HyperbandStrategyConfig = Shapes::StructureShape.new(name: 'HyperbandStrategyConfig')
     HyperbandStrategyMaxResource = Shapes::IntegerShape.new(name: 'HyperbandStrategyMaxResource')
     HyperbandStrategyMinResource = Shapes::IntegerShape.new(name: 'HyperbandStrategyMinResource')
+    IPAddressType = Shapes::StringShape.new(name: 'IPAddressType')
     IamIdentity = Shapes::StructureShape.new(name: 'IamIdentity')
     IamPolicyConstraints = Shapes::StructureShape.new(name: 'IamPolicyConstraints')
     IdempotencyToken = Shapes::StringShape.new(name: 'IdempotencyToken')
@@ -3593,6 +3597,10 @@ module Aws::SageMaker
     ClusterSummary.add_member(:training_plan_arns, Shapes::ShapeRef.new(shape: TrainingPlanArns, location_name: "TrainingPlanArns"))
     ClusterSummary.struct_class = Types::ClusterSummary
 
+    ClusterTieredStorageConfig.add_member(:mode, Shapes::ShapeRef.new(shape: ClusterConfigMode, required: true, location_name: "Mode"))
+    ClusterTieredStorageConfig.add_member(:instance_memory_allocation_percentage, Shapes::ShapeRef.new(shape: ClusterInstanceMemoryAllocationPercentage, location_name: "InstanceMemoryAllocationPercentage"))
+    ClusterTieredStorageConfig.struct_class = Types::ClusterTieredStorageConfig
+
     CodeEditorAppImageConfig.add_member(:file_system_config, Shapes::ShapeRef.new(shape: FileSystemConfig, location_name: "FileSystemConfig"))
     CodeEditorAppImageConfig.add_member(:container_config, Shapes::ShapeRef.new(shape: ContainerConfig, location_name: "ContainerConfig"))
     CodeEditorAppImageConfig.struct_class = Types::CodeEditorAppImageConfig
@@ -3853,6 +3861,7 @@ module Aws::SageMaker
     CreateClusterRequest.add_member(:tags, Shapes::ShapeRef.new(shape: TagList, location_name: "Tags"))
     CreateClusterRequest.add_member(:orchestrator, Shapes::ShapeRef.new(shape: ClusterOrchestrator, location_name: "Orchestrator"))
     CreateClusterRequest.add_member(:node_recovery, Shapes::ShapeRef.new(shape: ClusterNodeRecovery, location_name: "NodeRecovery"))
+    CreateClusterRequest.add_member(:tiered_storage_config, Shapes::ShapeRef.new(shape: ClusterTieredStorageConfig, location_name: "TieredStorageConfig"))
     CreateClusterRequest.add_member(:node_provisioning_mode, Shapes::ShapeRef.new(shape: ClusterNodeProvisioningMode, location_name: "NodeProvisioningMode"))
     CreateClusterRequest.add_member(:cluster_role, Shapes::ShapeRef.new(shape: RoleArn, location_name: "ClusterRole"))
     CreateClusterRequest.add_member(:auto_scaling, Shapes::ShapeRef.new(shape: ClusterAutoScalingConfig, location_name: "AutoScaling"))
@@ -4319,6 +4328,7 @@ module Aws::SageMaker
     CreateNotebookInstanceInput.add_member(:instance_type, Shapes::ShapeRef.new(shape: InstanceType, required: true, location_name: "InstanceType"))
     CreateNotebookInstanceInput.add_member(:subnet_id, Shapes::ShapeRef.new(shape: SubnetId, location_name: "SubnetId"))
     CreateNotebookInstanceInput.add_member(:security_group_ids, Shapes::ShapeRef.new(shape: SecurityGroupIds, location_name: "SecurityGroupIds"))
+    CreateNotebookInstanceInput.add_member(:ip_address_type, Shapes::ShapeRef.new(shape: IPAddressType, location_name: "IpAddressType"))
     CreateNotebookInstanceInput.add_member(:role_arn, Shapes::ShapeRef.new(shape: RoleArn, required: true, location_name: "RoleArn"))
     CreateNotebookInstanceInput.add_member(:kms_key_id, Shapes::ShapeRef.new(shape: KmsKeyId, location_name: "KmsKeyId"))
     CreateNotebookInstanceInput.add_member(:tags, Shapes::ShapeRef.new(shape: TagList, location_name: "Tags"))
@@ -5192,6 +5202,7 @@ module Aws::SageMaker
     DescribeClusterResponse.add_member(:restricted_instance_groups, Shapes::ShapeRef.new(shape: ClusterRestrictedInstanceGroupDetailsList, location_name: "RestrictedInstanceGroups"))
     DescribeClusterResponse.add_member(:vpc_config, Shapes::ShapeRef.new(shape: VpcConfig, location_name: "VpcConfig"))
     DescribeClusterResponse.add_member(:orchestrator, Shapes::ShapeRef.new(shape: ClusterOrchestrator, location_name: "Orchestrator"))
+    DescribeClusterResponse.add_member(:tiered_storage_config, Shapes::ShapeRef.new(shape: ClusterTieredStorageConfig, location_name: "TieredStorageConfig"))
     DescribeClusterResponse.add_member(:node_recovery, Shapes::ShapeRef.new(shape: ClusterNodeRecovery, location_name: "NodeRecovery"))
     DescribeClusterResponse.add_member(:node_provisioning_mode, Shapes::ShapeRef.new(shape: ClusterNodeProvisioningMode, location_name: "NodeProvisioningMode"))
     DescribeClusterResponse.add_member(:cluster_role, Shapes::ShapeRef.new(shape: RoleArn, location_name: "ClusterRole"))
@@ -5906,6 +5917,7 @@ module Aws::SageMaker
     DescribeNotebookInstanceOutput.add_member(:failure_reason, Shapes::ShapeRef.new(shape: FailureReason, location_name: "FailureReason"))
     DescribeNotebookInstanceOutput.add_member(:url, Shapes::ShapeRef.new(shape: NotebookInstanceUrl, location_name: "Url"))
     DescribeNotebookInstanceOutput.add_member(:instance_type, Shapes::ShapeRef.new(shape: InstanceType, location_name: "InstanceType"))
+    DescribeNotebookInstanceOutput.add_member(:ip_address_type, Shapes::ShapeRef.new(shape: IPAddressType, location_name: "IpAddressType"))
     DescribeNotebookInstanceOutput.add_member(:subnet_id, Shapes::ShapeRef.new(shape: SubnetId, location_name: "SubnetId"))
     DescribeNotebookInstanceOutput.add_member(:security_groups, Shapes::ShapeRef.new(shape: SecurityGroupIds, location_name: "SecurityGroups"))
     DescribeNotebookInstanceOutput.add_member(:role_arn, Shapes::ShapeRef.new(shape: RoleArn, location_name: "RoleArn"))
@@ -11304,6 +11316,7 @@ module Aws::SageMaker
     UpdateClusterRequest.add_member(:cluster_name, Shapes::ShapeRef.new(shape: ClusterNameOrArn, required: true, location_name: "ClusterName"))
     UpdateClusterRequest.add_member(:instance_groups, Shapes::ShapeRef.new(shape: ClusterInstanceGroupSpecifications, location_name: "InstanceGroups"))
     UpdateClusterRequest.add_member(:restricted_instance_groups, Shapes::ShapeRef.new(shape: ClusterRestrictedInstanceGroupSpecifications, location_name: "RestrictedInstanceGroups"))
+    UpdateClusterRequest.add_member(:tiered_storage_config, Shapes::ShapeRef.new(shape: ClusterTieredStorageConfig, location_name: "TieredStorageConfig"))
     UpdateClusterRequest.add_member(:node_recovery, Shapes::ShapeRef.new(shape: ClusterNodeRecovery, location_name: "NodeRecovery"))
     UpdateClusterRequest.add_member(:instance_groups_to_delete, Shapes::ShapeRef.new(shape: ClusterInstanceGroupsToDelete, location_name: "InstanceGroupsToDelete"))
     UpdateClusterRequest.add_member(:cluster_role, Shapes::ShapeRef.new(shape: RoleArn, location_name: "ClusterRole"))
@@ -11572,6 +11585,7 @@ module Aws::SageMaker
 
     UpdateNotebookInstanceInput.add_member(:notebook_instance_name, Shapes::ShapeRef.new(shape: NotebookInstanceName, required: true, location_name: "NotebookInstanceName"))
     UpdateNotebookInstanceInput.add_member(:instance_type, Shapes::ShapeRef.new(shape: InstanceType, location_name: "InstanceType"))
+    UpdateNotebookInstanceInput.add_member(:ip_address_type, Shapes::ShapeRef.new(shape: IPAddressType, location_name: "IpAddressType"))
     UpdateNotebookInstanceInput.add_member(:role_arn, Shapes::ShapeRef.new(shape: RoleArn, location_name: "RoleArn"))
     UpdateNotebookInstanceInput.add_member(:lifecycle_config_name, Shapes::ShapeRef.new(shape: NotebookInstanceLifecycleConfigName, location_name: "LifecycleConfigName"))
     UpdateNotebookInstanceInput.add_member(:disassociate_lifecycle_config, Shapes::ShapeRef.new(shape: DisassociateNotebookInstanceLifecycleConfig, location_name: "DisassociateLifecycleConfig", metadata: {"box" => true}))

data/lib/aws-sdk-sagemaker/types.rb

@@ -5848,6 +5848,36 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Defines the configuration for managed tier checkpointing in a HyperPod
+    # cluster. Managed tier checkpointing uses multiple storage tiers,
+    # including cluster CPU memory, to provide faster checkpoint operations
+    # and improved fault tolerance for large-scale model training. The
+    # system automatically saves checkpoints at high frequency to memory and
+    # periodically persists them to durable storage, like Amazon S3.
+    #
+    # @!attribute [rw] mode
+    #   Specifies whether managed tier checkpointing is enabled or disabled
+    #   for the HyperPod cluster. When set to `Enable`, the system installs
+    #   a memory management daemon that provides disaggregated memory as a
+    #   service for checkpoint storage. When set to `Disable`, the feature
+    #   is turned off and the memory management daemon is removed from the
+    #   cluster.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_memory_allocation_percentage
+    #   The percentage (int) of cluster memory to allocate for
+    #   checkpointing.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterTieredStorageConfig AWS API Documentation
+    #
+    class ClusterTieredStorageConfig < Struct.new(
+      :mode,
+      :instance_memory_allocation_percentage)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The configuration for the file system and kernels in a SageMaker image
     # running as a Code Editor app. The `FileSystemConfig` object is not
     # supported.
@@ -7366,6 +7396,13 @@ module Aws::SageMaker
     #   cluster instances.
     #   @return [String]
     #
+    # @!attribute [rw] tiered_storage_config
+    #   The configuration for managed tier checkpointing on the HyperPod
+    #   cluster. When enabled, this feature uses a multi-tier storage
+    #   approach for storing model checkpoints, providing faster checkpoint
+    #   operations and improved fault tolerance across cluster nodes.
+    #   @return [Types::ClusterTieredStorageConfig]
+    #
     # @!attribute [rw] node_provisioning_mode
     #   The mode for provisioning nodes in the cluster. You can specify the
     #   following modes:
@@ -7406,6 +7443,7 @@ module Aws::SageMaker
       :tags,
       :orchestrator,
       :node_recovery,
+      :tiered_storage_config,
       :node_provisioning_mode,
       :cluster_role,
       :auto_scaling)
@@ -9423,13 +9461,14 @@ module Aws::SageMaker
     #
     #   * The name can't end with "-metadata".
     #
-    #   * If you are using one of the following [built-in task types][1],
-    #     the attribute name *must* end with "-ref". If the task type you
-    #     are using is not listed below, the attribute name *must not* end
-    #     with "-ref".
+    #   * If you are using one of the [built-in task types][1] or one of the
+    #     following, the attribute name *must* end with "-ref".
     #
-    #     * Verification (`VerificationSemanticSegmentation`) labeling jobs
-    #       for this task type.
+    #     * Image semantic segmentation (`SemanticSegmentation)` and
+    #       adjustment (`AdjustmentSemanticSegmentation`) labeling jobs for
+    #       this task type. One exception is that verification
+    #       (`VerificationSemanticSegmentation`) *must not* end with
+    #       -"ref".
     #
     #     * Video frame object detection (`VideoObjectDetection`), and
     #       adjustment and verification (`AdjustmentVideoObjectDetection`)
@@ -10461,6 +10500,13 @@ module Aws::SageMaker
     #   groups must be for the same VPC as specified in the subnet.
     #   @return [Array<String>]
     #
+    # @!attribute [rw] ip_address_type
+    #   The IP address type for the notebook instance. Specify `ipv4` for
+    #   IPv4-only connectivity or `dualstack` for both IPv4 and IPv6
+    #   connectivity. When you specify `dualstack`, the subnet must support
+    #   IPv6 CIDR blocks. If not specified, defaults to `ipv4`.
+    #   @return [String]
+    #
     # @!attribute [rw] role_arn
     #   When you send any requests to Amazon Web Services resources from the
     #   notebook instance, SageMaker AI assumes this role to perform tasks
@@ -10603,6 +10649,7 @@ module Aws::SageMaker
       :instance_type,
       :subnet_id,
       :security_group_ids,
+      :ip_address_type,
       :role_arn,
       :kms_key_id,
       :tags,
@@ -15099,6 +15146,13 @@ module Aws::SageMaker
     #   The type of orchestrator used for the SageMaker HyperPod cluster.
     #   @return [Types::ClusterOrchestrator]
     #
+    # @!attribute [rw] tiered_storage_config
+    #   The current configuration for managed tier checkpointing on the
+    #   HyperPod cluster. For example, this shows whether the feature is
+    #   enabled and the percentage of cluster memory allocated for
+    #   checkpoint storage.
+    #   @return [Types::ClusterTieredStorageConfig]
+    #
     # @!attribute [rw] node_recovery
     #   The node recovery mode configured for the SageMaker HyperPod
     #   cluster.
@@ -15129,6 +15183,7 @@ module Aws::SageMaker
       :restricted_instance_groups,
       :vpc_config,
       :orchestrator,
+      :tiered_storage_config,
       :node_recovery,
       :node_provisioning_mode,
       :cluster_role,
@@ -19022,6 +19077,12 @@ module Aws::SageMaker
     #   The type of ML compute instance running on the notebook instance.
     #   @return [String]
     #
+    # @!attribute [rw] ip_address_type
+    #   The IP address type configured for the notebook instance. Returns
+    #   `ipv4` for IPv4-only connectivity or `dualstack` for both IPv4 and
+    #   IPv6 connectivity.
+    #   @return [String]
+    #
     # @!attribute [rw] subnet_id
     #   The ID of the VPC subnet.
     #   @return [String]
@@ -19155,6 +19216,7 @@ module Aws::SageMaker
       :failure_reason,
       :url,
       :instance_type,
+      :ip_address_type,
       :subnet_id,
       :security_groups,
       :role_arn,
@@ -19331,6 +19393,27 @@ module Aws::SageMaker
     #
     # @!attribute [rw] status
     #   The status of the SageMaker Partner AI App.
+    #
+    #   * Creating: SageMaker AI is creating the partner AI app. The partner
+    #     AI app is not available during creation.
+    #
+    #   * Updating: SageMaker AI is updating the partner AI app. The partner
+    #     AI app is not available when updating.
+    #
+    #   * Deleting: SageMaker AI is deleting the partner AI app. The partner
+    #     AI app is not available during deletion.
+    #
+    #   * Available: The partner AI app is provisioned and accessible.
+    #
+    #   * Failed: The partner AI app is in a failed state and isn't
+    #     available. SageMaker AI is investigating the issue. For further
+    #     guidance, contact Amazon Web Services Support.
+    #
+    #   * UpdateFailed: The partner AI app couldn't be updated but is
+    #     available.
+    #
+    #   * Deleted: The partner AI app is permanently deleted and not
+    #     available.
     #   @return [String]
     #
     # @!attribute [rw] creation_time
@@ -21565,8 +21648,7 @@ module Aws::SageMaker
     #   @return [Array<String>]
     #
     # @!attribute [rw] rootless_docker
-    #   Indicates whether to use rootless Docker. Default value is
-    #   `DISABLED`.
+    #   Indicates whether to use rootless Docker.
     #   @return [String]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DockerSettings AWS API Documentation
@@ -50623,6 +50705,13 @@ module Aws::SageMaker
     #   to be created in the SageMaker HyperPod cluster.
     #   @return [Array<Types::ClusterRestrictedInstanceGroupSpecification>]
     #
+    # @!attribute [rw] tiered_storage_config
+    #   Updates the configuration for managed tier checkpointing on the
+    #   HyperPod cluster. For example, you can enable or disable the feature
+    #   and modify the percentage of cluster memory allocated for checkpoint
+    #   storage.
+    #   @return [Types::ClusterTieredStorageConfig]
+    #
     # @!attribute [rw] node_recovery
     #   The node recovery mode to be applied to the SageMaker HyperPod
     #   cluster.
@@ -50650,6 +50739,7 @@ module Aws::SageMaker
       :cluster_name,
       :instance_groups,
       :restricted_instance_groups,
+      :tiered_storage_config,
       :node_recovery,
       :instance_groups_to_delete,
       :cluster_role,
@@ -52029,6 +52119,14 @@ module Aws::SageMaker
     #   The Amazon ML compute instance type.
     #   @return [String]
     #
+    # @!attribute [rw] ip_address_type
+    #   The IP address type for the notebook instance. Specify `ipv4` for
+    #   IPv4-only connectivity or `dualstack` for both IPv4 and IPv6
+    #   connectivity. The notebook instance must be stopped before updating
+    #   this setting. When you specify `dualstack`, the subnet must support
+    #   IPv6 addressing.
+    #   @return [String]
+    #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of the IAM role that SageMaker AI can
     #   assume to access the notebook instance. For more information, see
@@ -52155,6 +52253,7 @@ module Aws::SageMaker
     class UpdateNotebookInstanceInput < Struct.new(
       :notebook_instance_name,
       :instance_type,
+      :ip_address_type,
       :role_arn,
       :lifecycle_config_name,
       :disassociate_lifecycle_config,

data/lib/aws-sdk-sagemaker.rb

@@ -55,7 +55,7 @@ module Aws::SageMaker
   autoload :EndpointProvider, 'aws-sdk-sagemaker/endpoint_provider'
   autoload :Endpoints, 'aws-sdk-sagemaker/endpoints'
 
-  GEM_VERSION = '1.327.0'
+  GEM_VERSION = '1.328.0'
 
 end