aws-sdk-sagemaker 1.340.0 → 1.341.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f760b6f6d6902ca08349489fe9e34c577365953a32aa2f031a4db9e67aa9cb08
-  data.tar.gz: 5d01231c9d2c1d978d56deee398fb98fe79da54085d362172671724d6bce3233
+  metadata.gz: 26865b73cdcf61e9575a07d832b43aa93e24edce14672d0fec79742958394014
+  data.tar.gz: 0d0613bf3801d91d0d176695ccf4422a7e531ec6054f80a735069034eb79a2a1
 SHA512:
-  metadata.gz: a658467635f982e5e53c0cd5a428c96d8f17acb120d16c31184b8ac8ddd2f21c9062c41e6df84610f95227118ae6de6a47053642188e8ed29208e2d44cfa5863
-  data.tar.gz: ec789fc4a9c2d3014b4e19c7a5b628f158fdc8a3b262139e86e44147b83a65107844ac19445d924aa41f1e90791d961c6c1bf8abb2ef19a9de49e6a54e63cba0
+  metadata.gz: 06ee60e1d183dab830facd89ff879c1df147cc673f511e681cf0ecc2f45211585f034dbeb9893534cf8d1f4486554e59a8375d1acf5b8cd4398f8944245ad875
+  data.tar.gz: 913fb03b0fea5374c8eb21cfc8d4668e1705b493f94b9fb1ab30646ef2bc0bce18bc9deddbf5c0a20034dc466cb34b532de52eaeda6e822999d525659eda3585

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 Unreleased Changes
 ------------------
 
+1.341.0 (2025-11-21)
+------------------
+
+* Feature - Enhanced SageMaker HyperPod instance groups with support for MinInstanceCount, CapacityRequirements (Spot/On-Demand), and KubernetesConfig (labels and taints). Also Added speculative decoding and MaxInstanceCount for model optimization jobs.
+
 1.340.0 (2025-11-20)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.340.0
+1.341.0

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

@@ -2446,6 +2446,7 @@ module Aws::SageMaker
     #     instance_groups: [
     #       {
     #         instance_count: 1, # required
+    #         min_instance_count: 1,
     #         instance_group_name: "ClusterInstanceGroupName", # required
     #         instance_type: "ml.p4d.24xlarge", # required, accepts ml.p4d.24xlarge, ml.p4de.24xlarge, ml.p5.48xlarge, ml.p6e-gb200.36xlarge, ml.trn1.32xlarge, ml.trn1n.32xlarge, ml.g5.xlarge, ml.g5.2xlarge, ml.g5.4xlarge, ml.g5.8xlarge, ml.g5.12xlarge, ml.g5.16xlarge, ml.g5.24xlarge, ml.g5.48xlarge, ml.c5.large, ml.c5.xlarge, ml.c5.2xlarge, ml.c5.4xlarge, ml.c5.9xlarge, ml.c5.12xlarge, ml.c5.18xlarge, ml.c5.24xlarge, ml.c5n.large, ml.c5n.2xlarge, ml.c5n.4xlarge, ml.c5n.9xlarge, ml.c5n.18xlarge, ml.m5.large, ml.m5.xlarge, ml.m5.2xlarge, ml.m5.4xlarge, ml.m5.8xlarge, ml.m5.12xlarge, ml.m5.16xlarge, ml.m5.24xlarge, ml.t3.medium, ml.t3.large, ml.t3.xlarge, ml.t3.2xlarge, ml.g6.xlarge, ml.g6.2xlarge, ml.g6.4xlarge, ml.g6.8xlarge, ml.g6.16xlarge, ml.g6.12xlarge, ml.g6.24xlarge, ml.g6.48xlarge, ml.gr6.4xlarge, ml.gr6.8xlarge, ml.g6e.xlarge, ml.g6e.2xlarge, ml.g6e.4xlarge, ml.g6e.8xlarge, ml.g6e.16xlarge, ml.g6e.12xlarge, ml.g6e.24xlarge, ml.g6e.48xlarge, ml.p5e.48xlarge, ml.p5en.48xlarge, ml.p6-b200.48xlarge, ml.trn2.3xlarge, ml.trn2.48xlarge, ml.c6i.large, ml.c6i.xlarge, ml.c6i.2xlarge, ml.c6i.4xlarge, ml.c6i.8xlarge, ml.c6i.12xlarge, ml.c6i.16xlarge, ml.c6i.24xlarge, ml.c6i.32xlarge, ml.m6i.large, ml.m6i.xlarge, ml.m6i.2xlarge, ml.m6i.4xlarge, ml.m6i.8xlarge, ml.m6i.12xlarge, ml.m6i.16xlarge, ml.m6i.24xlarge, ml.m6i.32xlarge, ml.r6i.large, ml.r6i.xlarge, ml.r6i.2xlarge, ml.r6i.4xlarge, ml.r6i.8xlarge, ml.r6i.12xlarge, ml.r6i.16xlarge, ml.r6i.24xlarge, ml.r6i.32xlarge, ml.i3en.large, ml.i3en.xlarge, ml.i3en.2xlarge, ml.i3en.3xlarge, ml.i3en.6xlarge, ml.i3en.12xlarge, ml.i3en.24xlarge, ml.m7i.large, ml.m7i.xlarge, ml.m7i.2xlarge, ml.m7i.4xlarge, ml.m7i.8xlarge, ml.m7i.12xlarge, ml.m7i.16xlarge, ml.m7i.24xlarge, ml.m7i.48xlarge, ml.r7i.large, ml.r7i.xlarge, ml.r7i.2xlarge, ml.r7i.4xlarge, ml.r7i.8xlarge, ml.r7i.12xlarge, ml.r7i.16xlarge, ml.r7i.24xlarge, ml.r7i.48xlarge
     #         life_cycle_config: { # required
@@ -2491,6 +2492,24 @@ module Aws::SageMaker
     #           },
     #         },
     #         image_id: "ImageId",
+    #         kubernetes_config: {
+    #           labels: {
+    #             "ClusterKubernetesLabelKey" => "ClusterKubernetesLabelValue",
+    #           },
+    #           taints: [
+    #             {
+    #               key: "ClusterKubernetesTaintKey", # required
+    #               value: "ClusterKubernetesTaintValue",
+    #               effect: "NoSchedule", # required, accepts NoSchedule, PreferNoSchedule, NoExecute
+    #             },
+    #           ],
+    #         },
+    #         capacity_requirements: {
+    #           spot: {
+    #           },
+    #           on_demand: {
+    #           },
+    #         },
     #       },
     #     ],
     #     restricted_instance_groups: [
@@ -8570,6 +8589,9 @@ module Aws::SageMaker
     #   The type of instance that hosts the optimized model that you create
     #   with the optimization job.
     #
+    # @option params [Integer] :max_instance_count
+    #   The maximum number of instances to use for the optimization job.
+    #
     # @option params [Hash<String,String>] :optimization_environment
     #   The environment variables to set in the model container.
     #
@@ -8631,8 +8653,12 @@ module Aws::SageMaker
     #           accept_eula: false, # required
     #         },
     #       },
+    #       sage_maker_model: {
+    #         model_name: "ModelName",
+    #       },
     #     },
