aws-sdk-sagemaker 1.397.0 → 1.399.0

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

@@ -4988,6 +4988,31 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # The billable token usage for a training session, broken down by
+    # operation type.
+    #
+    # @!attribute [rw] prefill_token_count
+    #   The number of prefill tokens consumed.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] sample_token_count
+    #   The number of sample tokens consumed.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] training_token_count
+    #   The number of training tokens consumed.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/BillableTokenUsage AWS API Documentation
+    #
+    class BillableTokenUsage < Struct.new(
+      :prefill_token_count,
+      :sample_token_count,
+      :training_token_count)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Update policy for a blue/green deployment. If this update policy is
     # specified, SageMaker creates a new fleet during the deployment while
     # maintaining the old fleet. SageMaker flips traffic to the new fleet
@@ -10045,7 +10070,9 @@ module Aws::SageMaker
     end
 
     # @!attribute [rw] name
-    #   Name for the cluster policy.
+    #   The name for the cluster policy. The name must be unique within the
+    #   SageMaker AI HyperPod cluster specified by `ClusterArn`. You can use
+    #   the same name in other clusters within a Region or across Regions.
     #   @return [String]
     #
     # @!attribute [rw] cluster_arn
@@ -10250,7 +10277,10 @@ module Aws::SageMaker
     end
 
     # @!attribute [rw] name
-    #   Name to the compute allocation definition.
+    #   The name of the compute allocation definition. The name must be
+    #   unique within the SageMaker AI HyperPod cluster specified by
+    #   `ClusterArn`. You can use the same name in other clusters within a
+    #   Region or across Regions.
     #   @return [String]
     #
     # @!attribute [rw] description
@@ -14713,6 +14743,79 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] training_session_name
+    #   The name of the training session.
+    #   @return [String]
+    #
+    # @!attribute [rw] model_package_group_arn
+    #   The Amazon Resource Name (ARN) of the model package group where
+    #   checkpoints and fine-tuned model artifacts are published.
+    #   @return [String]
+    #
+    # @!attribute [rw] lora_config
+    #   The Low-Rank Adaptation (LoRA) configuration for the training
+    #   session, including the rank and alpha parameters.
+    #   @return [Types::LoraConfig]
+    #
+    # @!attribute [rw] role_arn
+    #   The Amazon Resource Name (ARN) of the IAM role that Amazon SageMaker
+    #   assumes to access resources on your behalf. The role must have
+    #   permissions to access the output S3 path and model package group.
+    #   @return [String]
+    #
+    # @!attribute [rw] vpc_config
+    #   The Virtual Private Cloud (VPC) configuration for the training
+    #   session. Specify subnets and security groups to run the training
+    #   session in your VPC.
+    #   @return [Types::VpcConfig]
+    #
+    # @!attribute [rw] model_config
+    #   The configuration for the base model, including the model ARN and
+    #   end-user license agreement (EULA) acceptance.
+    #   @return [Types::TrainingSessionModelConfig]
+    #
+    # @!attribute [rw] output_config
+    #   The output configuration for the training session, including the S3
+    #   location and optional encryption settings.
+    #   @return [Types::TrainingSessionOutputConfig]
+    #
+    # @!attribute [rw] tags
+    #   An array of key-value pairs to apply to the training session as
+    #   tags. For more information, see [Tagging Amazon Web Services
+    #   Resources][1].
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/general/latest/gr/aws_tagging.html
+    #   @return [Array<Types::Tag>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreateTrainingSessionRequest AWS API Documentation
+    #
+    class CreateTrainingSessionRequest < Struct.new(
+      :training_session_name,
+      :model_package_group_arn,
+      :lora_config,
+      :role_arn,
+      :vpc_config,
+      :model_config,
+      :output_config,
+      :tags)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] training_session_arn
+    #   The Amazon Resource Name (ARN) of the training session.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreateTrainingSessionResponse AWS API Documentation
+    #
+    class CreateTrainingSessionResponse < Struct.new(
+      :training_session_arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] transform_job_name
     #   The name of the transform job. The name must be unique within an
     #   Amazon Web Services Region in an Amazon Web Services account.
