aws-sdk-sagemaker 1.326.0 → 1.328.0

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

@@ -313,8 +313,14 @@ module Aws::SageMaker
     ClientSecret = Shapes::StringShape.new(name: 'ClientSecret')
     ClientToken = Shapes::StringShape.new(name: 'ClientToken')
     ClusterArn = Shapes::StringShape.new(name: 'ClusterArn')
+    ClusterAutoScalerType = Shapes::StringShape.new(name: 'ClusterAutoScalerType')
+    ClusterAutoScalingConfig = Shapes::StructureShape.new(name: 'ClusterAutoScalingConfig')
+    ClusterAutoScalingConfigOutput = Shapes::StructureShape.new(name: 'ClusterAutoScalingConfigOutput')
+    ClusterAutoScalingMode = Shapes::StringShape.new(name: 'ClusterAutoScalingMode')
+    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')
@@ -329,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')
@@ -369,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')
@@ -1194,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')
@@ -3391,6 +3400,16 @@ module Aws::SageMaker
     ClarifyTextConfig.add_member(:granularity, Shapes::ShapeRef.new(shape: ClarifyTextGranularity, required: true, location_name: "Granularity"))
     ClarifyTextConfig.struct_class = Types::ClarifyTextConfig
 
+    ClusterAutoScalingConfig.add_member(:mode, Shapes::ShapeRef.new(shape: ClusterAutoScalingMode, required: true, location_name: "Mode"))
+    ClusterAutoScalingConfig.add_member(:auto_scaler_type, Shapes::ShapeRef.new(shape: ClusterAutoScalerType, location_name: "AutoScalerType"))
+    ClusterAutoScalingConfig.struct_class = Types::ClusterAutoScalingConfig
+
+    ClusterAutoScalingConfigOutput.add_member(:mode, Shapes::ShapeRef.new(shape: ClusterAutoScalingMode, required: true, location_name: "Mode"))
+    ClusterAutoScalingConfigOutput.add_member(:auto_scaler_type, Shapes::ShapeRef.new(shape: ClusterAutoScalerType, location_name: "AutoScalerType"))
+    ClusterAutoScalingConfigOutput.add_member(:status, Shapes::ShapeRef.new(shape: ClusterAutoScalingStatus, required: true, location_name: "Status"))
+    ClusterAutoScalingConfigOutput.add_member(:failure_message, Shapes::ShapeRef.new(shape: String, location_name: "FailureMessage"))
+    ClusterAutoScalingConfigOutput.struct_class = Types::ClusterAutoScalingConfigOutput
+
     ClusterEbsVolumeConfig.add_member(:volume_size_in_gb, Shapes::ShapeRef.new(shape: ClusterEbsVolumeSizeInGB, location_name: "VolumeSizeInGB"))
     ClusterEbsVolumeConfig.add_member(:volume_kms_key_id, Shapes::ShapeRef.new(shape: KmsKeyId, location_name: "VolumeKmsKeyId"))
     ClusterEbsVolumeConfig.add_member(:root_volume, Shapes::ShapeRef.new(shape: Boolean, location_name: "RootVolume", metadata: {"box" => true}))
@@ -3578,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
@@ -3838,7 +3861,10 @@ 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"))
     CreateClusterRequest.struct_class = Types::CreateClusterRequest
 
     CreateClusterResponse.add_member(:cluster_arn, Shapes::ShapeRef.new(shape: ClusterArn, required: true, location_name: "ClusterArn"))
@@ -4302,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"))
@@ -5175,8 +5202,11 @@ 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"))
+    DescribeClusterResponse.add_member(:auto_scaling, Shapes::ShapeRef.new(shape: ClusterAutoScalingConfigOutput, location_name: "AutoScaling"))
     DescribeClusterResponse.struct_class = Types::DescribeClusterResponse
 
     DescribeClusterSchedulerConfigRequest.add_member(:cluster_scheduler_config_id, Shapes::ShapeRef.new(shape: ClusterSchedulerConfigId, required: true, location_name: "ClusterSchedulerConfigId"))
@@ -5887,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"))
@@ -11285,8 +11316,11 @@ 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"))
+    UpdateClusterRequest.add_member(:auto_scaling, Shapes::ShapeRef.new(shape: ClusterAutoScalingConfig, location_name: "AutoScaling"))
     UpdateClusterRequest.struct_class = Types::UpdateClusterRequest
 
     UpdateClusterResponse.add_member(:cluster_arn, Shapes::ShapeRef.new(shape: ClusterArn, required: true, location_name: "ClusterArn"))
@@ -11551,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