-    #     deployment_instance_type: "ml.p4d.24xlarge", # required, accepts ml.p4d.24xlarge, ml.p4de.24xlarge, ml.p5.48xlarge, ml.g5.xlarge, ml.g5.2xlarge, ml.g5.4xlarge, ml.g5.8xlarge, ml.g5.12xlarge, ml.g5.16xlarge, ml.g5.24xlarge, ml.g5.48xlarge, ml.g6.xlarge, ml.g6.2xlarge, ml.g6.4xlarge, ml.g6.8xlarge, ml.g6.12xlarge, ml.g6.16xlarge, ml.g6.24xlarge, ml.g6.48xlarge, ml.g6e.xlarge, ml.g6e.2xlarge, ml.g6e.4xlarge, ml.g6e.8xlarge, ml.g6e.12xlarge, ml.g6e.16xlarge, ml.g6e.24xlarge, ml.g6e.48xlarge, ml.inf2.xlarge, ml.inf2.8xlarge, ml.inf2.24xlarge, ml.inf2.48xlarge, ml.trn1.2xlarge, ml.trn1.32xlarge, ml.trn1n.32xlarge
+    #     deployment_instance_type: "ml.p4d.24xlarge", # required, accepts ml.p4d.24xlarge, ml.p4de.24xlarge, ml.p5.48xlarge, ml.p5e.48xlarge, ml.p5en.48xlarge, ml.g5.xlarge, ml.g5.2xlarge, ml.g5.4xlarge, ml.g5.8xlarge, ml.g5.12xlarge, ml.g5.16xlarge, ml.g5.24xlarge, ml.g5.48xlarge, ml.g6.xlarge, ml.g6.2xlarge, ml.g6.4xlarge, ml.g6.8xlarge, ml.g6.12xlarge, ml.g6.16xlarge, ml.g6.24xlarge, ml.g6.48xlarge, ml.g6e.xlarge, ml.g6e.2xlarge, ml.g6e.4xlarge, ml.g6e.8xlarge, ml.g6e.12xlarge, ml.g6e.16xlarge, ml.g6e.24xlarge, ml.g6e.48xlarge, ml.inf2.xlarge, ml.inf2.8xlarge, ml.inf2.24xlarge, ml.inf2.48xlarge, ml.trn1.2xlarge, ml.trn1.32xlarge, ml.trn1n.32xlarge
+    #     max_instance_count: 1,
     #     optimization_environment: {
     #       "NonEmptyString256" => "String256",
     #     },
@@ -8656,11 +8682,21 @@ module Aws::SageMaker
     #             "NonEmptyString256" => "String256",
     #           },
     #         },
+    #         model_speculative_decoding_config: {
+    #           technique: "EAGLE", # required, accepts EAGLE
+    #           training_data_source: {
+    #             s3_uri: "S3Uri", # required
+    #             s3_data_type: "S3Prefix", # required, accepts S3Prefix, ManifestFile
+    #           },
+    #         },
     #       },
     #     ],
     #     output_config: { # required
     #       kms_key_id: "KmsKeyId",
     #       s3_output_location: "S3Uri", # required
+    #       sage_maker_model: {
+    #         model_name: "ModelName",
+    #       },
     #     },
     #     stopping_condition: { # required
     #       max_runtime_in_seconds: 1,
@@ -13650,6 +13686,7 @@ module Aws::SageMaker
     #   resp.instance_groups #=> Array
     #   resp.instance_groups[0].current_count #=> Integer
     #   resp.instance_groups[0].target_count #=> Integer
+    #   resp.instance_groups[0].min_count #=> Integer
     #   resp.instance_groups[0].instance_group_name #=> String
     #   resp.instance_groups[0].instance_type #=> String, one of "ml.p4d.24xlarge", "ml.p4de.24xlarge", "ml.p5.48xlarge", "ml.p6e-gb200.36xlarge", "ml.trn1.32xlarge", "ml.trn1n.32xlarge", "ml.g5.xlarge", "ml.g5.2xlarge", "ml.g5.4xlarge", "ml.g5.8xlarge", "ml.g5.12xlarge", "ml.g5.16xlarge", "ml.g5.24xlarge", "ml.g5.48xlarge", "ml.c5.large", "ml.c5.xlarge", "ml.c5.2xlarge", "ml.c5.4xlarge", "ml.c5.9xlarge", "ml.c5.12xlarge", "ml.c5.18xlarge", "ml.c5.24xlarge", "ml.c5n.large", "ml.c5n.2xlarge", "ml.c5n.4xlarge", "ml.c5n.9xlarge", "ml.c5n.18xlarge", "ml.m5.large", "ml.m5.xlarge", "ml.m5.2xlarge", "ml.m5.4xlarge", "ml.m5.8xlarge", "ml.m5.12xlarge", "ml.m5.16xlarge", "ml.m5.24xlarge", "ml.t3.medium", "ml.t3.large", "ml.t3.xlarge", "ml.t3.2xlarge", "ml.g6.xlarge", "ml.g6.2xlarge", "ml.g6.4xlarge", "ml.g6.8xlarge", "ml.g6.16xlarge", "ml.g6.12xlarge", "ml.g6.24xlarge", "ml.g6.48xlarge", "ml.gr6.4xlarge", "ml.gr6.8xlarge", "ml.g6e.xlarge", "ml.g6e.2xlarge", "ml.g6e.4xlarge", "ml.g6e.8xlarge", "ml.g6e.16xlarge", "ml.g6e.12xlarge", "ml.g6e.24xlarge", "ml.g6e.48xlarge", "ml.p5e.48xlarge", "ml.p5en.48xlarge", "ml.p6-b200.48xlarge", "ml.trn2.3xlarge", "ml.trn2.48xlarge", "ml.c6i.large", "ml.c6i.xlarge", "ml.c6i.2xlarge", "ml.c6i.4xlarge", "ml.c6i.8xlarge", "ml.c6i.12xlarge", "ml.c6i.16xlarge", "ml.c6i.24xlarge", "ml.c6i.32xlarge", "ml.m6i.large", "ml.m6i.xlarge", "ml.m6i.2xlarge", "ml.m6i.4xlarge", "ml.m6i.8xlarge", "ml.m6i.12xlarge", "ml.m6i.16xlarge", "ml.m6i.24xlarge", "ml.m6i.32xlarge", "ml.r6i.large", "ml.r6i.xlarge", "ml.r6i.2xlarge", "ml.r6i.4xlarge", "ml.r6i.8xlarge", "ml.r6i.12xlarge", "ml.r6i.16xlarge", "ml.r6i.24xlarge", "ml.r6i.32xlarge", "ml.i3en.large", "ml.i3en.xlarge", "ml.i3en.2xlarge", "ml.i3en.3xlarge", "ml.i3en.6xlarge", "ml.i3en.12xlarge", "ml.i3en.24xlarge", "ml.m7i.large", "ml.m7i.xlarge", "ml.m7i.2xlarge", "ml.m7i.4xlarge", "ml.m7i.8xlarge", "ml.m7i.12xlarge", "ml.m7i.16xlarge", "ml.m7i.24xlarge", "ml.m7i.48xlarge", "ml.r7i.large", "ml.r7i.xlarge", "ml.r7i.2xlarge", "ml.r7i.4xlarge", "ml.r7i.8xlarge", "ml.r7i.12xlarge", "ml.r7i.16xlarge", "ml.r7i.24xlarge", "ml.r7i.48xlarge"
     #   resp.instance_groups[0].life_cycle_config.source_s3_uri #=> String
@@ -13679,6 +13716,20 @@ module Aws::SageMaker
     #   resp.instance_groups[0].scheduled_update_config.deployment_config.auto_rollback_configuration[0].alarm_name #=> String
     #   resp.instance_groups[0].current_image_id #=> String
     #   resp.instance_groups[0].desired_image_id #=> String
+    #   resp.instance_groups[0].active_operations #=> Hash
+    #   resp.instance_groups[0].active_operations["ActiveClusterOperationName"] #=> Integer
+    #   resp.instance_groups[0].kubernetes_config.current_labels #=> Hash
+    #   resp.instance_groups[0].kubernetes_config.current_labels["ClusterKubernetesLabelKey"] #=> String
+    #   resp.instance_groups[0].kubernetes_config.desired_labels #=> Hash
+    #   resp.instance_groups[0].kubernetes_config.desired_labels["ClusterKubernetesLabelKey"] #=> String
+    #   resp.instance_groups[0].kubernetes_config.current_taints #=> Array
+    #   resp.instance_groups[0].kubernetes_config.current_taints[0].key #=> String
+    #   resp.instance_groups[0].kubernetes_config.current_taints[0].value #=> String
+    #   resp.instance_groups[0].kubernetes_config.current_taints[0].effect #=> String, one of "NoSchedule", "PreferNoSchedule", "NoExecute"
+    #   resp.instance_groups[0].kubernetes_config.desired_taints #=> Array
+    #   resp.instance_groups[0].kubernetes_config.desired_taints[0].key #=> String
+    #   resp.instance_groups[0].kubernetes_config.desired_taints[0].value #=> String
+    #   resp.instance_groups[0].kubernetes_config.desired_taints[0].effect #=> String, one of "NoSchedule", "PreferNoSchedule", "NoExecute"
     #   resp.instance_groups[0].target_state_count #=> Integer
     #   resp.instance_groups[0].software_update_status #=> String, one of "Pending", "InProgress", "Succeeded", "Failed", "RollbackInProgress", "RollbackComplete"
     #   resp.instance_groups[0].active_software_update_config.rolling_update_policy.maximum_batch_size.type #=> String, one of "INSTANCE_COUNT", "CAPACITY_PERCENTAGE"
