aws-sdk-sagemaker 1.398.0 → 1.400.0

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

@@ -199,6 +199,7 @@ module Aws::SageMaker
     AthenaQueryString = Shapes::StringShape.new(name: 'AthenaQueryString')
     AthenaResultCompressionType = Shapes::StringShape.new(name: 'AthenaResultCompressionType')
     AthenaResultFormat = Shapes::StringShape.new(name: 'AthenaResultFormat')
+    AthenaStepMetadata = Shapes::StructureShape.new(name: 'AthenaStepMetadata')
     AthenaWorkGroup = Shapes::StringShape.new(name: 'AthenaWorkGroup')
     AttachClusterNodeNetworkInterfaceRequest = Shapes::StructureShape.new(name: 'AttachClusterNodeNetworkInterfaceRequest')
     AttachClusterNodeNetworkInterfaceResponse = Shapes::StructureShape.new(name: 'AttachClusterNodeNetworkInterfaceResponse')
@@ -281,6 +282,7 @@ module Aws::SageMaker
     AwsManagedHumanLoopRequestSource = Shapes::StringShape.new(name: 'AwsManagedHumanLoopRequestSource')
     BacktestResultsLocation = Shapes::StringShape.new(name: 'BacktestResultsLocation')
     BaseModel = Shapes::StructureShape.new(name: 'BaseModel')
+    BaseModelArn = Shapes::StringShape.new(name: 'BaseModelArn')
     BaseModelName = Shapes::StringShape.new(name: 'BaseModelName')
     BatchAddClusterNodesError = Shapes::StructureShape.new(name: 'BatchAddClusterNodesError')
     BatchAddClusterNodesErrorCode = Shapes::StringShape.new(name: 'BatchAddClusterNodesErrorCode')
@@ -330,6 +332,7 @@ module Aws::SageMaker
     Bias = Shapes::StructureShape.new(name: 'Bias')
     BillableTimeInSeconds = Shapes::IntegerShape.new(name: 'BillableTimeInSeconds')
     BillableTokenCount = Shapes::IntegerShape.new(name: 'BillableTokenCount')
+    BillableTokenUsage = Shapes::StructureShape.new(name: 'BillableTokenUsage')
     BlockedReason = Shapes::StringShape.new(name: 'BlockedReason')
     BlueGreenUpdatePolicy = Shapes::StructureShape.new(name: 'BlueGreenUpdatePolicy')
     Boolean = Shapes::BooleanShape.new(name: 'Boolean')
@@ -425,6 +428,11 @@ module Aws::SageMaker
     ClarifyTextConfig = Shapes::StructureShape.new(name: 'ClarifyTextConfig')
     ClarifyTextGranularity = Shapes::StringShape.new(name: 'ClarifyTextGranularity')
     ClarifyTextLanguage = Shapes::StringShape.new(name: 'ClarifyTextLanguage')
+    CleanupFailure = Shapes::StructureShape.new(name: 'CleanupFailure')
+    CleanupFailureErrorCode = Shapes::StringShape.new(name: 'CleanupFailureErrorCode')
+    CleanupFailureList = Shapes::ListShape.new(name: 'CleanupFailureList')
+    CleanupStepMetadata = Shapes::StructureShape.new(name: 'CleanupStepMetadata')
+    CleanupStepNameList = Shapes::ListShape.new(name: 'CleanupStepNameList')
     ClientId = Shapes::StringShape.new(name: 'ClientId')
     ClientSecret = Shapes::StringShape.new(name: 'ClientSecret')
     ClientToken = Shapes::StringShape.new(name: 'ClientToken')
@@ -752,6 +760,8 @@ module Aws::SageMaker
     CreateTrainingJobResponse = Shapes::StructureShape.new(name: 'CreateTrainingJobResponse')
     CreateTrainingPlanRequest = Shapes::StructureShape.new(name: 'CreateTrainingPlanRequest')
     CreateTrainingPlanResponse = Shapes::StructureShape.new(name: 'CreateTrainingPlanResponse')
+    CreateTrainingSessionRequest = Shapes::StructureShape.new(name: 'CreateTrainingSessionRequest')
+    CreateTrainingSessionResponse = Shapes::StructureShape.new(name: 'CreateTrainingSessionResponse')
     CreateTransformJobRequest = Shapes::StructureShape.new(name: 'CreateTransformJobRequest')
     CreateTransformJobResponse = Shapes::StructureShape.new(name: 'CreateTransformJobResponse')
     CreateTrialComponentRequest = Shapes::StructureShape.new(name: 'CreateTrialComponentRequest')
@@ -780,6 +790,7 @@ module Aws::SageMaker
     CustomImageContainerEnvironmentVariables = Shapes::MapShape.new(name: 'CustomImageContainerEnvironmentVariables')
     CustomImages = Shapes::ListShape.new(name: 'CustomImages')
     CustomPosixUserConfig = Shapes::StructureShape.new(name: 'CustomPosixUserConfig')
+    CustomerKmsKeyArn = Shapes::StringShape.new(name: 'CustomerKmsKeyArn')
     CustomerMetadataKey = Shapes::StringShape.new(name: 'CustomerMetadataKey')
     CustomerMetadataKeyList = Shapes::ListShape.new(name: 'CustomerMetadataKeyList')
     CustomerMetadataMap = Shapes::MapShape.new(name: 'CustomerMetadataMap')
@@ -894,6 +905,8 @@ module Aws::SageMaker
     DeleteTagsInput = Shapes::StructureShape.new(name: 'DeleteTagsInput')
     DeleteTagsOutput = Shapes::StructureShape.new(name: 'DeleteTagsOutput')
     DeleteTrainingJobRequest = Shapes::StructureShape.new(name: 'DeleteTrainingJobRequest')