@@ -4700,6 +4700,60 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Specifies the autoscaling configuration for a HyperPod cluster.
+    #
+    # @!attribute [rw] mode
+    #   Describes whether autoscaling is enabled or disabled for the
+    #   cluster. Valid values are `Enable` and `Disable`.
+    #   @return [String]
+    #
+    # @!attribute [rw] auto_scaler_type
+    #   The type of autoscaler to use. Currently supported value is
+    #   `Karpenter`.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterAutoScalingConfig AWS API Documentation
+    #
+    class ClusterAutoScalingConfig < Struct.new(
+      :mode,
+      :auto_scaler_type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The autoscaling configuration and status information for a HyperPod
+    # cluster.
+    #
+    # @!attribute [rw] mode
+    #   Describes whether autoscaling is enabled or disabled for the
+    #   cluster.
+    #   @return [String]
+    #
+    # @!attribute [rw] auto_scaler_type
+    #   The type of autoscaler configured for the cluster.
+    #   @return [String]
+    #
+    # @!attribute [rw] status
+    #   The current status of the autoscaling configuration. Valid values
+    #   are `InService`, `Failed`, `Creating`, and `Deleting`.
+    #   @return [String]
+    #
+    # @!attribute [rw] failure_message
+    #   If the autoscaling status is `Failed`, this field contains a message
+    #   describing the failure.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterAutoScalingConfigOutput AWS API Documentation
+    #
+    class ClusterAutoScalingConfigOutput < Struct.new(
+      :mode,
+      :auto_scaler_type,
+      :status,
+      :failure_message)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Defines the configuration for attaching an additional Amazon Elastic
     # Block Store (EBS) volume to each instance of the SageMaker HyperPod
     # cluster instance group. To learn more, see [SageMaker HyperPod release
@@ -5794,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.
@@ -7312,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:
@@ -7327,6 +7418,21 @@ module Aws::SageMaker
     #   ^
     #   @return [String]
     #
+    # @!attribute [rw] cluster_role
+    #   The Amazon Resource Name (ARN) of the IAM role that HyperPod assumes
+    #   to perform cluster autoscaling operations. This role must have
+    #   permissions for `sagemaker:BatchAddClusterNodes` and
+    #   `sagemaker:BatchDeleteClusterNodes`. This is only required when
+    #   autoscaling is enabled and when HyperPod is performing autoscaling
+    #   operations.
+    #   @return [String]
+    #
+    # @!attribute [rw] auto_scaling
+    #   The autoscaling configuration for the cluster. Enables automatic
+    #   scaling of cluster nodes based on workload demand using a
+    #   Karpenter-based system.
+    #   @return [Types::ClusterAutoScalingConfig]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreateClusterRequest AWS API Documentation
     #
     class CreateClusterRequest < Struct.new(
@@ -7337,7 +7443,10 @@ module Aws::SageMaker
       :tags,
       :orchestrator,
       :node_recovery,
-      :node_provisioning_mode)
+      :tiered_storage_config,
+      :node_provisioning_mode,
+      :cluster_role,
+      :auto_scaling)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -9352,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`)
@@ -10390,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
@@ -10532,6 +10649,7 @@ module Aws::SageMaker
       :instance_type,
       :subnet_id,
       :security_group_ids,
+      :ip_address_type,
       :role_arn,
       :kms_key_id,
       :tags,
@@ -15028,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.
@@ -15037,6 +15162,15 @@ module Aws::SageMaker
     #   The mode used for provisioning nodes in the cluster.
     #   @return [String]
     #
+    # @!attribute [rw] cluster_role
+    #   The Amazon Resource Name (ARN) of the IAM role that HyperPod uses
+    #   for cluster autoscaling operations.
+    #   @return [String]
+    #
+    # @!attribute [rw] auto_scaling
+    #   The current autoscaling configuration and status for the autoscaler.
+    #   @return [Types::ClusterAutoScalingConfigOutput]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeClusterResponse AWS API Documentation
     #
     class DescribeClusterResponse < Struct.new(
@@ -15049,8 +15183,11 @@ module Aws::SageMaker
       :restricted_instance_groups,
       :vpc_config,
       :orchestrator,
+      :tiered_storage_config,
       :node_recovery,
-      :node_provisioning_mode)
+      :node_provisioning_mode,
+      :cluster_role,
+      :auto_scaling)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -18940,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]
@@ -19073,6 +19216,7 @@ module Aws::SageMaker
       :failure_reason,
       :url,
       :instance_type,
+      :ip_address_type,
       :subnet_id,
       :security_groups,
       :role_arn,
@@ -19249,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
@@ -21483,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
@@ -50541,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.
@@ -50551,14 +50722,28 @@ module Aws::SageMaker
     #   as the separator between multiple names.
     #   @return [Array<String>]
     #
+    # @!attribute [rw] cluster_role
+    #   The Amazon Resource Name (ARN) of the IAM role that HyperPod assumes
+    #   for cluster autoscaling operations. Cannot be updated while
+    #   autoscaling is enabled.
+    #   @return [String]
+    #
+    # @!attribute [rw] auto_scaling
+    #   Updates the autoscaling configuration for the cluster. Use to enable
+    #   or disable automatic node scaling.
+    #   @return [Types::ClusterAutoScalingConfig]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UpdateClusterRequest AWS API Documentation
     #
     class UpdateClusterRequest < Struct.new(
       :cluster_name,
       :instance_groups,
       :restricted_instance_groups,
+      :tiered_storage_config,
       :node_recovery,
-      :instance_groups_to_delete)
+      :instance_groups_to_delete,
+      :cluster_role,
+      :auto_scaling)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -51934,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
@@ -52060,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.326.0'
+  GEM_VERSION = '1.328.0'
 
 end