@@ -17053,6 +17156,22 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] training_session_name
+    #   The name of the training session to delete.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DeleteTrainingSessionRequest AWS API Documentation
+    #
+    class DeleteTrainingSessionRequest < Struct.new(
+      :training_session_name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DeleteTrainingSessionResponse AWS API Documentation
+    #
+    class DeleteTrainingSessionResponse < Aws::EmptyStructure; end
+
     # @!attribute [rw] trial_component_name
     #   The name of the component to delete.
     #   @return [String]
@@ -24626,6 +24745,30 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] training_session_name
+    #   The name of the training session to describe.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeTrainingSessionRequest AWS API Documentation
+    #
+    class DescribeTrainingSessionRequest < Struct.new(
+      :training_session_name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] training_session_detail
+    #   The detailed information about the training session.
+    #   @return [Types::TrainingSessionDetail]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeTrainingSessionResponse AWS API Documentation
+    #
+    class DescribeTrainingSessionResponse < Struct.new(
+      :training_session_detail)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] transform_job_name
     #   The name of the transform job that you want to view details of.
     #   @return [String]
@@ -40294,6 +40437,67 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] next_token
+    #   If the response was truncated, use the token returned in the
+    #   previous response to retrieve the next page of results.
+    #   @return [String]
+    #
+    # @!attribute [rw] max_results
+    #   The maximum number of training sessions to return in the response.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] creation_time_after
+    #   A filter that returns only training sessions created after the
+    #   specified time.
+    #   @return [Time]
+    #
+    # @!attribute [rw] creation_time_before
+    #   A filter that returns only training sessions created before the
+    #   specified time.
+    #   @return [Time]
+    #
+    # @!attribute [rw] name_contains
+    #   A string in the training session name to filter results. Only
+    #   sessions whose name contains the specified string are returned.
+    #   @return [String]
+    #
+    # @!attribute [rw] status_equals
+    #   A filter that returns only training sessions with the specified
+    #   status.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListTrainingSessionsRequest AWS API Documentation
+    #
+    class ListTrainingSessionsRequest < Struct.new(
+      :next_token,
+      :max_results,
+      :creation_time_after,
+      :creation_time_before,
+      :name_contains,
+      :status_equals)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] next_token
+    #   If the response is truncated, this token retrieves the next set of
+    #   results.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_session_summaries
+    #   An array of `TrainingSessionSummary` objects that provide summary
+    #   information about the training sessions.
+    #   @return [Array<Types::TrainingSessionSummary>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListTrainingSessionsResponse AWS API Documentation
+    #
+    class ListTrainingSessionsResponse < Struct.new(
+      :next_token,
+      :training_session_summaries)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] creation_time_after
     #   A filter that returns only transform jobs created after the
     #   specified time.
@@ -40739,6 +40943,26 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # The Low-Rank Adaptation (LoRA) configuration for a training session,
+    # including the rank and alpha parameters.
+    #
+    # @!attribute [rw] rank
+    #   The rank of the Low-Rank Adaptation (LoRA) adapter matrices.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] alpha
+    #   The Low-Rank Adaptation (LoRA) scaling factor.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/LoraConfig AWS API Documentation
+    #
+    class LoraConfig < Struct.new(
+      :rank,
+      :alpha)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The MLflow configuration.
     #
     # @!attribute [rw] mlflow_resource_arn