+    DeleteTrainingSessionRequest = Shapes::StructureShape.new(name: 'DeleteTrainingSessionRequest')
+    DeleteTrainingSessionResponse = Shapes::StructureShape.new(name: 'DeleteTrainingSessionResponse')
     DeleteTrialComponentRequest = Shapes::StructureShape.new(name: 'DeleteTrialComponentRequest')
     DeleteTrialComponentResponse = Shapes::StructureShape.new(name: 'DeleteTrialComponentResponse')
     DeleteTrialRequest = Shapes::StructureShape.new(name: 'DeleteTrialRequest')
@@ -1057,6 +1070,8 @@ module Aws::SageMaker
     DescribeTrainingPlanExtensionHistoryResponse = Shapes::StructureShape.new(name: 'DescribeTrainingPlanExtensionHistoryResponse')
     DescribeTrainingPlanRequest = Shapes::StructureShape.new(name: 'DescribeTrainingPlanRequest')
     DescribeTrainingPlanResponse = Shapes::StructureShape.new(name: 'DescribeTrainingPlanResponse')
+    DescribeTrainingSessionRequest = Shapes::StructureShape.new(name: 'DescribeTrainingSessionRequest')
+    DescribeTrainingSessionResponse = Shapes::StructureShape.new(name: 'DescribeTrainingSessionResponse')
     DescribeTransformJobRequest = Shapes::StructureShape.new(name: 'DescribeTransformJobRequest')
     DescribeTransformJobResponse = Shapes::StructureShape.new(name: 'DescribeTransformJobResponse')
     DescribeTrialComponentRequest = Shapes::StructureShape.new(name: 'DescribeTrialComponentRequest')
@@ -1131,6 +1146,9 @@ module Aws::SageMaker
     DynamicScalingConfiguration = Shapes::StructureShape.new(name: 'DynamicScalingConfiguration')
     EFSFileSystem = Shapes::StructureShape.new(name: 'EFSFileSystem')
     EFSFileSystemConfig = Shapes::StructureShape.new(name: 'EFSFileSystemConfig')
+    EMRServerlessApplicationId = Shapes::StringShape.new(name: 'EMRServerlessApplicationId')
+    EMRServerlessJobRunId = Shapes::StringShape.new(name: 'EMRServerlessJobRunId')
+    EMRServerlessStepMetadata = Shapes::StructureShape.new(name: 'EMRServerlessStepMetadata')
     EMRStepMetadata = Shapes::StructureShape.new(name: 'EMRStepMetadata')
     EbsStorageSettings = Shapes::StructureShape.new(name: 'EbsStorageSettings')
     Ec2CapacityReservation = Shapes::StructureShape.new(name: 'Ec2CapacityReservation')
@@ -1311,6 +1329,9 @@ module Aws::SageMaker
     FlowDefinitionTaskKeywords = Shapes::ListShape.new(name: 'FlowDefinitionTaskKeywords')
     FlowDefinitionTaskTimeLimitInSeconds = Shapes::IntegerShape.new(name: 'FlowDefinitionTaskTimeLimitInSeconds')
     FlowDefinitionTaskTitle = Shapes::StringShape.new(name: 'FlowDefinitionTaskTitle')
+    ForEachSelectorResult = Shapes::StructureShape.new(name: 'ForEachSelectorResult')
+    ForEachStatus = Shapes::StringShape.new(name: 'ForEachStatus')
+    ForEachStepMetadata = Shapes::StructureShape.new(name: 'ForEachStepMetadata')
     ForecastFrequency = Shapes::StringShape.new(name: 'ForecastFrequency')
     ForecastHorizon = Shapes::IntegerShape.new(name: 'ForecastHorizon')
     ForecastQuantile = Shapes::StringShape.new(name: 'ForecastQuantile')
@@ -1577,6 +1598,10 @@ module Aws::SageMaker
     IotRoleAlias = Shapes::StringShape.new(name: 'IotRoleAlias')
     IsTrackingServerActive = Shapes::StringShape.new(name: 'IsTrackingServerActive')
     ItemIdentifierAttributeName = Shapes::StringShape.new(name: 'ItemIdentifierAttributeName')
+    IterationContext = Shapes::StructureShape.new(name: 'IterationContext')
+    JavaBoolean = Shapes::BooleanShape.new(name: 'JavaBoolean')
+    JavaInteger = Shapes::IntegerShape.new(name: 'JavaInteger')
+    JavaLong = Shapes::IntegerShape.new(name: 'JavaLong')
     Job = Shapes::StructureShape.new(name: 'Job')
     JobArn = Shapes::StringShape.new(name: 'JobArn')
     JobCategory = Shapes::StringShape.new(name: 'JobCategory')
@@ -1829,6 +1854,8 @@ module Aws::SageMaker
     ListTrainingJobsResponse = Shapes::StructureShape.new(name: 'ListTrainingJobsResponse')
     ListTrainingPlansRequest = Shapes::StructureShape.new(name: 'ListTrainingPlansRequest')
     ListTrainingPlansResponse = Shapes::StructureShape.new(name: 'ListTrainingPlansResponse')
+    ListTrainingSessionsRequest = Shapes::StructureShape.new(name: 'ListTrainingSessionsRequest')
+    ListTrainingSessionsResponse = Shapes::StructureShape.new(name: 'ListTrainingSessionsResponse')
     ListTransformJobsRequest = Shapes::StructureShape.new(name: 'ListTransformJobsRequest')
     ListTransformJobsResponse = Shapes::StructureShape.new(name: 'ListTransformJobsResponse')
     ListTrialComponentKey256 = Shapes::ListShape.new(name: 'ListTrialComponentKey256')
@@ -1849,6 +1876,8 @@ module Aws::SageMaker
     LocalPath = Shapes::StringShape.new(name: 'LocalPath')
     Long = Shapes::IntegerShape.new(name: 'Long')
     LongS3Uri = Shapes::StringShape.new(name: 'LongS3Uri')
+    LoraAlpha = Shapes::IntegerShape.new(name: 'LoraAlpha')
+    LoraConfig = Shapes::StructureShape.new(name: 'LoraConfig')
     MIGProfileType = Shapes::StringShape.new(name: 'MIGProfileType')
     MLFramework = Shapes::StringShape.new(name: 'MLFramework')
     MLflowArn = Shapes::StringShape.new(name: 'MLflowArn')