@@ -13789,6 +13840,7 @@ module Aws::SageMaker
     #   resp.event_details.event_details.event_metadata.instance_group.ami_override #=> String
     #   resp.event_details.event_details.event_metadata.instance_group_scaling.instance_count #=> Integer
     #   resp.event_details.event_details.event_metadata.instance_group_scaling.target_count #=> Integer
+    #   resp.event_details.event_details.event_metadata.instance_group_scaling.min_count #=> Integer
     #   resp.event_details.event_details.event_metadata.instance_group_scaling.failure_message #=> String
     #   resp.event_details.event_details.event_metadata.instance.customer_eni #=> String
     #   resp.event_details.event_details.event_metadata.instance.additional_enis.efa_enis #=> Array
@@ -13866,6 +13918,19 @@ module Aws::SageMaker
     #   resp.node_details.current_image_id #=> String
     #   resp.node_details.desired_image_id #=> String
     #   resp.node_details.ultra_server_info.id #=> String
+    #   resp.node_details.kubernetes_config.current_labels #=> Hash
+    #   resp.node_details.kubernetes_config.current_labels["ClusterKubernetesLabelKey"] #=> String
+    #   resp.node_details.kubernetes_config.desired_labels #=> Hash
+    #   resp.node_details.kubernetes_config.desired_labels["ClusterKubernetesLabelKey"] #=> String
+    #   resp.node_details.kubernetes_config.current_taints #=> Array
+    #   resp.node_details.kubernetes_config.current_taints[0].key #=> String
+    #   resp.node_details.kubernetes_config.current_taints[0].value #=> String
+    #   resp.node_details.kubernetes_config.current_taints[0].effect #=> String, one of "NoSchedule", "PreferNoSchedule", "NoExecute"
+    #   resp.node_details.kubernetes_config.desired_taints #=> Array
+    #   resp.node_details.kubernetes_config.desired_taints[0].key #=> String
+    #   resp.node_details.kubernetes_config.desired_taints[0].value #=> String
+    #   resp.node_details.kubernetes_config.desired_taints[0].effect #=> String, one of "NoSchedule", "PreferNoSchedule", "NoExecute"
+    #   resp.node_details.capacity_type #=> String, one of "Spot", "OnDemand"
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeClusterNode AWS API Documentation
     #
@@ -17608,6 +17673,7 @@ module Aws::SageMaker
     #   * {Types::DescribeOptimizationJobResponse#model_source #model_source} => Types::OptimizationJobModelSource
     #   * {Types::DescribeOptimizationJobResponse#optimization_environment #optimization_environment} => Hash&lt;String,String&gt;
     #   * {Types::DescribeOptimizationJobResponse#deployment_instance_type #deployment_instance_type} => String
+    #   * {Types::DescribeOptimizationJobResponse#max_instance_count #max_instance_count} => Integer
     #   * {Types::DescribeOptimizationJobResponse#optimization_configs #optimization_configs} => Array&lt;Types::OptimizationConfig&gt;
     #   * {Types::DescribeOptimizationJobResponse#output_config #output_config} => Types::OptimizationJobOutputConfig
     #   * {Types::DescribeOptimizationJobResponse#optimization_output #optimization_output} => Types::OptimizationOutput
@@ -17633,9 +17699,11 @@ module Aws::SageMaker
     #   resp.optimization_job_name #=> String
     #   resp.model_source.s3.s3_uri #=> String
     #   resp.model_source.s3.model_access_config.accept_eula #=> Boolean
+    #   resp.model_source.sage_maker_model.model_name #=> String
     #   resp.optimization_environment #=> Hash
     #   resp.optimization_environment["NonEmptyString256"] #=> String
-    #   resp.deployment_instance_type #=> String, one of "ml.p4d.24xlarge", "ml.p4de.24xlarge", "ml.p5.48xlarge", "ml.g5.xlarge", "ml.g5.2xlarge", "ml.g5.4xlarge", "ml.g5.8xlarge", "ml.g5.12xlarge", "ml.g5.16xlarge", "ml.g5.24xlarge", "ml.g5.48xlarge", "ml.g6.xlarge", "ml.g6.2xlarge", "ml.g6.4xlarge", "ml.g6.8xlarge", "ml.g6.12xlarge", "ml.g6.16xlarge", "ml.g6.24xlarge", "ml.g6.48xlarge", "ml.g6e.xlarge", "ml.g6e.2xlarge", "ml.g6e.4xlarge", "ml.g6e.8xlarge", "ml.g6e.12xlarge", "ml.g6e.16xlarge", "ml.g6e.24xlarge", "ml.g6e.48xlarge", "ml.inf2.xlarge", "ml.inf2.8xlarge", "ml.inf2.24xlarge", "ml.inf2.48xlarge", "ml.trn1.2xlarge", "ml.trn1.32xlarge", "ml.trn1n.32xlarge"
+    #   resp.deployment_instance_type #=> String, one of "ml.p4d.24xlarge", "ml.p4de.24xlarge", "ml.p5.48xlarge", "ml.p5e.48xlarge", "ml.p5en.48xlarge", "ml.g5.xlarge", "ml.g5.2xlarge", "ml.g5.4xlarge", "ml.g5.8xlarge", "ml.g5.12xlarge", "ml.g5.16xlarge", "ml.g5.24xlarge", "ml.g5.48xlarge", "ml.g6.xlarge", "ml.g6.2xlarge", "ml.g6.4xlarge", "ml.g6.8xlarge", "ml.g6.12xlarge", "ml.g6.16xlarge", "ml.g6.24xlarge", "ml.g6.48xlarge", "ml.g6e.xlarge", "ml.g6e.2xlarge", "ml.g6e.4xlarge", "ml.g6e.8xlarge", "ml.g6e.12xlarge", "ml.g6e.16xlarge", "ml.g6e.24xlarge", "ml.g6e.48xlarge", "ml.inf2.xlarge", "ml.inf2.8xlarge", "ml.inf2.24xlarge", "ml.inf2.48xlarge", "ml.trn1.2xlarge", "ml.trn1.32xlarge", "ml.trn1n.32xlarge"
+    #   resp.max_instance_count #=> Integer
     #   resp.optimization_configs #=> Array
     #   resp.optimization_configs[0].model_quantization_config.image #=> String
     #   resp.optimization_configs[0].model_quantization_config.override_environment #=> Hash
@@ -17646,8 +17714,12 @@ module Aws::SageMaker
     #   resp.optimization_configs[0].model_sharding_config.image #=> String
     #   resp.optimization_configs[0].model_sharding_config.override_environment #=> Hash
     #   resp.optimization_configs[0].model_sharding_config.override_environment["NonEmptyString256"] #=> String
+    #   resp.optimization_configs[0].model_speculative_decoding_config.technique #=> String, one of "EAGLE"
+    #   resp.optimization_configs[0].model_speculative_decoding_config.training_data_source.s3_uri #=> String
+    #   resp.optimization_configs[0].model_speculative_decoding_config.training_data_source.s3_data_type #=> String, one of "S3Prefix", "ManifestFile"
     #   resp.output_config.kms_key_id #=> String
     #   resp.output_config.s3_output_location #=> String