@@ -52995,6 +53219,22 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] training_session_name
+    #   The name of the training session to stop.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/StopTrainingSessionRequest AWS API Documentation
+    #
+    class StopTrainingSessionRequest < Struct.new(
+      :training_session_name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/StopTrainingSessionResponse AWS API Documentation
+    #
+    class StopTrainingSessionResponse < Aws::EmptyStructure; end
+
     # @!attribute [rw] transform_job_name
     #   The name of the batch transform job to stop.
     #   @return [String]
@@ -55211,6 +55451,190 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Contains detailed information about a training session.
+    #
+    # @!attribute [rw] training_session_arn
+    #   The Amazon Resource Name (ARN) of the training session.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_session_name
+    #   The name of the training session.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_session_status
+    #   The current status of the training session.
+    #   @return [String]
+    #
+    # @!attribute [rw] model_package_group_arn
+    #   The Amazon Resource Name (ARN) of the model package group where
+    #   checkpoints are published.
+    #   @return [String]
+    #
+    # @!attribute [rw] creation_time
+    #   The date and time that the training session was created.
+    #   @return [Time]
+    #
+    # @!attribute [rw] last_modified_time
+    #   The date and time that the training session was last modified.
+    #   @return [Time]
+    #
+    # @!attribute [rw] end_time
+    #   The date and time that the training session ended.
+    #   @return [Time]
+    #
+    # @!attribute [rw] error_details
+    #   If the training session failed or is blocked, detailed error
+    #   information.
+    #   @return [String]
+    #
+    # @!attribute [rw] last_received_sequence_number
+    #   The sequence number of the last training operation received by the
+    #   session.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] billable_token_usage
+    #   The billable token usage for the training session.
+    #   @return [Types::BillableTokenUsage]
+    #
+    # @!attribute [rw] vpc_config
+    #   The Virtual Private Cloud (VPC) configuration for the training
+    #   session.
+    #   @return [Types::VpcConfig]
+    #
+    # @!attribute [rw] role_arn
+    #   The Amazon Resource Name (ARN) of the IAM role associated with the
+    #   training session.
+    #   @return [String]
+    #
+    # @!attribute [rw] lora_config
+    #   The Low-Rank Adaptation (LoRA) configuration for the training
+    #   session.
+    #   @return [Types::LoraConfig]
+    #
+    # @!attribute [rw] model_config
+    #   The configuration for the base model.
+    #   @return [Types::TrainingSessionModelConfig]
+    #
+    # @!attribute [rw] output_config
+    #   The output configuration for the training session.
+    #   @return [Types::TrainingSessionOutputConfig]
+    #
+    # @!attribute [rw] tags
+    #   The tags associated with the training session.
+    #   @return [Array<Types::Tag>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingSessionDetail AWS API Documentation
+    #
+    class TrainingSessionDetail < Struct.new(
+      :training_session_arn,
+      :training_session_name,
+      :training_session_status,
+      :model_package_group_arn,
+      :creation_time,
+      :last_modified_time,
+      :end_time,
+      :error_details,
+      :last_received_sequence_number,
+      :billable_token_usage,
+      :vpc_config,
+      :role_arn,
+      :lora_config,
+      :model_config,
+      :output_config,
+      :tags)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The configuration for the base model used in a training session.
+    #
+    # @!attribute [rw] base_model_arn
+    #   The Amazon Resource Name (ARN) of the base foundation model from
+    #   SageMaker Hub.
+    #   @return [String]
+    #
+    # @!attribute [rw] accept_eula
+    #   Specifies whether you accept the end-user license agreement (EULA)
+    #   for gated models.
+    #   @return [Boolean]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingSessionModelConfig AWS API Documentation
+    #
+    class TrainingSessionModelConfig < Struct.new(
+      :base_model_arn,
+      :accept_eula)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The output configuration for a training session, including the Amazon
+    # S3 location for checkpoints and model artifacts, and optional
+    # encryption settings.
+    #
+    # @!attribute [rw] s3_location
+    #   The Amazon S3 location where training session outputs, including
+    #   checkpoints and model artifacts, are stored.
+    #   @return [Types::TrainingSessionS3Location]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingSessionOutputConfig AWS API Documentation
+    #
+    class TrainingSessionOutputConfig < Struct.new(
+      :s3_location)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Specifies the Amazon S3 path and optional encryption settings for
+    # training session output data, including checkpoints, model artifacts,
+    # and invocation results.
+    #
+    # @!attribute [rw] path
+    #   The Amazon S3 path where training session outputs are stored.
+    #   @return [String]
+    #
+    # @!attribute [rw] kms_key_arn
+    #   The Amazon Resource Name (ARN) of the KMS key used to encrypt output
+    #   data. If not specified, the default Amazon S3 encryption is used.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingSessionS3Location AWS API Documentation
+    #
+    class TrainingSessionS3Location < Struct.new(
+      :path,
+      :kms_key_arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains summary information about a training session.
+    #
+    # @!attribute [rw] training_session_arn
+    #   The Amazon Resource Name (ARN) of the training session.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_session_status
+    #   The current status of the training session.
+    #   @return [String]
+    #
+    # @!attribute [rw] creation_time
+    #   The date and time that the training session was created.
+    #   @return [Time]
+    #
+    # @!attribute [rw] last_modified_time
+    #   The date and time that the training session was last modified.
+    #   @return [Time]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingSessionSummary AWS API Documentation
+    #
+    class TrainingSessionSummary < Struct.new(
+      :training_session_arn,
+      :training_session_status,
+      :creation_time,
+      :last_modified_time)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Defines how the algorithm is used for a training job.
     #
     # @!attribute [rw] training_image

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.397.0'
+  GEM_VERSION = '1.399.0'
 
 end
 

data/sig/client.rbs

@@ -166,7 +166,7 @@ module Aws
                                          instance_group_name: ::String,
                                          increment_target_count_by: ::Integer,
                                          availability_zones: Array[::String]?,
-                                         instance_types: Array[("ml.p4d.24xlarge" | "ml.p4de.24xlarge" | "ml.p5.48xlarge" | "ml.p5.4xlarge" | "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" | "ml.r5d.16xlarge" | "ml.g7e.2xlarge" | "ml.g7e.4xlarge" | "ml.g7e.8xlarge" | "ml.g7e.12xlarge" | "ml.g7e.24xlarge" | "ml.g7e.48xlarge" | "ml.p6-b300.48xlarge" | "ml.g4dn.xlarge" | "ml.g4dn.2xlarge" | "ml.g4dn.4xlarge" | "ml.g4dn.8xlarge" | "ml.g4dn.12xlarge" | "ml.g4dn.16xlarge" | "ml.c6g.medium" | "ml.c6g.large" | "ml.c6g.xlarge" | "ml.c6g.2xlarge" | "ml.c6g.4xlarge" | "ml.c6g.8xlarge" | "ml.c6g.12xlarge" | "ml.c6g.16xlarge" | "ml.c7g.medium" | "ml.c7g.large" | "ml.c7g.xlarge" | "ml.c7g.2xlarge" | "ml.c7g.4xlarge" | "ml.c7g.8xlarge" | "ml.c7g.12xlarge" | "ml.c7g.16xlarge" | "ml.c8g.medium" | "ml.c8g.large" | "ml.c8g.xlarge" | "ml.c8g.2xlarge" | "ml.c8g.4xlarge" | "ml.c8g.8xlarge" | "ml.c8g.12xlarge" | "ml.c8g.16xlarge" | "ml.c8g.24xlarge" | "ml.c8g.48xlarge" | "ml.c6a.large" | "ml.c6a.xlarge" | "ml.c6a.2xlarge" | "ml.c6a.4xlarge" | "ml.c6a.8xlarge" | "ml.c6a.12xlarge" | "ml.c6a.16xlarge" | "ml.c6a.24xlarge" | "ml.c6a.32xlarge" | "ml.c6a.48xlarge" | "ml.m6a.large" | "ml.m6a.xlarge" | "ml.m6a.2xlarge" | "ml.m6a.4xlarge" | "ml.m6a.8xlarge" | "ml.m6a.12xlarge" | "ml.m6a.16xlarge" | "ml.m6a.24xlarge" | "ml.m6a.32xlarge" | "ml.m6a.48xlarge" | "ml.m6g.medium" | "ml.m6g.large" | "ml.m6g.xlarge" | "ml.m6g.2xlarge" | "ml.m6g.4xlarge" | "ml.m6g.8xlarge" | "ml.m6g.12xlarge" | "ml.m6g.16xlarge" | "ml.m7g.medium" | "ml.m7g.large" | "ml.m7g.xlarge" | "ml.m7g.2xlarge" | "ml.m7g.4xlarge" | "ml.m7g.8xlarge" | "ml.m7g.12xlarge" | "ml.m7g.16xlarge" | "ml.m8g.medium" | "ml.m8g.large" | "ml.m8g.xlarge" | "ml.m8g.2xlarge" | "ml.m8g.4xlarge" | "ml.m8g.8xlarge" | "ml.m8g.12xlarge" | "ml.m8g.16xlarge" | "ml.m8g.24xlarge" | "ml.m8g.48xlarge" | "ml.g7.2xlarge" | "ml.g7.4xlarge" | "ml.g7.8xlarge" | "ml.g7.12xlarge" | "ml.g7.24xlarge" | "ml.g7.48xlarge")]?
+                                         instance_types: Array[("ml.p4d.24xlarge" | "ml.p4de.24xlarge" | "ml.p5.48xlarge" | "ml.p5.4xlarge" | "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" | "ml.r5d.16xlarge" | "ml.g7e.2xlarge" | "ml.g7e.4xlarge" | "ml.g7e.8xlarge" | "ml.g7e.12xlarge" | "ml.g7e.24xlarge" | "ml.g7e.48xlarge" | "ml.p6-b300.48xlarge" | "ml.g4dn.xlarge" | "ml.g4dn.2xlarge" | "ml.g4dn.4xlarge" | "ml.g4dn.8xlarge" | "ml.g4dn.12xlarge" | "ml.g4dn.16xlarge" | "ml.c6g.medium" | "ml.c6g.large" | "ml.c6g.xlarge" | "ml.c6g.2xlarge" | "ml.c6g.4xlarge" | "ml.c6g.8xlarge" | "ml.c6g.12xlarge" | "ml.c6g.16xlarge" | "ml.c7g.medium" | "ml.c7g.large" | "ml.c7g.xlarge" | "ml.c7g.2xlarge" | "ml.c7g.4xlarge" | "ml.c7g.8xlarge" | "ml.c7g.12xlarge" | "ml.c7g.16xlarge" | "ml.c8g.medium" | "ml.c8g.large" | "ml.c8g.xlarge" | "ml.c8g.2xlarge" | "ml.c8g.4xlarge" | "ml.c8g.8xlarge" | "ml.c8g.12xlarge" | "ml.c8g.16xlarge" | "ml.c8g.24xlarge" | "ml.c8g.48xlarge" | "ml.c6a.large" | "ml.c6a.xlarge" | "ml.c6a.2xlarge" | "ml.c6a.4xlarge" | "ml.c6a.8xlarge" | "ml.c6a.12xlarge" | "ml.c6a.16xlarge" | "ml.c6a.24xlarge" | "ml.c6a.32xlarge" | "ml.c6a.48xlarge" | "ml.m6a.large" | "ml.m6a.xlarge" | "ml.m6a.2xlarge" | "ml.m6a.4xlarge" | "ml.m6a.8xlarge" | "ml.m6a.12xlarge" | "ml.m6a.16xlarge" | "ml.m6a.24xlarge" | "ml.m6a.32xlarge" | "ml.m6a.48xlarge" | "ml.m6g.medium" | "ml.m6g.large" | "ml.m6g.xlarge" | "ml.m6g.2xlarge" | "ml.m6g.4xlarge" | "ml.m6g.8xlarge" | "ml.m6g.12xlarge" | "ml.m6g.16xlarge" | "ml.m7g.medium" | "ml.m7g.large" | "ml.m7g.xlarge" | "ml.m7g.2xlarge" | "ml.m7g.4xlarge" | "ml.m7g.8xlarge" | "ml.m7g.12xlarge" | "ml.m7g.16xlarge" | "ml.m8g.medium" | "ml.m8g.large" | "ml.m8g.xlarge" | "ml.m8g.2xlarge" | "ml.m8g.4xlarge" | "ml.m8g.8xlarge" | "ml.m8g.12xlarge" | "ml.m8g.16xlarge" | "ml.m8g.24xlarge" | "ml.m8g.48xlarge" | "ml.g7.2xlarge" | "ml.g7.4xlarge" | "ml.g7.8xlarge" | "ml.g7.12xlarge" | "ml.g7.24xlarge" | "ml.g7.48xlarge" | "ml.c8a.16xlarge" | "ml.m8a.16xlarge")]?
                                        }
                                      ]
                                    ) -> _BatchAddClusterNodesResponseSuccess