@@ -2198,6 +2227,7 @@ module Aws::SageMaker
     OutputDataConfig = Shapes::StructureShape.new(name: 'OutputDataConfig')
     OutputParameter = Shapes::StructureShape.new(name: 'OutputParameter')
     OutputParameterList = Shapes::ListShape.new(name: 'OutputParameterList')
+    OutputS3Path = Shapes::StringShape.new(name: 'OutputS3Path')
     OwnershipSettings = Shapes::StructureShape.new(name: 'OwnershipSettings')
     OwnershipSettingsSummary = Shapes::StructureShape.new(name: 'OwnershipSettingsSummary')
     PaginationToken = Shapes::StringShape.new(name: 'PaginationToken')
@@ -2384,6 +2414,7 @@ module Aws::SageMaker
     RStudioServerProDomainSettingsForUpdate = Shapes::StructureShape.new(name: 'RStudioServerProDomainSettingsForUpdate')
     RStudioServerProUserGroup = Shapes::StringShape.new(name: 'RStudioServerProUserGroup')
     RandomSeed = Shapes::IntegerShape.new(name: 'RandomSeed')
+    Rank = Shapes::IntegerShape.new(name: 'Rank')
     RealTimeInferenceConfig = Shapes::StructureShape.new(name: 'RealTimeInferenceConfig')
     RealTimeInferenceRecommendation = Shapes::StructureShape.new(name: 'RealTimeInferenceRecommendation')
     RealTimeInferenceRecommendations = Shapes::ListShape.new(name: 'RealTimeInferenceRecommendations')
@@ -2675,6 +2706,8 @@ module Aws::SageMaker
     StopPipelineExecutionResponse = Shapes::StructureShape.new(name: 'StopPipelineExecutionResponse')
     StopProcessingJobRequest = Shapes::StructureShape.new(name: 'StopProcessingJobRequest')
     StopTrainingJobRequest = Shapes::StructureShape.new(name: 'StopTrainingJobRequest')
+    StopTrainingSessionRequest = Shapes::StructureShape.new(name: 'StopTrainingSessionRequest')
+    StopTrainingSessionResponse = Shapes::StructureShape.new(name: 'StopTrainingSessionResponse')
     StopTransformJobRequest = Shapes::StructureShape.new(name: 'StopTransformJobRequest')
     StoppingCondition = Shapes::StructureShape.new(name: 'StoppingCondition')
     StorageType = Shapes::StringShape.new(name: 'StorageType')
@@ -2682,6 +2715,7 @@ module Aws::SageMaker
     String = Shapes::StringShape.new(name: 'String')
     String1024 = Shapes::StringShape.new(name: 'String1024')
     String128 = Shapes::StringShape.new(name: 'String128')
+    String12800 = Shapes::StringShape.new(name: 'String12800')
     String200 = Shapes::StringShape.new(name: 'String200')
     String2048 = Shapes::StringShape.new(name: 'String2048')
     String256 = Shapes::StringShape.new(name: 'String256')
@@ -2833,6 +2867,15 @@ module Aws::SageMaker
     TrainingRepositoryAccessMode = Shapes::StringShape.new(name: 'TrainingRepositoryAccessMode')
     TrainingRepositoryAuthConfig = Shapes::StructureShape.new(name: 'TrainingRepositoryAuthConfig')
     TrainingRepositoryCredentialsProviderArn = Shapes::StringShape.new(name: 'TrainingRepositoryCredentialsProviderArn')
+    TrainingSessionArn = Shapes::StringShape.new(name: 'TrainingSessionArn')
+    TrainingSessionDetail = Shapes::StructureShape.new(name: 'TrainingSessionDetail')
+    TrainingSessionModelConfig = Shapes::StructureShape.new(name: 'TrainingSessionModelConfig')
+    TrainingSessionName = Shapes::StringShape.new(name: 'TrainingSessionName')
+    TrainingSessionOutputConfig = Shapes::StructureShape.new(name: 'TrainingSessionOutputConfig')
+    TrainingSessionS3Location = Shapes::StructureShape.new(name: 'TrainingSessionS3Location')
+    TrainingSessionStatus = Shapes::StringShape.new(name: 'TrainingSessionStatus')
+    TrainingSessionSummaries = Shapes::ListShape.new(name: 'TrainingSessionSummaries')
+    TrainingSessionSummary = Shapes::StructureShape.new(name: 'TrainingSessionSummary')
     TrainingSpecification = Shapes::StructureShape.new(name: 'TrainingSpecification')
     TrainingStepIndex = Shapes::IntegerShape.new(name: 'TrainingStepIndex')
     TrainingTimeInSeconds = Shapes::IntegerShape.new(name: 'TrainingTimeInSeconds')
@@ -3532,6 +3575,10 @@ module Aws::SageMaker
     AthenaDatasetDefinition.add_member(:output_compression, Shapes::ShapeRef.new(shape: AthenaResultCompressionType, location_name: "OutputCompression"))
     AthenaDatasetDefinition.struct_class = Types::AthenaDatasetDefinition
 
+    AthenaStepMetadata.add_member(:query_execution_id, Shapes::ShapeRef.new(shape: String256, location_name: "QueryExecutionId"))
+    AthenaStepMetadata.add_member(:s3_output_location, Shapes::ShapeRef.new(shape: String1024, location_name: "S3OutputLocation"))
+    AthenaStepMetadata.struct_class = Types::AthenaStepMetadata
+
     AttachClusterNodeNetworkInterfaceRequest.add_member(:cluster_name, Shapes::ShapeRef.new(shape: ClusterNameOrArn, required: true, location_name: "ClusterName"))
     AttachClusterNodeNetworkInterfaceRequest.add_member(:node_id, Shapes::ShapeRef.new(shape: ClusterNodeId, required: true, location_name: "NodeId"))
     AttachClusterNodeNetworkInterfaceRequest.add_member(:network_interface_id, Shapes::ShapeRef.new(shape: ClusterNetworkInterfaceId, required: true, location_name: "NetworkInterfaceId"))