+    #   resp.output_config.sage_maker_model.model_name #=> String
     #   resp.optimization_output.recommended_inference_image #=> String
     #   resp.role_arn #=> String
     #   resp.stopping_condition.max_runtime_in_seconds #=> Integer
@@ -24473,7 +24545,8 @@ module Aws::SageMaker
     #   resp.optimization_job_summaries[0].optimization_start_time #=> Time
     #   resp.optimization_job_summaries[0].optimization_end_time #=> Time
     #   resp.optimization_job_summaries[0].last_modified_time #=> Time
-    #   resp.optimization_job_summaries[0].deployment_instance_type #=> String, one of "ml.p4d.24xlarge", "ml.p4de.24xlarge", "ml.p5.48xlarge", "ml.g5.xlarge", "ml.g5.2xlarge", "ml.g5.4xlarge", "ml.g5.8xlarge", "ml.g5.12xlarge", "ml.g5.16xlarge", "ml.g5.24xlarge", "ml.g5.48xlarge", "ml.g6.xlarge", "ml.g6.2xlarge", "ml.g6.4xlarge", "ml.g6.8xlarge", "ml.g6.12xlarge", "ml.g6.16xlarge", "ml.g6.24xlarge", "ml.g6.48xlarge", "ml.g6e.xlarge", "ml.g6e.2xlarge", "ml.g6e.4xlarge", "ml.g6e.8xlarge", "ml.g6e.12xlarge", "ml.g6e.16xlarge", "ml.g6e.24xlarge", "ml.g6e.48xlarge", "ml.inf2.xlarge", "ml.inf2.8xlarge", "ml.inf2.24xlarge", "ml.inf2.48xlarge", "ml.trn1.2xlarge", "ml.trn1.32xlarge", "ml.trn1n.32xlarge"
+    #   resp.optimization_job_summaries[0].deployment_instance_type #=> String, one of "ml.p4d.24xlarge", "ml.p4de.24xlarge", "ml.p5.48xlarge", "ml.p5e.48xlarge", "ml.p5en.48xlarge", "ml.g5.xlarge", "ml.g5.2xlarge", "ml.g5.4xlarge", "ml.g5.8xlarge", "ml.g5.12xlarge", "ml.g5.16xlarge", "ml.g5.24xlarge", "ml.g5.48xlarge", "ml.g6.xlarge", "ml.g6.2xlarge", "ml.g6.4xlarge", "ml.g6.8xlarge", "ml.g6.12xlarge", "ml.g6.16xlarge", "ml.g6.24xlarge", "ml.g6.48xlarge", "ml.g6e.xlarge", "ml.g6e.2xlarge", "ml.g6e.4xlarge", "ml.g6e.8xlarge", "ml.g6e.12xlarge", "ml.g6e.16xlarge", "ml.g6e.24xlarge", "ml.g6e.48xlarge", "ml.inf2.xlarge", "ml.inf2.8xlarge", "ml.inf2.24xlarge", "ml.inf2.48xlarge", "ml.trn1.2xlarge", "ml.trn1.32xlarge", "ml.trn1n.32xlarge"
+    #   resp.optimization_job_summaries[0].max_instance_count #=> Integer
     #   resp.optimization_job_summaries[0].optimization_types #=> Array
     #   resp.optimization_job_summaries[0].optimization_types[0] #=> String
     #   resp.next_token #=> String
@@ -27812,6 +27885,7 @@ module Aws::SageMaker
     #     instance_groups: [
     #       {
     #         instance_count: 1, # required
+    #         min_instance_count: 1,
     #         instance_group_name: "ClusterInstanceGroupName", # required
     #         instance_type: "ml.p4d.24xlarge", # required, accepts ml.p4d.24xlarge, ml.p4de.24xlarge, ml.p5.48xlarge, ml.p6e-gb200.36xlarge, ml.trn1.32xlarge, ml.trn1n.32xlarge, ml.g5.xlarge, ml.g5.2xlarge, ml.g5.4xlarge, ml.g5.8xlarge, ml.g5.12xlarge, ml.g5.16xlarge, ml.g5.24xlarge, ml.g5.48xlarge, ml.c5.large, ml.c5.xlarge, ml.c5.2xlarge, ml.c5.4xlarge, ml.c5.9xlarge, ml.c5.12xlarge, ml.c5.18xlarge, ml.c5.24xlarge, ml.c5n.large, ml.c5n.2xlarge, ml.c5n.4xlarge, ml.c5n.9xlarge, ml.c5n.18xlarge, ml.m5.large, ml.m5.xlarge, ml.m5.2xlarge, ml.m5.4xlarge, ml.m5.8xlarge, ml.m5.12xlarge, ml.m5.16xlarge, ml.m5.24xlarge, ml.t3.medium, ml.t3.large, ml.t3.xlarge, ml.t3.2xlarge, ml.g6.xlarge, ml.g6.2xlarge, ml.g6.4xlarge, ml.g6.8xlarge, ml.g6.16xlarge, ml.g6.12xlarge, ml.g6.24xlarge, ml.g6.48xlarge, ml.gr6.4xlarge, ml.gr6.8xlarge, ml.g6e.xlarge, ml.g6e.2xlarge, ml.g6e.4xlarge, ml.g6e.8xlarge, ml.g6e.16xlarge, ml.g6e.12xlarge, ml.g6e.24xlarge, ml.g6e.48xlarge, ml.p5e.48xlarge, ml.p5en.48xlarge, ml.p6-b200.48xlarge, ml.trn2.3xlarge, ml.trn2.48xlarge, ml.c6i.large, ml.c6i.xlarge, ml.c6i.2xlarge, ml.c6i.4xlarge, ml.c6i.8xlarge, ml.c6i.12xlarge, ml.c6i.16xlarge, ml.c6i.24xlarge, ml.c6i.32xlarge, ml.m6i.large, ml.m6i.xlarge, ml.m6i.2xlarge, ml.m6i.4xlarge, ml.m6i.8xlarge, ml.m6i.12xlarge, ml.m6i.16xlarge, ml.m6i.24xlarge, ml.m6i.32xlarge, ml.r6i.large, ml.r6i.xlarge, ml.r6i.2xlarge, ml.r6i.4xlarge, ml.r6i.8xlarge, ml.r6i.12xlarge, ml.r6i.16xlarge, ml.r6i.24xlarge, ml.r6i.32xlarge, ml.i3en.large, ml.i3en.xlarge, ml.i3en.2xlarge, ml.i3en.3xlarge, ml.i3en.6xlarge, ml.i3en.12xlarge, ml.i3en.24xlarge, ml.m7i.large, ml.m7i.xlarge, ml.m7i.2xlarge, ml.m7i.4xlarge, ml.m7i.8xlarge, ml.m7i.12xlarge, ml.m7i.16xlarge, ml.m7i.24xlarge, ml.m7i.48xlarge, ml.r7i.large, ml.r7i.xlarge, ml.r7i.2xlarge, ml.r7i.4xlarge, ml.r7i.8xlarge, ml.r7i.12xlarge, ml.r7i.16xlarge, ml.r7i.24xlarge, ml.r7i.48xlarge
     #         life_cycle_config: { # required
@@ -27857,6 +27931,24 @@ module Aws::SageMaker
     #           },
     #         },
     #         image_id: "ImageId",
+    #         kubernetes_config: {
+    #           labels: {
+    #             "ClusterKubernetesLabelKey" => "ClusterKubernetesLabelValue",
+    #           },
+    #           taints: [
+    #             {
+    #               key: "ClusterKubernetesTaintKey", # required
+    #               value: "ClusterKubernetesTaintValue",
+    #               effect: "NoSchedule", # required, accepts NoSchedule, PreferNoSchedule, NoExecute
+    #             },
+    #           ],
+    #         },
+    #         capacity_requirements: {
+    #           spot: {
+    #           },
+    #           on_demand: {
+    #           },
+    #         },
     #       },
     #     ],
     #     restricted_instance_groups: [