@@ -2845,6 +2845,42 @@ module Aws
                                 ) -> _CreateTrainingPlanResponseSuccess
                               | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _CreateTrainingPlanResponseSuccess
 
+      interface _CreateTrainingSessionResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::CreateTrainingSessionResponse]
+        def training_session_arn: () -> ::String
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#create_training_session-instance_method
+      def create_training_session: (
+                                     training_session_name: ::String,
+                                     ?model_package_group_arn: ::String,
+                                     lora_config: {
+                                       rank: ::Integer,
+                                       alpha: ::Integer
+                                     },
+                                     role_arn: ::String,
+                                     ?vpc_config: {
+                                       security_group_ids: Array[::String],
+                                       subnets: Array[::String]
+                                     },
+                                     model_config: {
+                                       base_model_arn: ::String,
+                                       accept_eula: bool?
+                                     },
+                                     output_config: {
+                                       s3_location: {
+                                         path: ::String,
+                                         kms_key_arn: ::String?
+                                       }
+                                     },
+                                     ?tags: Array[
+                                       {
+                                         key: ::String,
+                                         value: ::String
+                                       }
+                                     ]
+                                   ) -> _CreateTrainingSessionResponseSuccess
+                                 | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _CreateTrainingSessionResponseSuccess
+
       interface _CreateTransformJobResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::CreateTransformJobResponse]
         def transform_job_arn: () -> ::String