@@ -3905,6 +3952,11 @@ module Aws::SageMaker
     Bias.add_member(:post_training_report, Shapes::ShapeRef.new(shape: MetricsSource, location_name: "PostTrainingReport"))
     Bias.struct_class = Types::Bias
 
+    BillableTokenUsage.add_member(:prefill_token_count, Shapes::ShapeRef.new(shape: JavaLong, location_name: "PrefillTokenCount"))
+    BillableTokenUsage.add_member(:sample_token_count, Shapes::ShapeRef.new(shape: JavaLong, location_name: "SampleTokenCount"))
+    BillableTokenUsage.add_member(:training_token_count, Shapes::ShapeRef.new(shape: JavaLong, location_name: "TrainingTokenCount"))
+    BillableTokenUsage.struct_class = Types::BillableTokenUsage
+
     BlueGreenUpdatePolicy.add_member(:traffic_routing_configuration, Shapes::ShapeRef.new(shape: TrafficRoutingConfig, required: true, location_name: "TrafficRoutingConfiguration"))
     BlueGreenUpdatePolicy.add_member(:termination_wait_in_seconds, Shapes::ShapeRef.new(shape: TerminationWaitInSeconds, location_name: "TerminationWaitInSeconds"))
     BlueGreenUpdatePolicy.add_member(:maximum_execution_timeout_in_seconds, Shapes::ShapeRef.new(shape: MaximumExecutionTimeoutInSeconds, location_name: "MaximumExecutionTimeoutInSeconds"))
@@ -4096,6 +4148,20 @@ module Aws::SageMaker
     ClarifyTextConfig.add_member(:granularity, Shapes::ShapeRef.new(shape: ClarifyTextGranularity, required: true, location_name: "Granularity"))
     ClarifyTextConfig.struct_class = Types::ClarifyTextConfig
 
+    CleanupFailure.add_member(:step_name, Shapes::ShapeRef.new(shape: String256, location_name: "StepName"))
+    CleanupFailure.add_member(:error_code, Shapes::ShapeRef.new(shape: CleanupFailureErrorCode, location_name: "ErrorCode"))
+    CleanupFailure.add_member(:error_message, Shapes::ShapeRef.new(shape: String2048, location_name: "ErrorMessage"))
+    CleanupFailure.struct_class = Types::CleanupFailure
+
+    CleanupFailureList.member = Shapes::ShapeRef.new(shape: CleanupFailure)
+
+    CleanupStepMetadata.add_member(:cleaned_steps, Shapes::ShapeRef.new(shape: CleanupStepNameList, location_name: "CleanedSteps"))
+    CleanupStepMetadata.add_member(:failed_steps, Shapes::ShapeRef.new(shape: CleanupFailureList, location_name: "FailedSteps"))
+    CleanupStepMetadata.add_member(:skipped_steps, Shapes::ShapeRef.new(shape: CleanupStepNameList, location_name: "SkippedSteps"))
+    CleanupStepMetadata.struct_class = Types::CleanupStepMetadata
+
+    CleanupStepNameList.member = Shapes::ShapeRef.new(shape: String256)
+
     ClusterAccountingDatabase.add_member(:endpoint, Shapes::ShapeRef.new(shape: ClusterAccountingDatabaseEndpoint, required: true, location_name: "Endpoint"))
     ClusterAccountingDatabase.add_member(:port, Shapes::ShapeRef.new(shape: ClusterAccountingDatabasePort, location_name: "Port"))
     ClusterAccountingDatabase.add_member(:name, Shapes::ShapeRef.new(shape: ClusterAccountingDatabaseName, location_name: "Name"))
@@ -5474,6 +5540,19 @@ module Aws::SageMaker
     CreateTrainingPlanResponse.add_member(:training_plan_arn, Shapes::ShapeRef.new(shape: TrainingPlanArn, required: true, location_name: "TrainingPlanArn"))
     CreateTrainingPlanResponse.struct_class = Types::CreateTrainingPlanResponse
 
+    CreateTrainingSessionRequest.add_member(:training_session_name, Shapes::ShapeRef.new(shape: TrainingSessionName, required: true, location_name: "TrainingSessionName"))
+    CreateTrainingSessionRequest.add_member(:model_package_group_arn, Shapes::ShapeRef.new(shape: ModelPackageGroupArn, location_name: "ModelPackageGroupArn"))
+    CreateTrainingSessionRequest.add_member(:lora_config, Shapes::ShapeRef.new(shape: LoraConfig, required: true, location_name: "LoraConfig"))
+    CreateTrainingSessionRequest.add_member(:role_arn, Shapes::ShapeRef.new(shape: RoleArn, required: true, location_name: "RoleArn"))
+    CreateTrainingSessionRequest.add_member(:vpc_config, Shapes::ShapeRef.new(shape: VpcConfig, location_name: "VpcConfig"))
+    CreateTrainingSessionRequest.add_member(:model_config, Shapes::ShapeRef.new(shape: TrainingSessionModelConfig, required: true, location_name: "ModelConfig"))
+    CreateTrainingSessionRequest.add_member(:output_config, Shapes::ShapeRef.new(shape: TrainingSessionOutputConfig, required: true, location_name: "OutputConfig"))
+    CreateTrainingSessionRequest.add_member(:tags, Shapes::ShapeRef.new(shape: TagList, location_name: "Tags"))
+    CreateTrainingSessionRequest.struct_class = Types::CreateTrainingSessionRequest
+
+    CreateTrainingSessionResponse.add_member(:training_session_arn, Shapes::ShapeRef.new(shape: TrainingSessionArn, required: true, location_name: "TrainingSessionArn"))
+    CreateTrainingSessionResponse.struct_class = Types::CreateTrainingSessionResponse
+
     CreateTransformJobRequest.add_member(:transform_job_name, Shapes::ShapeRef.new(shape: TransformJobName, required: true, location_name: "TransformJobName"))
     CreateTransformJobRequest.add_member(:model_name, Shapes::ShapeRef.new(shape: ModelName, required: true, location_name: "ModelName"))
     CreateTransformJobRequest.add_member(:max_concurrent_transforms, Shapes::ShapeRef.new(shape: MaxConcurrentTransforms, location_name: "MaxConcurrentTransforms"))