@@ -31579,7 +31671,7 @@ module Aws::SageMaker
         tracer: tracer
       )
       context[:gem_name] = 'aws-sdk-sagemaker'
-      context[:gem_version] = '1.340.0'
+      context[:gem_version] = '1.341.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -26,6 +26,9 @@ module Aws::SageMaker
     ActionSummaries = Shapes::ListShape.new(name: 'ActionSummaries')
     ActionSummary = Shapes::StructureShape.new(name: 'ActionSummary')
     ActivationState = Shapes::StringShape.new(name: 'ActivationState')
+    ActiveClusterOperationCount = Shapes::IntegerShape.new(name: 'ActiveClusterOperationCount')
+    ActiveClusterOperationName = Shapes::StringShape.new(name: 'ActiveClusterOperationName')
+    ActiveOperations = Shapes::MapShape.new(name: 'ActiveOperations')
     AddAssociationRequest = Shapes::StructureShape.new(name: 'AddAssociationRequest')
     AddAssociationResponse = Shapes::StructureShape.new(name: 'AddAssociationResponse')
     AddClusterNodeSpecification = Shapes::StructureShape.new(name: 'AddClusterNodeSpecification')
@@ -342,6 +345,8 @@ module Aws::SageMaker
     ClusterAutoScalingStatus = Shapes::StringShape.new(name: 'ClusterAutoScalingStatus')
     ClusterAvailabilityZone = Shapes::StringShape.new(name: 'ClusterAvailabilityZone')
     ClusterAvailabilityZoneId = Shapes::StringShape.new(name: 'ClusterAvailabilityZoneId')
+    ClusterCapacityRequirements = Shapes::StructureShape.new(name: 'ClusterCapacityRequirements')
+    ClusterCapacityType = Shapes::StringShape.new(name: 'ClusterCapacityType')
     ClusterConfigMode = Shapes::StringShape.new(name: 'ClusterConfigMode')
     ClusterEbsVolumeConfig = Shapes::StructureShape.new(name: 'ClusterEbsVolumeConfig')
     ClusterEbsVolumeSizeInGB = Shapes::IntegerShape.new(name: 'ClusterEbsVolumeSizeInGB')
@@ -364,6 +369,17 @@ module Aws::SageMaker
     ClusterInstanceStorageConfig = Shapes::UnionShape.new(name: 'ClusterInstanceStorageConfig')
     ClusterInstanceStorageConfigs = Shapes::ListShape.new(name: 'ClusterInstanceStorageConfigs')
     ClusterInstanceType = Shapes::StringShape.new(name: 'ClusterInstanceType')
+    ClusterKubernetesConfig = Shapes::StructureShape.new(name: 'ClusterKubernetesConfig')
+    ClusterKubernetesConfigDetails = Shapes::StructureShape.new(name: 'ClusterKubernetesConfigDetails')
+    ClusterKubernetesConfigNodeDetails = Shapes::StructureShape.new(name: 'ClusterKubernetesConfigNodeDetails')
+    ClusterKubernetesLabelKey = Shapes::StringShape.new(name: 'ClusterKubernetesLabelKey')
+    ClusterKubernetesLabelValue = Shapes::StringShape.new(name: 'ClusterKubernetesLabelValue')
+    ClusterKubernetesLabels = Shapes::MapShape.new(name: 'ClusterKubernetesLabels')
+    ClusterKubernetesTaint = Shapes::StructureShape.new(name: 'ClusterKubernetesTaint')
+    ClusterKubernetesTaintEffect = Shapes::StringShape.new(name: 'ClusterKubernetesTaintEffect')
+    ClusterKubernetesTaintKey = Shapes::StringShape.new(name: 'ClusterKubernetesTaintKey')
+    ClusterKubernetesTaintValue = Shapes::StringShape.new(name: 'ClusterKubernetesTaintValue')
+    ClusterKubernetesTaints = Shapes::ListShape.new(name: 'ClusterKubernetesTaints')
     ClusterLifeCycleConfig = Shapes::StructureShape.new(name: 'ClusterLifeCycleConfig')
     ClusterLifeCycleConfigFileName = Shapes::StringShape.new(name: 'ClusterLifeCycleConfigFileName')
     ClusterMetadata = Shapes::StructureShape.new(name: 'ClusterMetadata')
@@ -379,6 +395,7 @@ module Aws::SageMaker
     ClusterNodeSummaries = Shapes::ListShape.new(name: 'ClusterNodeSummaries')
     ClusterNodeSummary = Shapes::StructureShape.new(name: 'ClusterNodeSummary')
     ClusterNonNegativeInstanceCount = Shapes::IntegerShape.new(name: 'ClusterNonNegativeInstanceCount')
+    ClusterOnDemandOptions = Shapes::StructureShape.new(name: 'ClusterOnDemandOptions')
     ClusterOrchestrator = Shapes::StructureShape.new(name: 'ClusterOrchestrator')
     ClusterOrchestratorEksConfig = Shapes::StructureShape.new(name: 'ClusterOrchestratorEksConfig')
     ClusterPrivateDnsHostname = Shapes::StringShape.new(name: 'ClusterPrivateDnsHostname')
@@ -394,6 +411,7 @@ module Aws::SageMaker
     ClusterSchedulerConfigSummaryList = Shapes::ListShape.new(name: 'ClusterSchedulerConfigSummaryList')
     ClusterSchedulerPriorityClassName = Shapes::StringShape.new(name: 'ClusterSchedulerPriorityClassName')
     ClusterSortBy = Shapes::StringShape.new(name: 'ClusterSortBy')
+    ClusterSpotOptions = Shapes::StructureShape.new(name: 'ClusterSpotOptions')
     ClusterStatus = Shapes::StringShape.new(name: 'ClusterStatus')
     ClusterSummaries = Shapes::ListShape.new(name: 'ClusterSummaries')
     ClusterSummary = Shapes::StructureShape.new(name: 'ClusterSummary')
@@ -1743,6 +1761,10 @@ module Aws::SageMaker
     ModelSetupTime = Shapes::IntegerShape.new(name: 'ModelSetupTime')
     ModelShardingConfig = Shapes::StructureShape.new(name: 'ModelShardingConfig')
     ModelSortKey = Shapes::StringShape.new(name: 'ModelSortKey')
+    ModelSpeculativeDecodingConfig = Shapes::StructureShape.new(name: 'ModelSpeculativeDecodingConfig')
+    ModelSpeculativeDecodingS3DataType = Shapes::StringShape.new(name: 'ModelSpeculativeDecodingS3DataType')
+    ModelSpeculativeDecodingTechnique = Shapes::StringShape.new(name: 'ModelSpeculativeDecodingTechnique')
+    ModelSpeculativeDecodingTrainingDataSource = Shapes::StructureShape.new(name: 'ModelSpeculativeDecodingTrainingDataSource')
     ModelStepMetadata = Shapes::StructureShape.new(name: 'ModelStepMetadata')
     ModelSummary = Shapes::StructureShape.new(name: 'ModelSummary')
     ModelSummaryList = Shapes::ListShape.new(name: 'ModelSummaryList')
@@ -1877,6 +1899,7 @@ module Aws::SageMaker
     OptimizationJobArn = Shapes::StringShape.new(name: 'OptimizationJobArn')
     OptimizationJobDeploymentInstanceType = Shapes::StringShape.new(name: 'OptimizationJobDeploymentInstanceType')
     OptimizationJobEnvironmentVariables = Shapes::MapShape.new(name: 'OptimizationJobEnvironmentVariables')
+    OptimizationJobMaxInstanceCount = Shapes::IntegerShape.new(name: 'OptimizationJobMaxInstanceCount')
     OptimizationJobModelSource = Shapes::StructureShape.new(name: 'OptimizationJobModelSource')
     OptimizationJobModelSourceS3 = Shapes::StructureShape.new(name: 'OptimizationJobModelSourceS3')
     OptimizationJobOutputConfig = Shapes::StructureShape.new(name: 'OptimizationJobOutputConfig')