@@ -3486,6 +3522,15 @@ module Aws
                                ) -> ::Seahorse::Client::_ResponseSuccess[::Aws::EmptyStructure]
                              | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> ::Seahorse::Client::_ResponseSuccess[::Aws::EmptyStructure]
 
+      interface _DeleteTrainingSessionResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::DeleteTrainingSessionResponse]
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#delete_training_session-instance_method
+      def delete_training_session: (
+                                     training_session_name: ::String
+                                   ) -> _DeleteTrainingSessionResponseSuccess
+                                 | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DeleteTrainingSessionResponseSuccess
+
       interface _DeleteTrialResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::DeleteTrialResponse]
         def trial_arn: () -> ::String
@@ -5116,6 +5161,16 @@ module Aws
                                                     ) -> _DescribeTrainingPlanExtensionHistoryResponseSuccess
                                                   | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DescribeTrainingPlanExtensionHistoryResponseSuccess
 
+      interface _DescribeTrainingSessionResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::DescribeTrainingSessionResponse]
+        def training_session_detail: () -> Types::TrainingSessionDetail
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#describe_training_session-instance_method
+      def describe_training_session: (
+                                       training_session_name: ::String
+                                     ) -> _DescribeTrainingSessionResponseSuccess
+                                   | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DescribeTrainingSessionResponseSuccess
+
       interface _DescribeTransformJobResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::DescribeTransformJobResponse]
         def transform_job_name: () -> ::String