@@ -5964,6 +6043,11 @@ module Aws::SageMaker
     DeleteTrainingJobRequest.add_member(:training_job_name, Shapes::ShapeRef.new(shape: TrainingJobName, required: true, location_name: "TrainingJobName"))
     DeleteTrainingJobRequest.struct_class = Types::DeleteTrainingJobRequest
 
+    DeleteTrainingSessionRequest.add_member(:training_session_name, Shapes::ShapeRef.new(shape: TrainingSessionName, required: true, location_name: "TrainingSessionName"))
+    DeleteTrainingSessionRequest.struct_class = Types::DeleteTrainingSessionRequest
+
+    DeleteTrainingSessionResponse.struct_class = Types::DeleteTrainingSessionResponse
+
     DeleteTrialComponentRequest.add_member(:trial_component_name, Shapes::ShapeRef.new(shape: ExperimentEntityName, required: true, location_name: "TrialComponentName"))
     DeleteTrialComponentRequest.struct_class = Types::DeleteTrialComponentRequest
 
@@ -7332,6 +7416,12 @@ module Aws::SageMaker
     DescribeTrainingPlanResponse.add_member(:reserved_capacity_summaries, Shapes::ShapeRef.new(shape: ReservedCapacitySummaries, location_name: "ReservedCapacitySummaries"))
     DescribeTrainingPlanResponse.struct_class = Types::DescribeTrainingPlanResponse
 
+    DescribeTrainingSessionRequest.add_member(:training_session_name, Shapes::ShapeRef.new(shape: TrainingSessionName, required: true, location_name: "TrainingSessionName"))
+    DescribeTrainingSessionRequest.struct_class = Types::DescribeTrainingSessionRequest
+
+    DescribeTrainingSessionResponse.add_member(:training_session_detail, Shapes::ShapeRef.new(shape: TrainingSessionDetail, required: true, location_name: "TrainingSessionDetail"))
+    DescribeTrainingSessionResponse.struct_class = Types::DescribeTrainingSessionResponse
+
     DescribeTransformJobRequest.add_member(:transform_job_name, Shapes::ShapeRef.new(shape: TransformJobName, required: true, location_name: "TransformJobName"))
     DescribeTransformJobRequest.struct_class = Types::DescribeTransformJobRequest
 
@@ -7591,6 +7681,12 @@ module Aws::SageMaker
     EFSFileSystemConfig.add_member(:file_system_path, Shapes::ShapeRef.new(shape: FileSystemPath, location_name: "FileSystemPath"))
     EFSFileSystemConfig.struct_class = Types::EFSFileSystemConfig
 
+    EMRServerlessStepMetadata.add_member(:application_id, Shapes::ShapeRef.new(shape: EMRServerlessApplicationId, location_name: "ApplicationId"))
+    EMRServerlessStepMetadata.add_member(:job_run_id, Shapes::ShapeRef.new(shape: EMRServerlessJobRunId, location_name: "JobRunId"))
+    EMRServerlessStepMetadata.add_member(:job_run_name, Shapes::ShapeRef.new(shape: String256, location_name: "JobRunName"))
+    EMRServerlessStepMetadata.add_member(:log_file_path, Shapes::ShapeRef.new(shape: String1024, location_name: "LogFilePath"))
+    EMRServerlessStepMetadata.struct_class = Types::EMRServerlessStepMetadata
+
     EMRStepMetadata.add_member(:cluster_id, Shapes::ShapeRef.new(shape: String256, location_name: "ClusterId"))
     EMRStepMetadata.add_member(:step_id, Shapes::ShapeRef.new(shape: String256, location_name: "StepId"))
     EMRStepMetadata.add_member(:step_name, Shapes::ShapeRef.new(shape: String256, location_name: "StepName"))
@@ -8011,6 +8107,19 @@ module Aws::SageMaker
 
     FlowDefinitionTaskKeywords.member = Shapes::ShapeRef.new(shape: FlowDefinitionTaskKeyword)
 
+    ForEachSelectorResult.add_member(:step_name, Shapes::ShapeRef.new(shape: String256, location_name: "StepName"))
+    ForEachSelectorResult.add_member(:index, Shapes::ShapeRef.new(shape: Integer, location_name: "Index", metadata: {"box" => true}))
+    ForEachSelectorResult.struct_class = Types::ForEachSelectorResult
+
+    ForEachStepMetadata.add_member(:status, Shapes::ShapeRef.new(shape: ForEachStatus, location_name: "Status"))
+    ForEachStepMetadata.add_member(:input_item_count, Shapes::ShapeRef.new(shape: Integer, location_name: "InputItemCount", metadata: {"box" => true}))
+    ForEachStepMetadata.add_member(:total_step_count, Shapes::ShapeRef.new(shape: Integer, location_name: "TotalStepCount", metadata: {"box" => true}))
+    ForEachStepMetadata.add_member(:succeeded_step_count, Shapes::ShapeRef.new(shape: Integer, location_name: "SucceededStepCount", metadata: {"box" => true}))
+    ForEachStepMetadata.add_member(:failed_step_count, Shapes::ShapeRef.new(shape: Integer, location_name: "FailedStepCount", metadata: {"box" => true}))
+    ForEachStepMetadata.add_member(:stopped_step_count, Shapes::ShapeRef.new(shape: Integer, location_name: "StoppedStepCount", metadata: {"box" => true}))
+    ForEachStepMetadata.add_member(:selector, Shapes::ShapeRef.new(shape: ForEachSelectorResult, location_name: "Selector"))
+    ForEachStepMetadata.struct_class = Types::ForEachStepMetadata
+
     ForecastQuantiles.member = Shapes::ShapeRef.new(shape: ForecastQuantile)
 
     GenerativeAiSettings.add_member(:amazon_bedrock_role_arn, Shapes::ShapeRef.new(shape: RoleArn, location_name: "AmazonBedrockRoleArn"))