@@ -1886,6 +1909,7 @@ module Aws::SageMaker
     OptimizationModelAcceptEula = Shapes::BooleanShape.new(name: 'OptimizationModelAcceptEula')
     OptimizationModelAccessConfig = Shapes::StructureShape.new(name: 'OptimizationModelAccessConfig')
     OptimizationOutput = Shapes::StructureShape.new(name: 'OptimizationOutput')
+    OptimizationSageMakerModel = Shapes::StructureShape.new(name: 'OptimizationSageMakerModel')
     OptimizationType = Shapes::StringShape.new(name: 'OptimizationType')
     OptimizationTypes = Shapes::ListShape.new(name: 'OptimizationTypes')
     OptimizationVpcConfig = Shapes::StructureShape.new(name: 'OptimizationVpcConfig')
@@ -2749,6 +2773,9 @@ module Aws::SageMaker
     ActionSummary.add_member(:last_modified_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "LastModifiedTime"))
     ActionSummary.struct_class = Types::ActionSummary
 
+    ActiveOperations.key = Shapes::ShapeRef.new(shape: ActiveClusterOperationName)
+    ActiveOperations.value = Shapes::ShapeRef.new(shape: ActiveClusterOperationCount)
+
     AddAssociationRequest.add_member(:source_arn, Shapes::ShapeRef.new(shape: AssociationEntityArn, required: true, location_name: "SourceArn"))
     AddAssociationRequest.add_member(:destination_arn, Shapes::ShapeRef.new(shape: AssociationEntityArn, required: true, location_name: "DestinationArn"))
     AddAssociationRequest.add_member(:association_type, Shapes::ShapeRef.new(shape: AssociationEdgeType, location_name: "AssociationType"))
@@ -3516,6 +3543,10 @@ module Aws::SageMaker
     ClusterAutoScalingConfigOutput.add_member(:failure_message, Shapes::ShapeRef.new(shape: String, location_name: "FailureMessage"))
     ClusterAutoScalingConfigOutput.struct_class = Types::ClusterAutoScalingConfigOutput
 
+    ClusterCapacityRequirements.add_member(:spot, Shapes::ShapeRef.new(shape: ClusterSpotOptions, location_name: "Spot"))
+    ClusterCapacityRequirements.add_member(:on_demand, Shapes::ShapeRef.new(shape: ClusterOnDemandOptions, location_name: "OnDemand"))
+    ClusterCapacityRequirements.struct_class = Types::ClusterCapacityRequirements
+
     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}))
@@ -3546,6 +3577,7 @@ module Aws::SageMaker
 
     ClusterInstanceGroupDetails.add_member(:current_count, Shapes::ShapeRef.new(shape: ClusterNonNegativeInstanceCount, location_name: "CurrentCount"))
     ClusterInstanceGroupDetails.add_member(:target_count, Shapes::ShapeRef.new(shape: ClusterInstanceCount, location_name: "TargetCount"))
+    ClusterInstanceGroupDetails.add_member(:min_count, Shapes::ShapeRef.new(shape: ClusterInstanceCount, location_name: "MinCount"))
     ClusterInstanceGroupDetails.add_member(:instance_group_name, Shapes::ShapeRef.new(shape: ClusterInstanceGroupName, location_name: "InstanceGroupName"))
     ClusterInstanceGroupDetails.add_member(:instance_type, Shapes::ShapeRef.new(shape: ClusterInstanceType, location_name: "InstanceType"))
     ClusterInstanceGroupDetails.add_member(:life_cycle_config, Shapes::ShapeRef.new(shape: ClusterLifeCycleConfig, location_name: "LifeCycleConfig"))
@@ -3560,6 +3592,9 @@ module Aws::SageMaker
     ClusterInstanceGroupDetails.add_member(:scheduled_update_config, Shapes::ShapeRef.new(shape: ScheduledUpdateConfig, location_name: "ScheduledUpdateConfig"))
     ClusterInstanceGroupDetails.add_member(:current_image_id, Shapes::ShapeRef.new(shape: ImageId, location_name: "CurrentImageId"))
     ClusterInstanceGroupDetails.add_member(:desired_image_id, Shapes::ShapeRef.new(shape: ImageId, location_name: "DesiredImageId"))
+    ClusterInstanceGroupDetails.add_member(:active_operations, Shapes::ShapeRef.new(shape: ActiveOperations, location_name: "ActiveOperations"))
+    ClusterInstanceGroupDetails.add_member(:kubernetes_config, Shapes::ShapeRef.new(shape: ClusterKubernetesConfigDetails, location_name: "KubernetesConfig"))
+    ClusterInstanceGroupDetails.add_member(:capacity_requirements, Shapes::ShapeRef.new(shape: ClusterCapacityRequirements, location_name: "CapacityRequirements"))
     ClusterInstanceGroupDetails.add_member(:target_state_count, Shapes::ShapeRef.new(shape: ClusterInstanceCount, location_name: "TargetStateCount"))
     ClusterInstanceGroupDetails.add_member(:software_update_status, Shapes::ShapeRef.new(shape: SoftwareUpdateStatus, location_name: "SoftwareUpdateStatus"))
     ClusterInstanceGroupDetails.add_member(:active_software_update_config, Shapes::ShapeRef.new(shape: DeploymentConfiguration, location_name: "ActiveSoftwareUpdateConfig"))
@@ -3568,6 +3603,7 @@ module Aws::SageMaker
     ClusterInstanceGroupDetailsList.member = Shapes::ShapeRef.new(shape: ClusterInstanceGroupDetails)
 
     ClusterInstanceGroupSpecification.add_member(:instance_count, Shapes::ShapeRef.new(shape: ClusterInstanceCount, required: true, location_name: "InstanceCount"))
+    ClusterInstanceGroupSpecification.add_member(:min_instance_count, Shapes::ShapeRef.new(shape: ClusterInstanceCount, location_name: "MinInstanceCount"))
     ClusterInstanceGroupSpecification.add_member(:instance_group_name, Shapes::ShapeRef.new(shape: ClusterInstanceGroupName, required: true, location_name: "InstanceGroupName"))
     ClusterInstanceGroupSpecification.add_member(:instance_type, Shapes::ShapeRef.new(shape: ClusterInstanceType, required: true, location_name: "InstanceType"))
     ClusterInstanceGroupSpecification.add_member(:life_cycle_config, Shapes::ShapeRef.new(shape: ClusterLifeCycleConfig, required: true, location_name: "LifeCycleConfig"))
@@ -3579,6 +3615,8 @@ module Aws::SageMaker
     ClusterInstanceGroupSpecification.add_member(:override_vpc_config, Shapes::ShapeRef.new(shape: VpcConfig, location_name: "OverrideVpcConfig"))
     ClusterInstanceGroupSpecification.add_member(:scheduled_update_config, Shapes::ShapeRef.new(shape: ScheduledUpdateConfig, location_name: "ScheduledUpdateConfig"))
     ClusterInstanceGroupSpecification.add_member(:image_id, Shapes::ShapeRef.new(shape: ImageId, location_name: "ImageId"))
+    ClusterInstanceGroupSpecification.add_member(:kubernetes_config, Shapes::ShapeRef.new(shape: ClusterKubernetesConfig, location_name: "KubernetesConfig"))
+    ClusterInstanceGroupSpecification.add_member(:capacity_requirements, Shapes::ShapeRef.new(shape: ClusterCapacityRequirements, location_name: "CapacityRequirements"))
     ClusterInstanceGroupSpecification.struct_class = Types::ClusterInstanceGroupSpecification
 
     ClusterInstanceGroupSpecifications.member = Shapes::ShapeRef.new(shape: ClusterInstanceGroupSpecification)
@@ -3601,6 +3639,32 @@ module Aws::SageMaker
 
     ClusterInstanceStorageConfigs.member = Shapes::ShapeRef.new(shape: ClusterInstanceStorageConfig)
 