@@ -6906,6 +6961,22 @@ module Aws
                                ) -> _ListTrainingPlansResponseSuccess
                              | (?Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _ListTrainingPlansResponseSuccess
 
+      interface _ListTrainingSessionsResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::ListTrainingSessionsResponse]
+        def next_token: () -> ::String
+        def training_session_summaries: () -> ::Array[Types::TrainingSessionSummary]
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#list_training_sessions-instance_method
+      def list_training_sessions: (
+                                    ?next_token: ::String,
+                                    ?max_results: ::Integer,
+                                    ?creation_time_after: ::Time,
+                                    ?creation_time_before: ::Time,
+                                    ?name_contains: ::String,
+                                    ?status_equals: ("Created" | "Blocked" | "Running" | "Failed" | "Scheduled" | "Stopping" | "Stopped" | "Expired" | "Deleting" | "DeletionFailed")
+                                  ) -> _ListTrainingSessionsResponseSuccess
+                                | (?Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _ListTrainingSessionsResponseSuccess
+
       interface _ListTransformJobsResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::ListTransformJobsResponse]
         def transform_job_summaries: () -> ::Array[Types::TransformJobSummary]
@@ -7422,6 +7493,15 @@ module Aws
                              ) -> ::Seahorse::Client::_ResponseSuccess[::Aws::EmptyStructure]
                            | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> ::Seahorse::Client::_ResponseSuccess[::Aws::EmptyStructure]
 
+      interface _StopTrainingSessionResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::StopTrainingSessionResponse]
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#stop_training_session-instance_method
+      def stop_training_session: (
+                                   training_session_name: ::String
+                                 ) -> _StopTrainingSessionResponseSuccess
+                               | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _StopTrainingSessionResponseSuccess
+
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#stop_transform_job-instance_method
       def stop_transform_job: (
                                 transform_job_name: ::String