@@ -8711,6 +8820,11 @@ module Aws::SageMaker
 
     IntegerParameterRanges.member = Shapes::ShapeRef.new(shape: IntegerParameterRange)
 
+    IterationContext.add_member(:parent_step_name, Shapes::ShapeRef.new(shape: String256, location_name: "ParentStepName"))
+    IterationContext.add_member(:index, Shapes::ShapeRef.new(shape: Integer, location_name: "Index", metadata: {"box" => true}))
+    IterationContext.add_member(:item, Shapes::ShapeRef.new(shape: String12800, location_name: "Item"))
+    IterationContext.struct_class = Types::IterationContext
+
     Job.add_member(:job_name, Shapes::ShapeRef.new(shape: JobName, location_name: "JobName"))
     Job.add_member(:job_arn, Shapes::ShapeRef.new(shape: JobArn, location_name: "JobArn"))
     Job.add_member(:role_arn, Shapes::ShapeRef.new(shape: RoleArn, location_name: "RoleArn"))
@@ -10062,6 +10176,18 @@ module Aws::SageMaker
     ListTrainingPlansResponse.add_member(:training_plan_summaries, Shapes::ShapeRef.new(shape: TrainingPlanSummaries, required: true, location_name: "TrainingPlanSummaries"))
     ListTrainingPlansResponse.struct_class = Types::ListTrainingPlansResponse
 
+    ListTrainingSessionsRequest.add_member(:next_token, Shapes::ShapeRef.new(shape: NextToken, location_name: "NextToken"))
+    ListTrainingSessionsRequest.add_member(:max_results, Shapes::ShapeRef.new(shape: MaxResults, location_name: "MaxResults"))
+    ListTrainingSessionsRequest.add_member(:creation_time_after, Shapes::ShapeRef.new(shape: Timestamp, location_name: "CreationTimeAfter"))
+    ListTrainingSessionsRequest.add_member(:creation_time_before, Shapes::ShapeRef.new(shape: Timestamp, location_name: "CreationTimeBefore"))
+    ListTrainingSessionsRequest.add_member(:name_contains, Shapes::ShapeRef.new(shape: NameContains, location_name: "NameContains"))
+    ListTrainingSessionsRequest.add_member(:status_equals, Shapes::ShapeRef.new(shape: TrainingSessionStatus, location_name: "StatusEquals"))
+    ListTrainingSessionsRequest.struct_class = Types::ListTrainingSessionsRequest
+
+    ListTrainingSessionsResponse.add_member(:next_token, Shapes::ShapeRef.new(shape: NextToken, location_name: "NextToken"))
+    ListTrainingSessionsResponse.add_member(:training_session_summaries, Shapes::ShapeRef.new(shape: TrainingSessionSummaries, location_name: "TrainingSessionSummaries"))
+    ListTrainingSessionsResponse.struct_class = Types::ListTrainingSessionsResponse
+
     ListTransformJobsRequest.add_member(:creation_time_after, Shapes::ShapeRef.new(shape: Timestamp, location_name: "CreationTimeAfter"))
     ListTransformJobsRequest.add_member(:creation_time_before, Shapes::ShapeRef.new(shape: Timestamp, location_name: "CreationTimeBefore"))
     ListTransformJobsRequest.add_member(:last_modified_time_after, Shapes::ShapeRef.new(shape: Timestamp, location_name: "LastModifiedTimeAfter"))
@@ -10152,6 +10278,10 @@ module Aws::SageMaker
     ListWorkteamsResponse.add_member(:next_token, Shapes::ShapeRef.new(shape: NextToken, location_name: "NextToken"))
     ListWorkteamsResponse.struct_class = Types::ListWorkteamsResponse
 
+    LoraConfig.add_member(:rank, Shapes::ShapeRef.new(shape: Rank, required: true, location_name: "Rank"))
+    LoraConfig.add_member(:alpha, Shapes::ShapeRef.new(shape: LoraAlpha, required: true, location_name: "Alpha"))
+    LoraConfig.struct_class = Types::LoraConfig
+
     MLflowConfiguration.add_member(:mlflow_resource_arn, Shapes::ShapeRef.new(shape: MLflowArn, location_name: "MlflowResourceArn"))
     MLflowConfiguration.add_member(:mlflow_experiment_name, Shapes::ShapeRef.new(shape: MlflowExperimentEntityName, location_name: "MlflowExperimentName"))
     MLflowConfiguration.struct_class = Types::MLflowConfiguration
@@ -11160,6 +11290,7 @@ module Aws::SageMaker
     PipelineExecutionStep.add_member(:metadata, Shapes::ShapeRef.new(shape: PipelineExecutionStepMetadata, location_name: "Metadata"))
     PipelineExecutionStep.add_member(:attempt_count, Shapes::ShapeRef.new(shape: Integer, location_name: "AttemptCount", metadata: {"box" => true}))
     PipelineExecutionStep.add_member(:selective_execution_result, Shapes::ShapeRef.new(shape: SelectiveExecutionResult, location_name: "SelectiveExecutionResult"))
+    PipelineExecutionStep.add_member(:iteration_context, Shapes::ShapeRef.new(shape: IterationContext, location_name: "IterationContext"))
     PipelineExecutionStep.struct_class = Types::PipelineExecutionStep
 
     PipelineExecutionStepList.member = Shapes::ShapeRef.new(shape: PipelineExecutionStep)
@@ -11174,6 +11305,7 @@ module Aws::SageMaker
     PipelineExecutionStepMetadata.add_member(:callback, Shapes::ShapeRef.new(shape: CallbackStepMetadata, location_name: "Callback"))
     PipelineExecutionStepMetadata.add_member(:lambda, Shapes::ShapeRef.new(shape: LambdaStepMetadata, location_name: "Lambda"))
     PipelineExecutionStepMetadata.add_member(:emr, Shapes::ShapeRef.new(shape: EMRStepMetadata, location_name: "EMR"))