+    ClusterKubernetesConfig.add_member(:labels, Shapes::ShapeRef.new(shape: ClusterKubernetesLabels, location_name: "Labels"))
+    ClusterKubernetesConfig.add_member(:taints, Shapes::ShapeRef.new(shape: ClusterKubernetesTaints, location_name: "Taints"))
+    ClusterKubernetesConfig.struct_class = Types::ClusterKubernetesConfig
+
+    ClusterKubernetesConfigDetails.add_member(:current_labels, Shapes::ShapeRef.new(shape: ClusterKubernetesLabels, location_name: "CurrentLabels"))
+    ClusterKubernetesConfigDetails.add_member(:desired_labels, Shapes::ShapeRef.new(shape: ClusterKubernetesLabels, location_name: "DesiredLabels"))
+    ClusterKubernetesConfigDetails.add_member(:current_taints, Shapes::ShapeRef.new(shape: ClusterKubernetesTaints, location_name: "CurrentTaints"))
+    ClusterKubernetesConfigDetails.add_member(:desired_taints, Shapes::ShapeRef.new(shape: ClusterKubernetesTaints, location_name: "DesiredTaints"))
+    ClusterKubernetesConfigDetails.struct_class = Types::ClusterKubernetesConfigDetails
+
+    ClusterKubernetesConfigNodeDetails.add_member(:current_labels, Shapes::ShapeRef.new(shape: ClusterKubernetesLabels, location_name: "CurrentLabels"))
+    ClusterKubernetesConfigNodeDetails.add_member(:desired_labels, Shapes::ShapeRef.new(shape: ClusterKubernetesLabels, location_name: "DesiredLabels"))
+    ClusterKubernetesConfigNodeDetails.add_member(:current_taints, Shapes::ShapeRef.new(shape: ClusterKubernetesTaints, location_name: "CurrentTaints"))
+    ClusterKubernetesConfigNodeDetails.add_member(:desired_taints, Shapes::ShapeRef.new(shape: ClusterKubernetesTaints, location_name: "DesiredTaints"))
+    ClusterKubernetesConfigNodeDetails.struct_class = Types::ClusterKubernetesConfigNodeDetails
+
+    ClusterKubernetesLabels.key = Shapes::ShapeRef.new(shape: ClusterKubernetesLabelKey)
+    ClusterKubernetesLabels.value = Shapes::ShapeRef.new(shape: ClusterKubernetesLabelValue)
+
+    ClusterKubernetesTaint.add_member(:key, Shapes::ShapeRef.new(shape: ClusterKubernetesTaintKey, required: true, location_name: "Key"))
+    ClusterKubernetesTaint.add_member(:value, Shapes::ShapeRef.new(shape: ClusterKubernetesTaintValue, location_name: "Value"))
+    ClusterKubernetesTaint.add_member(:effect, Shapes::ShapeRef.new(shape: ClusterKubernetesTaintEffect, required: true, location_name: "Effect"))
+    ClusterKubernetesTaint.struct_class = Types::ClusterKubernetesTaint
+
+    ClusterKubernetesTaints.member = Shapes::ShapeRef.new(shape: ClusterKubernetesTaint)
+
     ClusterLifeCycleConfig.add_member(:source_s3_uri, Shapes::ShapeRef.new(shape: S3Uri, required: true, location_name: "SourceS3Uri"))
     ClusterLifeCycleConfig.add_member(:on_create, Shapes::ShapeRef.new(shape: ClusterLifeCycleConfigFileName, required: true, location_name: "OnCreate"))
     ClusterLifeCycleConfig.struct_class = Types::ClusterLifeCycleConfig
@@ -3628,6 +3692,8 @@ module Aws::SageMaker
     ClusterNodeDetails.add_member(:current_image_id, Shapes::ShapeRef.new(shape: ImageId, location_name: "CurrentImageId"))
     ClusterNodeDetails.add_member(:desired_image_id, Shapes::ShapeRef.new(shape: ImageId, location_name: "DesiredImageId"))
     ClusterNodeDetails.add_member(:ultra_server_info, Shapes::ShapeRef.new(shape: UltraServerInfo, location_name: "UltraServerInfo"))
+    ClusterNodeDetails.add_member(:kubernetes_config, Shapes::ShapeRef.new(shape: ClusterKubernetesConfigNodeDetails, location_name: "KubernetesConfig"))
+    ClusterNodeDetails.add_member(:capacity_type, Shapes::ShapeRef.new(shape: ClusterCapacityType, location_name: "CapacityType"))
     ClusterNodeDetails.struct_class = Types::ClusterNodeDetails
 
     ClusterNodeIds.member = Shapes::ShapeRef.new(shape: ClusterNodeId)
@@ -3647,6 +3713,8 @@ module Aws::SageMaker
     ClusterNodeSummary.add_member(:private_dns_hostname, Shapes::ShapeRef.new(shape: ClusterPrivateDnsHostname, location_name: "PrivateDnsHostname"))
     ClusterNodeSummary.struct_class = Types::ClusterNodeSummary
 
+    ClusterOnDemandOptions.struct_class = Types::ClusterOnDemandOptions
+
     ClusterOrchestrator.add_member(:eks, Shapes::ShapeRef.new(shape: ClusterOrchestratorEksConfig, required: true, location_name: "Eks"))
     ClusterOrchestrator.struct_class = Types::ClusterOrchestrator
 
@@ -3698,6 +3766,8 @@ module Aws::SageMaker
 
     ClusterSchedulerConfigSummaryList.member = Shapes::ShapeRef.new(shape: ClusterSchedulerConfigSummary)
 
+    ClusterSpotOptions.struct_class = Types::ClusterSpotOptions
+
     ClusterSummaries.member = Shapes::ShapeRef.new(shape: ClusterSummary)
 
     ClusterSummary.add_member(:cluster_arn, Shapes::ShapeRef.new(shape: ClusterArn, required: true, location_name: "ClusterArn"))
@@ -4471,6 +4541,7 @@ module Aws::SageMaker
     CreateOptimizationJobRequest.add_member(:role_arn, Shapes::ShapeRef.new(shape: RoleArn, required: true, location_name: "RoleArn"))
     CreateOptimizationJobRequest.add_member(:model_source, Shapes::ShapeRef.new(shape: OptimizationJobModelSource, required: true, location_name: "ModelSource"))
     CreateOptimizationJobRequest.add_member(:deployment_instance_type, Shapes::ShapeRef.new(shape: OptimizationJobDeploymentInstanceType, required: true, location_name: "DeploymentInstanceType"))
+    CreateOptimizationJobRequest.add_member(:max_instance_count, Shapes::ShapeRef.new(shape: OptimizationJobMaxInstanceCount, location_name: "MaxInstanceCount"))
     CreateOptimizationJobRequest.add_member(:optimization_environment, Shapes::ShapeRef.new(shape: OptimizationJobEnvironmentVariables, location_name: "OptimizationEnvironment"))
     CreateOptimizationJobRequest.add_member(:optimization_configs, Shapes::ShapeRef.new(shape: OptimizationConfigs, required: true, location_name: "OptimizationConfigs"))
     CreateOptimizationJobRequest.add_member(:output_config, Shapes::ShapeRef.new(shape: OptimizationJobOutputConfig, required: true, location_name: "OutputConfig"))
@@ -6071,6 +6142,7 @@ module Aws::SageMaker
     DescribeOptimizationJobResponse.add_member(:model_source, Shapes::ShapeRef.new(shape: OptimizationJobModelSource, required: true, location_name: "ModelSource"))
     DescribeOptimizationJobResponse.add_member(:optimization_environment, Shapes::ShapeRef.new(shape: OptimizationJobEnvironmentVariables, location_name: "OptimizationEnvironment"))
     DescribeOptimizationJobResponse.add_member(:deployment_instance_type, Shapes::ShapeRef.new(shape: OptimizationJobDeploymentInstanceType, required: true, location_name: "DeploymentInstanceType"))