+    PipelineExecutionStepMetadata.add_member(:emr_serverless, Shapes::ShapeRef.new(shape: EMRServerlessStepMetadata, location_name: "EMRServerless"))
     PipelineExecutionStepMetadata.add_member(:quality_check, Shapes::ShapeRef.new(shape: QualityCheckStepMetadata, location_name: "QualityCheck"))
     PipelineExecutionStepMetadata.add_member(:clarify_check, Shapes::ShapeRef.new(shape: ClarifyCheckStepMetadata, location_name: "ClarifyCheck"))
     PipelineExecutionStepMetadata.add_member(:fail, Shapes::ShapeRef.new(shape: FailStepMetadata, location_name: "Fail"))
@@ -11187,6 +11319,9 @@ module Aws::SageMaker
     PipelineExecutionStepMetadata.add_member(:inference_component, Shapes::ShapeRef.new(shape: InferenceComponentMetadata, location_name: "InferenceComponent"))
     PipelineExecutionStepMetadata.add_member(:lineage, Shapes::ShapeRef.new(shape: LineageMetadata, location_name: "Lineage"))
     PipelineExecutionStepMetadata.add_member(:job, Shapes::ShapeRef.new(shape: JobStepMetadata, location_name: "Job"))
+    PipelineExecutionStepMetadata.add_member(:athena, Shapes::ShapeRef.new(shape: AthenaStepMetadata, location_name: "Athena"))
+    PipelineExecutionStepMetadata.add_member(:cleanup, Shapes::ShapeRef.new(shape: CleanupStepMetadata, location_name: "Cleanup"))
+    PipelineExecutionStepMetadata.add_member(:for_each, Shapes::ShapeRef.new(shape: ForEachStepMetadata, location_name: "ForEach"))
     PipelineExecutionStepMetadata.struct_class = Types::PipelineExecutionStepMetadata
 
     PipelineExecutionSummary.add_member(:pipeline_execution_arn, Shapes::ShapeRef.new(shape: PipelineExecutionArn, location_name: "PipelineExecutionArn"))
@@ -12278,6 +12413,11 @@ module Aws::SageMaker
     StopTrainingJobRequest.add_member(:training_job_name, Shapes::ShapeRef.new(shape: TrainingJobName, required: true, location_name: "TrainingJobName"))
     StopTrainingJobRequest.struct_class = Types::StopTrainingJobRequest
 
+    StopTrainingSessionRequest.add_member(:training_session_name, Shapes::ShapeRef.new(shape: TrainingSessionName, required: true, location_name: "TrainingSessionName"))
+    StopTrainingSessionRequest.struct_class = Types::StopTrainingSessionRequest
+
+    StopTrainingSessionResponse.struct_class = Types::StopTrainingSessionResponse
+
     StopTransformJobRequest.add_member(:transform_job_name, Shapes::ShapeRef.new(shape: TransformJobName, required: true, location_name: "TransformJobName"))
     StopTransformJobRequest.struct_class = Types::StopTransformJobRequest
 
@@ -12608,6 +12748,43 @@ module Aws::SageMaker
     TrainingRepositoryAuthConfig.add_member(:training_repository_credentials_provider_arn, Shapes::ShapeRef.new(shape: TrainingRepositoryCredentialsProviderArn, required: true, location_name: "TrainingRepositoryCredentialsProviderArn"))
     TrainingRepositoryAuthConfig.struct_class = Types::TrainingRepositoryAuthConfig
 
+    TrainingSessionDetail.add_member(:training_session_arn, Shapes::ShapeRef.new(shape: TrainingSessionArn, required: true, location_name: "TrainingSessionArn"))
+    TrainingSessionDetail.add_member(:training_session_name, Shapes::ShapeRef.new(shape: TrainingSessionName, required: true, location_name: "TrainingSessionName"))
+    TrainingSessionDetail.add_member(:training_session_status, Shapes::ShapeRef.new(shape: TrainingSessionStatus, required: true, location_name: "TrainingSessionStatus"))
+    TrainingSessionDetail.add_member(:model_package_group_arn, Shapes::ShapeRef.new(shape: ModelPackageGroupArn, location_name: "ModelPackageGroupArn"))
+    TrainingSessionDetail.add_member(:creation_time, Shapes::ShapeRef.new(shape: Timestamp, required: true, location_name: "CreationTime"))
+    TrainingSessionDetail.add_member(:last_modified_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "LastModifiedTime"))
+    TrainingSessionDetail.add_member(:end_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "EndTime"))
+    TrainingSessionDetail.add_member(:error_details, Shapes::ShapeRef.new(shape: FailureReason, location_name: "ErrorDetails"))
+    TrainingSessionDetail.add_member(:last_received_sequence_number, Shapes::ShapeRef.new(shape: JavaInteger, location_name: "LastReceivedSequenceNumber"))
+    TrainingSessionDetail.add_member(:billable_token_usage, Shapes::ShapeRef.new(shape: BillableTokenUsage, location_name: "BillableTokenUsage"))
+    TrainingSessionDetail.add_member(:vpc_config, Shapes::ShapeRef.new(shape: VpcConfig, location_name: "VpcConfig"))
+    TrainingSessionDetail.add_member(:role_arn, Shapes::ShapeRef.new(shape: RoleArn, location_name: "RoleArn"))
+    TrainingSessionDetail.add_member(:lora_config, Shapes::ShapeRef.new(shape: LoraConfig, location_name: "LoraConfig"))
+    TrainingSessionDetail.add_member(:model_config, Shapes::ShapeRef.new(shape: TrainingSessionModelConfig, location_name: "ModelConfig"))
+    TrainingSessionDetail.add_member(:output_config, Shapes::ShapeRef.new(shape: TrainingSessionOutputConfig, location_name: "OutputConfig"))
+    TrainingSessionDetail.add_member(:tags, Shapes::ShapeRef.new(shape: TagList, location_name: "Tags"))
+    TrainingSessionDetail.struct_class = Types::TrainingSessionDetail
+
+    TrainingSessionModelConfig.add_member(:base_model_arn, Shapes::ShapeRef.new(shape: BaseModelArn, required: true, location_name: "BaseModelArn"))
+    TrainingSessionModelConfig.add_member(:accept_eula, Shapes::ShapeRef.new(shape: JavaBoolean, location_name: "AcceptEula"))
+    TrainingSessionModelConfig.struct_class = Types::TrainingSessionModelConfig
+
+    TrainingSessionOutputConfig.add_member(:s3_location, Shapes::ShapeRef.new(shape: TrainingSessionS3Location, required: true, location_name: "S3Location"))
+    TrainingSessionOutputConfig.struct_class = Types::TrainingSessionOutputConfig
+
+    TrainingSessionS3Location.add_member(:path, Shapes::ShapeRef.new(shape: OutputS3Path, required: true, location_name: "Path"))
+    TrainingSessionS3Location.add_member(:kms_key_arn, Shapes::ShapeRef.new(shape: CustomerKmsKeyArn, location_name: "KmsKeyArn"))
+    TrainingSessionS3Location.struct_class = Types::TrainingSessionS3Location
+
+    TrainingSessionSummaries.member = Shapes::ShapeRef.new(shape: TrainingSessionSummary)
+
+    TrainingSessionSummary.add_member(:training_session_arn, Shapes::ShapeRef.new(shape: TrainingSessionArn, required: true, location_name: "TrainingSessionArn"))
+    TrainingSessionSummary.add_member(:training_session_status, Shapes::ShapeRef.new(shape: TrainingSessionStatus, required: true, location_name: "TrainingSessionStatus"))
+    TrainingSessionSummary.add_member(:creation_time, Shapes::ShapeRef.new(shape: Timestamp, required: true, location_name: "CreationTime"))
+    TrainingSessionSummary.add_member(:last_modified_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "LastModifiedTime"))
+    TrainingSessionSummary.struct_class = Types::TrainingSessionSummary
+
     TrainingSpecification.add_member(:training_image, Shapes::ShapeRef.new(shape: ContainerImage, required: true, location_name: "TrainingImage"))
     TrainingSpecification.add_member(:training_image_digest, Shapes::ShapeRef.new(shape: ImageDigest, location_name: "TrainingImageDigest"))
     TrainingSpecification.add_member(:supported_hyper_parameters, Shapes::ShapeRef.new(shape: HyperParameterSpecifications, location_name: "SupportedHyperParameters"))
@@ -14250,6 +14427,17 @@ module Aws::SageMaker
         o.errors << Shapes::ShapeRef.new(shape: ResourceLimitExceeded)
       end)
 
+      api.add_operation(:create_training_session, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "CreateTrainingSession"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: CreateTrainingSessionRequest)
+        o.output = Shapes::ShapeRef.new(shape: CreateTrainingSessionResponse)
+        o.errors << Shapes::ShapeRef.new(shape: ResourceNotFound)
+        o.errors << Shapes::ShapeRef.new(shape: ResourceInUse)
+        o.errors << Shapes::ShapeRef.new(shape: ResourceLimitExceeded)
+      end)
+
       api.add_operation(:create_transform_job, Seahorse::Model::Operation.new.tap do |o|
         o.name = "CreateTransformJob"
         o.http_method = "POST"
@@ -14829,6 +15017,15 @@ module Aws::SageMaker
         o.errors << Shapes::ShapeRef.new(shape: ResourceInUse)
       end)
 
+      api.add_operation(:delete_training_session, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "DeleteTrainingSession"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: DeleteTrainingSessionRequest)
+        o.output = Shapes::ShapeRef.new(shape: DeleteTrainingSessionResponse)
+        o.errors << Shapes::ShapeRef.new(shape: ResourceNotFound)
+      end)
+
       api.add_operation(:delete_trial, Seahorse::Model::Operation.new.tap do |o|
         o.name = "DeleteTrial"
         o.http_method = "POST"
@@ -15506,6 +15703,15 @@ module Aws::SageMaker
         )
       end)
 
+      api.add_operation(:describe_training_session, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "DescribeTrainingSession"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: DescribeTrainingSessionRequest)
+        o.output = Shapes::ShapeRef.new(shape: DescribeTrainingSessionResponse)
+        o.errors << Shapes::ShapeRef.new(shape: ResourceNotFound)
+      end)
+
       api.add_operation(:describe_transform_job, Seahorse::Model::Operation.new.tap do |o|
         o.name = "DescribeTransformJob"
         o.http_method = "POST"
@@ -16830,6 +17036,20 @@ module Aws::SageMaker
         )
       end)
 
+      api.add_operation(:list_training_sessions, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "ListTrainingSessions"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: ListTrainingSessionsRequest)
+        o.output = Shapes::ShapeRef.new(shape: ListTrainingSessionsResponse)
+        o[:pager] = Aws::Pager.new(
+          limit_key: "max_results",
+          tokens: {
+            "next_token" => "next_token"
+          }
+        )
+      end)
+
       api.add_operation(:list_transform_jobs, Seahorse::Model::Operation.new.tap do |o|
         o.name = "ListTransformJobs"
         o.http_method = "POST"
@@ -17267,6 +17487,15 @@ module Aws::SageMaker
         o.errors << Shapes::ShapeRef.new(shape: ResourceNotFound)
       end)
 
+      api.add_operation(:stop_training_session, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "StopTrainingSession"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: StopTrainingSessionRequest)
+        o.output = Shapes::ShapeRef.new(shape: StopTrainingSessionResponse)
+        o.errors << Shapes::ShapeRef.new(shape: ResourceNotFound)
+      end)
+
       api.add_operation(:stop_transform_job, Seahorse::Model::Operation.new.tap do |o|
         o.name = "StopTransformJob"
         o.http_method = "POST"