+    DescribeOptimizationJobResponse.add_member(:max_instance_count, Shapes::ShapeRef.new(shape: OptimizationJobMaxInstanceCount, location_name: "MaxInstanceCount"))
     DescribeOptimizationJobResponse.add_member(:optimization_configs, Shapes::ShapeRef.new(shape: OptimizationConfigs, required: true, location_name: "OptimizationConfigs"))
     DescribeOptimizationJobResponse.add_member(:output_config, Shapes::ShapeRef.new(shape: OptimizationJobOutputConfig, required: true, location_name: "OutputConfig"))
     DescribeOptimizationJobResponse.add_member(:optimization_output, Shapes::ShapeRef.new(shape: OptimizationOutput, location_name: "OptimizationOutput"))
@@ -7561,6 +7633,7 @@ module Aws::SageMaker
 
     InstanceGroupScalingMetadata.add_member(:instance_count, Shapes::ShapeRef.new(shape: InstanceCount, location_name: "InstanceCount"))
     InstanceGroupScalingMetadata.add_member(:target_count, Shapes::ShapeRef.new(shape: TargetCount, location_name: "TargetCount"))
+    InstanceGroupScalingMetadata.add_member(:min_count, Shapes::ShapeRef.new(shape: InstanceCount, location_name: "MinCount"))
     InstanceGroupScalingMetadata.add_member(:failure_message, Shapes::ShapeRef.new(shape: String, location_name: "FailureMessage"))
     InstanceGroupScalingMetadata.struct_class = Types::InstanceGroupScalingMetadata
 
@@ -9319,6 +9392,14 @@ module Aws::SageMaker
     ModelShardingConfig.add_member(:override_environment, Shapes::ShapeRef.new(shape: OptimizationJobEnvironmentVariables, location_name: "OverrideEnvironment"))
     ModelShardingConfig.struct_class = Types::ModelShardingConfig
 
+    ModelSpeculativeDecodingConfig.add_member(:technique, Shapes::ShapeRef.new(shape: ModelSpeculativeDecodingTechnique, required: true, location_name: "Technique"))
+    ModelSpeculativeDecodingConfig.add_member(:training_data_source, Shapes::ShapeRef.new(shape: ModelSpeculativeDecodingTrainingDataSource, location_name: "TrainingDataSource"))
+    ModelSpeculativeDecodingConfig.struct_class = Types::ModelSpeculativeDecodingConfig
+
+    ModelSpeculativeDecodingTrainingDataSource.add_member(:s3_uri, Shapes::ShapeRef.new(shape: S3Uri, required: true, location_name: "S3Uri"))
+    ModelSpeculativeDecodingTrainingDataSource.add_member(:s3_data_type, Shapes::ShapeRef.new(shape: ModelSpeculativeDecodingS3DataType, required: true, location_name: "S3DataType"))
+    ModelSpeculativeDecodingTrainingDataSource.struct_class = Types::ModelSpeculativeDecodingTrainingDataSource
+
     ModelStepMetadata.add_member(:arn, Shapes::ShapeRef.new(shape: String256, location_name: "Arn"))
     ModelStepMetadata.struct_class = Types::ModelStepMetadata
 
@@ -9630,10 +9711,12 @@ module Aws::SageMaker
     OptimizationConfig.add_member(:model_quantization_config, Shapes::ShapeRef.new(shape: ModelQuantizationConfig, location_name: "ModelQuantizationConfig"))
     OptimizationConfig.add_member(:model_compilation_config, Shapes::ShapeRef.new(shape: ModelCompilationConfig, location_name: "ModelCompilationConfig"))
     OptimizationConfig.add_member(:model_sharding_config, Shapes::ShapeRef.new(shape: ModelShardingConfig, location_name: "ModelShardingConfig"))
+    OptimizationConfig.add_member(:model_speculative_decoding_config, Shapes::ShapeRef.new(shape: ModelSpeculativeDecodingConfig, location_name: "ModelSpeculativeDecodingConfig"))
     OptimizationConfig.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
     OptimizationConfig.add_member_subclass(:model_quantization_config, Types::OptimizationConfig::ModelQuantizationConfig)
     OptimizationConfig.add_member_subclass(:model_compilation_config, Types::OptimizationConfig::ModelCompilationConfig)
     OptimizationConfig.add_member_subclass(:model_sharding_config, Types::OptimizationConfig::ModelShardingConfig)
+    OptimizationConfig.add_member_subclass(:model_speculative_decoding_config, Types::OptimizationConfig::ModelSpeculativeDecodingConfig)
     OptimizationConfig.add_member_subclass(:unknown, Types::OptimizationConfig::Unknown)
     OptimizationConfig.struct_class = Types::OptimizationConfig
 
@@ -9643,6 +9726,7 @@ module Aws::SageMaker
     OptimizationJobEnvironmentVariables.value = Shapes::ShapeRef.new(shape: String256)
 
     OptimizationJobModelSource.add_member(:s3, Shapes::ShapeRef.new(shape: OptimizationJobModelSourceS3, location_name: "S3"))
+    OptimizationJobModelSource.add_member(:sage_maker_model, Shapes::ShapeRef.new(shape: OptimizationSageMakerModel, location_name: "SageMakerModel"))
     OptimizationJobModelSource.struct_class = Types::OptimizationJobModelSource
 
     OptimizationJobModelSourceS3.add_member(:s3_uri, Shapes::ShapeRef.new(shape: S3Uri, location_name: "S3Uri"))
@@ -9651,6 +9735,7 @@ module Aws::SageMaker
 
     OptimizationJobOutputConfig.add_member(:kms_key_id, Shapes::ShapeRef.new(shape: KmsKeyId, location_name: "KmsKeyId"))
     OptimizationJobOutputConfig.add_member(:s3_output_location, Shapes::ShapeRef.new(shape: S3Uri, required: true, location_name: "S3OutputLocation"))
+    OptimizationJobOutputConfig.add_member(:sage_maker_model, Shapes::ShapeRef.new(shape: OptimizationSageMakerModel, location_name: "SageMakerModel"))
     OptimizationJobOutputConfig.struct_class = Types::OptimizationJobOutputConfig
 
     OptimizationJobSummaries.member = Shapes::ShapeRef.new(shape: OptimizationJobSummary)
@@ -9663,6 +9748,7 @@ module Aws::SageMaker
     OptimizationJobSummary.add_member(:optimization_end_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "OptimizationEndTime"))
     OptimizationJobSummary.add_member(:last_modified_time, Shapes::ShapeRef.new(shape: LastModifiedTime, location_name: "LastModifiedTime"))
     OptimizationJobSummary.add_member(:deployment_instance_type, Shapes::ShapeRef.new(shape: OptimizationJobDeploymentInstanceType, required: true, location_name: "DeploymentInstanceType"))
+    OptimizationJobSummary.add_member(:max_instance_count, Shapes::ShapeRef.new(shape: OptimizationJobMaxInstanceCount, location_name: "MaxInstanceCount"))
     OptimizationJobSummary.add_member(:optimization_types, Shapes::ShapeRef.new(shape: OptimizationTypes, required: true, location_name: "OptimizationTypes"))
     OptimizationJobSummary.struct_class = Types::OptimizationJobSummary
 
@@ -9672,6 +9758,9 @@ module Aws::SageMaker
     OptimizationOutput.add_member(:recommended_inference_image, Shapes::ShapeRef.new(shape: OptimizationContainerImage, location_name: "RecommendedInferenceImage"))
     OptimizationOutput.struct_class = Types::OptimizationOutput
 
+    OptimizationSageMakerModel.add_member(:model_name, Shapes::ShapeRef.new(shape: ModelName, location_name: "ModelName"))
+    OptimizationSageMakerModel.struct_class = Types::OptimizationSageMakerModel
+
     OptimizationTypes.member = Shapes::ShapeRef.new(shape: OptimizationType)
 
     OptimizationVpcConfig.add_member(:security_group_ids, Shapes::ShapeRef.new(shape: OptimizationVpcSecurityGroupIds, required: true, location_name: "SecurityGroupIds"))