aws-sdk-sagemaker 1.342.0 → 1.343.0

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

@@ -105,6 +105,8 @@ module Aws::SageMaker
     AssociateTrialComponentResponse = Shapes::StructureShape.new(name: 'AssociateTrialComponentResponse')
     AssociationEdgeType = Shapes::StringShape.new(name: 'AssociationEdgeType')
     AssociationEntityArn = Shapes::StringShape.new(name: 'AssociationEntityArn')
+    AssociationInfo = Shapes::StructureShape.new(name: 'AssociationInfo')
+    AssociationInfoList = Shapes::ListShape.new(name: 'AssociationInfoList')
     AssociationSummaries = Shapes::ListShape.new(name: 'AssociationSummaries')
     AssociationSummary = Shapes::StructureShape.new(name: 'AssociationSummary')
     AssumableRoleArns = Shapes::ListShape.new(name: 'AssumableRoleArns')
@@ -196,6 +198,7 @@ module Aws::SageMaker
     AvailableUpgrade = Shapes::StructureShape.new(name: 'AvailableUpgrade')
     AwsManagedHumanLoopRequestSource = Shapes::StringShape.new(name: 'AwsManagedHumanLoopRequestSource')
     BacktestResultsLocation = Shapes::StringShape.new(name: 'BacktestResultsLocation')
+    BaseModel = Shapes::StructureShape.new(name: 'BaseModel')
     BaseModelName = Shapes::StringShape.new(name: 'BaseModelName')
     BatchAddClusterNodesError = Shapes::StructureShape.new(name: 'BatchAddClusterNodesError')
     BatchAddClusterNodesErrorCode = Shapes::StringShape.new(name: 'BatchAddClusterNodesErrorCode')
@@ -237,9 +240,14 @@ module Aws::SageMaker
     BatchReplaceClusterNodesResponse = Shapes::StructureShape.new(name: 'BatchReplaceClusterNodesResponse')
     BatchStrategy = Shapes::StringShape.new(name: 'BatchStrategy')
     BatchTransformInput = Shapes::StructureShape.new(name: 'BatchTransformInput')
+    BedrockCustomModelDeploymentMetadata = Shapes::StructureShape.new(name: 'BedrockCustomModelDeploymentMetadata')
+    BedrockCustomModelMetadata = Shapes::StructureShape.new(name: 'BedrockCustomModelMetadata')
+    BedrockModelImportMetadata = Shapes::StructureShape.new(name: 'BedrockModelImportMetadata')
+    BedrockProvisionedModelThroughputMetadata = Shapes::StructureShape.new(name: 'BedrockProvisionedModelThroughputMetadata')
     BestObjectiveNotImproving = Shapes::StructureShape.new(name: 'BestObjectiveNotImproving')
     Bias = Shapes::StructureShape.new(name: 'Bias')
     BillableTimeInSeconds = Shapes::IntegerShape.new(name: 'BillableTimeInSeconds')
+    BillableTokenCount = Shapes::IntegerShape.new(name: 'BillableTokenCount')
     BlockedReason = Shapes::StringShape.new(name: 'BlockedReason')
     BlueGreenUpdatePolicy = Shapes::StructureShape.new(name: 'BlueGreenUpdatePolicy')
     Boolean = Shapes::BooleanShape.new(name: 'Boolean')
@@ -645,6 +653,7 @@ module Aws::SageMaker
     CustomerMetadataKeyList = Shapes::ListShape.new(name: 'CustomerMetadataKeyList')
     CustomerMetadataMap = Shapes::MapShape.new(name: 'CustomerMetadataMap')
     CustomerMetadataValue = Shapes::StringShape.new(name: 'CustomerMetadataValue')
+    CustomizationTechnique = Shapes::StringShape.new(name: 'CustomizationTechnique')
     CustomizedMetricSpecification = Shapes::StructureShape.new(name: 'CustomizedMetricSpecification')
     DataCaptureConfig = Shapes::StructureShape.new(name: 'DataCaptureConfig')
     DataCaptureConfigSummary = Shapes::StructureShape.new(name: 'DataCaptureConfigSummary')
@@ -660,6 +669,7 @@ module Aws::SageMaker
     DataSourceName = Shapes::StringShape.new(name: 'DataSourceName')
     Database = Shapes::StringShape.new(name: 'Database')
     DatasetDefinition = Shapes::StructureShape.new(name: 'DatasetDefinition')
+    DatasetSource = Shapes::StructureShape.new(name: 'DatasetSource')
     DebugHookConfig = Shapes::StructureShape.new(name: 'DebugHookConfig')
     DebugRuleConfiguration = Shapes::StructureShape.new(name: 'DebugRuleConfiguration')
     DebugRuleConfigurations = Shapes::ListShape.new(name: 'DebugRuleConfigurations')
@@ -1049,6 +1059,8 @@ module Aws::SageMaker
     EnvironmentParameters = Shapes::ListShape.new(name: 'EnvironmentParameters')
     EnvironmentValue = Shapes::StringShape.new(name: 'EnvironmentValue')
     ErrorInfo = Shapes::StructureShape.new(name: 'ErrorInfo')
+    EvaluationType = Shapes::StringShape.new(name: 'EvaluationType')
+    EvaluatorArn = Shapes::StringShape.new(name: 'EvaluatorArn')
     EventDetails = Shapes::StructureShape.new(name: 'EventDetails')
     EventId = Shapes::StringShape.new(name: 'EventId')
     EventMetadata = Shapes::UnionShape.new(name: 'EventMetadata')
@@ -1197,6 +1209,7 @@ module Aws::SageMaker
     HubContentSupportStatus = Shapes::StringShape.new(name: 'HubContentSupportStatus')
     HubContentType = Shapes::StringShape.new(name: 'HubContentType')
     HubContentVersion = Shapes::StringShape.new(name: 'HubContentVersion')
+    HubDataSetArn = Shapes::StringShape.new(name: 'HubDataSetArn')
     HubDescription = Shapes::StringShape.new(name: 'HubDescription')
     HubDisplayName = Shapes::StringShape.new(name: 'HubDisplayName')
     HubInfo = Shapes::StructureShape.new(name: 'HubInfo')
@@ -1309,6 +1322,7 @@ module Aws::SageMaker
     InferenceComponentDataCacheConfig = Shapes::StructureShape.new(name: 'InferenceComponentDataCacheConfig')
     InferenceComponentDataCacheConfigSummary = Shapes::StructureShape.new(name: 'InferenceComponentDataCacheConfigSummary')
     InferenceComponentDeploymentConfig = Shapes::StructureShape.new(name: 'InferenceComponentDeploymentConfig')
+    InferenceComponentMetadata = Shapes::StructureShape.new(name: 'InferenceComponentMetadata')
     InferenceComponentName = Shapes::StringShape.new(name: 'InferenceComponentName')
     InferenceComponentNameContains = Shapes::StringShape.new(name: 'InferenceComponentNameContains')
     InferenceComponentRollingUpdatePolicy = Shapes::StructureShape.new(name: 'InferenceComponentRollingUpdatePolicy')
@@ -1433,6 +1447,7 @@ module Aws::SageMaker
     LineageGroupNameOrArn = Shapes::StringShape.new(name: 'LineageGroupNameOrArn')
     LineageGroupSummaries = Shapes::ListShape.new(name: 'LineageGroupSummaries')
     LineageGroupSummary = Shapes::StructureShape.new(name: 'LineageGroupSummary')
+    LineageMetadata = Shapes::StructureShape.new(name: 'LineageMetadata')
     LineageType = Shapes::StringShape.new(name: 'LineageType')
     ListActionsRequest = Shapes::StructureShape.new(name: 'ListActionsRequest')
     ListActionsResponse = Shapes::StructureShape.new(name: 'ListActionsResponse')
@@ -1622,11 +1637,14 @@ module Aws::SageMaker
     LongS3Uri = Shapes::StringShape.new(name: 'LongS3Uri')
     MIGProfileType = Shapes::StringShape.new(name: 'MIGProfileType')
     MLFramework = Shapes::StringShape.new(name: 'MLFramework')
+    MLflowArn = Shapes::StringShape.new(name: 'MLflowArn')
+    MLflowConfiguration = Shapes::StructureShape.new(name: 'MLflowConfiguration')
     MaintenanceStatus = Shapes::StringShape.new(name: 'MaintenanceStatus')
     MajorMinorVersion = Shapes::StringShape.new(name: 'MajorMinorVersion')
     ManagedInstanceScalingMaxInstanceCount = Shapes::IntegerShape.new(name: 'ManagedInstanceScalingMaxInstanceCount')
     ManagedInstanceScalingMinInstanceCount = Shapes::IntegerShape.new(name: 'ManagedInstanceScalingMinInstanceCount')
     ManagedInstanceScalingStatus = Shapes::StringShape.new(name: 'ManagedInstanceScalingStatus')
+    MapString2048 = Shapes::MapShape.new(name: 'MapString2048')
     MaxAutoMLJobRuntimeInSeconds = Shapes::IntegerShape.new(name: 'MaxAutoMLJobRuntimeInSeconds')
     MaxCandidates = Shapes::IntegerShape.new(name: 'MaxCandidates')
     MaxConcurrentInvocationsPerInstance = Shapes::IntegerShape.new(name: 'MaxConcurrentInvocationsPerInstance')
@@ -1669,6 +1687,7 @@ module Aws::SageMaker
     MetricsConfig = Shapes::StructureShape.new(name: 'MetricsConfig')
     MetricsSource = Shapes::StructureShape.new(name: 'MetricsSource')
     MinimumInstanceMetadataServiceVersion = Shapes::StringShape.new(name: 'MinimumInstanceMetadataServiceVersion')
+    MlFlowResourceArn = Shapes::StringShape.new(name: 'MlFlowResourceArn')
     MlReservationArn = Shapes::StringShape.new(name: 'MlReservationArn')
     MlTools = Shapes::StringShape.new(name: 'MlTools')
     MlflowAppArn = Shapes::StringShape.new(name: 'MlflowAppArn')
@@ -1677,6 +1696,13 @@ module Aws::SageMaker
     MlflowAppSummaries = Shapes::ListShape.new(name: 'MlflowAppSummaries')
     MlflowAppSummary = Shapes::StructureShape.new(name: 'MlflowAppSummary')
     MlflowAppUrl = Shapes::StringShape.new(name: 'MlflowAppUrl')
+    MlflowConfig = Shapes::StructureShape.new(name: 'MlflowConfig')
+    MlflowDetails = Shapes::StructureShape.new(name: 'MlflowDetails')
+    MlflowExperimentEntityName = Shapes::StringShape.new(name: 'MlflowExperimentEntityName')
+    MlflowExperimentId = Shapes::StringShape.new(name: 'MlflowExperimentId')
+    MlflowExperimentName = Shapes::StringShape.new(name: 'MlflowExperimentName')
+    MlflowRunId = Shapes::StringShape.new(name: 'MlflowRunId')
+    MlflowRunName = Shapes::StringShape.new(name: 'MlflowRunName')
     MlflowVersion = Shapes::StringShape.new(name: 'MlflowVersion')
     Model = Shapes::StructureShape.new(name: 'Model')
     ModelAccessConfig = Shapes::StructureShape.new(name: 'ModelAccessConfig')
@@ -1747,6 +1773,7 @@ module Aws::SageMaker
     ModelPackage = Shapes::StructureShape.new(name: 'ModelPackage')
     ModelPackageArn = Shapes::StringShape.new(name: 'ModelPackageArn')
     ModelPackageArnList = Shapes::ListShape.new(name: 'ModelPackageArnList')
+    ModelPackageConfig = Shapes::StructureShape.new(name: 'ModelPackageConfig')
     ModelPackageContainerDefinition = Shapes::StructureShape.new(name: 'ModelPackageContainerDefinition')
     ModelPackageContainerDefinitionList = Shapes::ListShape.new(name: 'ModelPackageContainerDefinitionList')
     ModelPackageFrameworkVersion = Shapes::StringShape.new(name: 'ModelPackageFrameworkVersion')
@@ -1757,6 +1784,7 @@ module Aws::SageMaker
     ModelPackageGroupSummary = Shapes::StructureShape.new(name: 'ModelPackageGroupSummary')
     ModelPackageGroupSummaryList = Shapes::ListShape.new(name: 'ModelPackageGroupSummaryList')
     ModelPackageModelCard = Shapes::StructureShape.new(name: 'ModelPackageModelCard')
+    ModelPackageRegistrationType = Shapes::StringShape.new(name: 'ModelPackageRegistrationType')
     ModelPackageSecurityConfig = Shapes::StructureShape.new(name: 'ModelPackageSecurityConfig')
     ModelPackageSortBy = Shapes::StringShape.new(name: 'ModelPackageSortBy')
     ModelPackageSourceUri = Shapes::StringShape.new(name: 'ModelPackageSourceUri')
@@ -1975,6 +2003,7 @@ module Aws::SageMaker
     PartnerAppSummaries = Shapes::ListShape.new(name: 'PartnerAppSummaries')
     PartnerAppSummary = Shapes::StructureShape.new(name: 'PartnerAppSummary')
     PartnerAppType = Shapes::StringShape.new(name: 'PartnerAppType')
+    Peft = Shapes::StringShape.new(name: 'Peft')
     PendingDeploymentSummary = Shapes::StructureShape.new(name: 'PendingDeploymentSummary')
     PendingProductionVariantSummary = Shapes::StructureShape.new(name: 'PendingProductionVariantSummary')
     PendingProductionVariantSummaryList = Shapes::ListShape.new(name: 'PendingProductionVariantSummaryList')
@@ -2130,6 +2159,7 @@ module Aws::SageMaker
     RealTimeInferenceRecommendation = Shapes::StructureShape.new(name: 'RealTimeInferenceRecommendation')
     RealTimeInferenceRecommendations = Shapes::ListShape.new(name: 'RealTimeInferenceRecommendations')
     RealtimeInferenceInstanceTypes = Shapes::ListShape.new(name: 'RealtimeInferenceInstanceTypes')
+    RecipeName = Shapes::StringShape.new(name: 'RecipeName')
     RecommendationFailureReason = Shapes::StringShape.new(name: 'RecommendationFailureReason')
     RecommendationJobArn = Shapes::StringShape.new(name: 'RecommendationJobArn')
     RecommendationJobCompilationJobName = Shapes::StringShape.new(name: 'RecommendationJobCompilationJobName')
@@ -2296,6 +2326,9 @@ module Aws::SageMaker
     SendPipelineExecutionStepFailureResponse = Shapes::StructureShape.new(name: 'SendPipelineExecutionStepFailureResponse')
     SendPipelineExecutionStepSuccessRequest = Shapes::StructureShape.new(name: 'SendPipelineExecutionStepSuccessRequest')
     SendPipelineExecutionStepSuccessResponse = Shapes::StructureShape.new(name: 'SendPipelineExecutionStepSuccessResponse')
+    ServerlessJobBaseModelArn = Shapes::StringShape.new(name: 'ServerlessJobBaseModelArn')
+    ServerlessJobConfig = Shapes::StructureShape.new(name: 'ServerlessJobConfig')
+    ServerlessJobType = Shapes::StringShape.new(name: 'ServerlessJobType')
     ServerlessMaxConcurrency = Shapes::IntegerShape.new(name: 'ServerlessMaxConcurrency')
     ServerlessMemorySizeInMB = Shapes::IntegerShape.new(name: 'ServerlessMemorySizeInMB')
     ServerlessProvisionedConcurrency = Shapes::IntegerShape.new(name: 'ServerlessProvisionedConcurrency')
@@ -2485,6 +2518,7 @@ module Aws::SageMaker
     TokenValue = Shapes::StringShape.new(name: 'TokenValue')
     TotalHits = Shapes::StructureShape.new(name: 'TotalHits')
     TotalInstanceCount = Shapes::IntegerShape.new(name: 'TotalInstanceCount')
+    TotalStepCountPerEpoch = Shapes::IntegerShape.new(name: 'TotalStepCountPerEpoch')
     TrackingServerArn = Shapes::StringShape.new(name: 'TrackingServerArn')
     TrackingServerMaintenanceStatus = Shapes::StringShape.new(name: 'TrackingServerMaintenanceStatus')
     TrackingServerName = Shapes::StringShape.new(name: 'TrackingServerName')
@@ -2505,6 +2539,8 @@ module Aws::SageMaker
     TrainingEnvironmentKey = Shapes::StringShape.new(name: 'TrainingEnvironmentKey')
     TrainingEnvironmentMap = Shapes::MapShape.new(name: 'TrainingEnvironmentMap')
     TrainingEnvironmentValue = Shapes::StringShape.new(name: 'TrainingEnvironmentValue')
+    TrainingEpochCount = Shapes::IntegerShape.new(name: 'TrainingEpochCount')
+    TrainingEpochIndex = Shapes::IntegerShape.new(name: 'TrainingEpochIndex')
     TrainingImageConfig = Shapes::StructureShape.new(name: 'TrainingImageConfig')
     TrainingInputMode = Shapes::StringShape.new(name: 'TrainingInputMode')
     TrainingInstanceCount = Shapes::IntegerShape.new(name: 'TrainingInstanceCount')
@@ -2540,10 +2576,12 @@ module Aws::SageMaker
     TrainingPlanStatusMessage = Shapes::StringShape.new(name: 'TrainingPlanStatusMessage')
     TrainingPlanSummaries = Shapes::ListShape.new(name: 'TrainingPlanSummaries')
     TrainingPlanSummary = Shapes::StructureShape.new(name: 'TrainingPlanSummary')
+    TrainingProgressInfo = Shapes::StructureShape.new(name: 'TrainingProgressInfo')
     TrainingRepositoryAccessMode = Shapes::StringShape.new(name: 'TrainingRepositoryAccessMode')
     TrainingRepositoryAuthConfig = Shapes::StructureShape.new(name: 'TrainingRepositoryAuthConfig')
     TrainingRepositoryCredentialsProviderArn = Shapes::StringShape.new(name: 'TrainingRepositoryCredentialsProviderArn')
     TrainingSpecification = Shapes::StructureShape.new(name: 'TrainingSpecification')
+    TrainingStepIndex = Shapes::IntegerShape.new(name: 'TrainingStepIndex')
     TrainingTimeInSeconds = Shapes::IntegerShape.new(name: 'TrainingTimeInSeconds')
     TransformAmiVersion = Shapes::StringShape.new(name: 'TransformAmiVersion')
     TransformDataSource = Shapes::StructureShape.new(name: 'TransformDataSource')
@@ -2979,6 +3017,12 @@ module Aws::SageMaker
     AssociateTrialComponentResponse.add_member(:trial_arn, Shapes::ShapeRef.new(shape: TrialArn, location_name: "TrialArn"))
     AssociateTrialComponentResponse.struct_class = Types::AssociateTrialComponentResponse
 
+    AssociationInfo.add_member(:source_arn, Shapes::ShapeRef.new(shape: String2048, required: true, location_name: "SourceArn"))
+    AssociationInfo.add_member(:destination_arn, Shapes::ShapeRef.new(shape: String2048, required: true, location_name: "DestinationArn"))
+    AssociationInfo.struct_class = Types::AssociationInfo
+
+    AssociationInfoList.member = Shapes::ShapeRef.new(shape: AssociationInfo)
+
     AssociationSummaries.member = Shapes::ShapeRef.new(shape: AssociationSummary)
 
     AssociationSummary.add_member(:source_arn, Shapes::ShapeRef.new(shape: AssociationEntityArn, location_name: "SourceArn"))
@@ -3215,6 +3259,11 @@ module Aws::SageMaker
     AvailableUpgrade.add_member(:release_notes, Shapes::ShapeRef.new(shape: ReleaseNotesList, location_name: "ReleaseNotes"))
     AvailableUpgrade.struct_class = Types::AvailableUpgrade
 
+    BaseModel.add_member(:hub_content_name, Shapes::ShapeRef.new(shape: HubContentName, location_name: "HubContentName"))
+    BaseModel.add_member(:hub_content_version, Shapes::ShapeRef.new(shape: HubContentVersion, location_name: "HubContentVersion"))
+    BaseModel.add_member(:recipe_name, Shapes::ShapeRef.new(shape: RecipeName, location_name: "RecipeName"))
+    BaseModel.struct_class = Types::BaseModel
+
     BatchAddClusterNodesError.add_member(:instance_group_name, Shapes::ShapeRef.new(shape: InstanceGroupName, required: true, location_name: "InstanceGroupName"))
     BatchAddClusterNodesError.add_member(:error_code, Shapes::ShapeRef.new(shape: BatchAddClusterNodesErrorCode, required: true, location_name: "ErrorCode"))
     BatchAddClusterNodesError.add_member(:failed_count, Shapes::ShapeRef.new(shape: BatchAddFailureCount, required: true, location_name: "FailedCount"))
@@ -3284,6 +3333,7 @@ module Aws::SageMaker
     BatchDescribeModelPackageSummary.add_member(:inference_specification, Shapes::ShapeRef.new(shape: InferenceSpecification, required: true, location_name: "InferenceSpecification"))
     BatchDescribeModelPackageSummary.add_member(:model_package_status, Shapes::ShapeRef.new(shape: ModelPackageStatus, required: true, location_name: "ModelPackageStatus"))
     BatchDescribeModelPackageSummary.add_member(:model_approval_status, Shapes::ShapeRef.new(shape: ModelApprovalStatus, location_name: "ModelApprovalStatus"))
+    BatchDescribeModelPackageSummary.add_member(:model_package_registration_type, Shapes::ShapeRef.new(shape: ModelPackageRegistrationType, location_name: "ModelPackageRegistrationType"))
     BatchDescribeModelPackageSummary.struct_class = Types::BatchDescribeModelPackageSummary
 
     BatchRebootClusterNodeLogicalIdsError.add_member(:node_logical_id, Shapes::ShapeRef.new(shape: ClusterNodeLogicalId, required: true, location_name: "NodeLogicalId"))
@@ -3358,6 +3408,18 @@ module Aws::SageMaker
     BatchTransformInput.add_member(:exclude_features_attribute, Shapes::ShapeRef.new(shape: ExcludeFeaturesAttribute, location_name: "ExcludeFeaturesAttribute"))
     BatchTransformInput.struct_class = Types::BatchTransformInput
 
+    BedrockCustomModelDeploymentMetadata.add_member(:arn, Shapes::ShapeRef.new(shape: String1024, location_name: "Arn"))
+    BedrockCustomModelDeploymentMetadata.struct_class = Types::BedrockCustomModelDeploymentMetadata
+
+    BedrockCustomModelMetadata.add_member(:arn, Shapes::ShapeRef.new(shape: String1024, location_name: "Arn"))
+    BedrockCustomModelMetadata.struct_class = Types::BedrockCustomModelMetadata
+
+    BedrockModelImportMetadata.add_member(:arn, Shapes::ShapeRef.new(shape: String1024, location_name: "Arn"))
+    BedrockModelImportMetadata.struct_class = Types::BedrockModelImportMetadata
+
+    BedrockProvisionedModelThroughputMetadata.add_member(:arn, Shapes::ShapeRef.new(shape: String1024, location_name: "Arn"))
+    BedrockProvisionedModelThroughputMetadata.struct_class = Types::BedrockProvisionedModelThroughputMetadata
+
     BestObjectiveNotImproving.add_member(:max_number_of_training_jobs_not_improving, Shapes::ShapeRef.new(shape: MaxNumberOfTrainingJobsNotImproving, location_name: "MaxNumberOfTrainingJobsNotImproving"))
     BestObjectiveNotImproving.struct_class = Types::BestObjectiveNotImproving
 
@@ -3739,7 +3801,7 @@ module Aws::SageMaker
 
     ClusterOnDemandOptions.struct_class = Types::ClusterOnDemandOptions
 
-    ClusterOrchestrator.add_member(:eks, Shapes::ShapeRef.new(shape: ClusterOrchestratorEksConfig, location_name: "Eks"))
+    ClusterOrchestrator.add_member(:eks, Shapes::ShapeRef.new(shape: ClusterOrchestratorEksConfig, required: true, location_name: "Eks"))
     ClusterOrchestrator.struct_class = Types::ClusterOrchestrator
 
     ClusterOrchestratorEksConfig.add_member(:cluster_arn, Shapes::ShapeRef.new(shape: EksClusterArn, required: true, location_name: "ClusterArn"))
@@ -4495,6 +4557,7 @@ module Aws::SageMaker
     CreateModelPackageInput.add_member(:model_package_name, Shapes::ShapeRef.new(shape: EntityName, location_name: "ModelPackageName"))
     CreateModelPackageInput.add_member(:model_package_group_name, Shapes::ShapeRef.new(shape: ArnOrName, location_name: "ModelPackageGroupName"))
     CreateModelPackageInput.add_member(:model_package_description, Shapes::ShapeRef.new(shape: EntityDescription, location_name: "ModelPackageDescription"))
+    CreateModelPackageInput.add_member(:model_package_registration_type, Shapes::ShapeRef.new(shape: ModelPackageRegistrationType, location_name: "ModelPackageRegistrationType"))
     CreateModelPackageInput.add_member(:inference_specification, Shapes::ShapeRef.new(shape: InferenceSpecification, location_name: "InferenceSpecification"))
     CreateModelPackageInput.add_member(:validation_specification, Shapes::ShapeRef.new(shape: ModelPackageValidationSpecification, location_name: "ValidationSpecification"))
     CreateModelPackageInput.add_member(:source_algorithm_specification, Shapes::ShapeRef.new(shape: SourceAlgorithmSpecification, location_name: "SourceAlgorithmSpecification"))
@@ -4741,6 +4804,9 @@ module Aws::SageMaker
     CreateTrainingJobRequest.add_member(:remote_debug_config, Shapes::ShapeRef.new(shape: RemoteDebugConfig, location_name: "RemoteDebugConfig"))
     CreateTrainingJobRequest.add_member(:infra_check_config, Shapes::ShapeRef.new(shape: InfraCheckConfig, location_name: "InfraCheckConfig"))
     CreateTrainingJobRequest.add_member(:session_chaining_config, Shapes::ShapeRef.new(shape: SessionChainingConfig, location_name: "SessionChainingConfig"))
+    CreateTrainingJobRequest.add_member(:serverless_job_config, Shapes::ShapeRef.new(shape: ServerlessJobConfig, location_name: "ServerlessJobConfig"))
+    CreateTrainingJobRequest.add_member(:mlflow_config, Shapes::ShapeRef.new(shape: MlflowConfig, location_name: "MlflowConfig"))
+    CreateTrainingJobRequest.add_member(:model_package_config, Shapes::ShapeRef.new(shape: ModelPackageConfig, location_name: "ModelPackageConfig"))
     CreateTrainingJobRequest.struct_class = Types::CreateTrainingJobRequest
 
     CreateTrainingJobResponse.add_member(:training_job_arn, Shapes::ShapeRef.new(shape: TrainingJobArn, required: true, location_name: "TrainingJobArn"))
@@ -4932,6 +4998,7 @@ module Aws::SageMaker
 
     DataSource.add_member(:s3_data_source, Shapes::ShapeRef.new(shape: S3DataSource, location_name: "S3DataSource"))
     DataSource.add_member(:file_system_data_source, Shapes::ShapeRef.new(shape: FileSystemDataSource, location_name: "FileSystemDataSource"))
+    DataSource.add_member(:dataset_source, Shapes::ShapeRef.new(shape: DatasetSource, location_name: "DatasetSource"))
     DataSource.struct_class = Types::DataSource
 
     DatasetDefinition.add_member(:athena_dataset_definition, Shapes::ShapeRef.new(shape: AthenaDatasetDefinition, location_name: "AthenaDatasetDefinition"))
@@ -4941,6 +5008,9 @@ module Aws::SageMaker
     DatasetDefinition.add_member(:input_mode, Shapes::ShapeRef.new(shape: InputMode, location_name: "InputMode"))
     DatasetDefinition.struct_class = Types::DatasetDefinition
 
+    DatasetSource.add_member(:dataset_arn, Shapes::ShapeRef.new(shape: HubDataSetArn, required: true, location_name: "DatasetArn"))
+    DatasetSource.struct_class = Types::DatasetSource
+
     DebugHookConfig.add_member(:local_path, Shapes::ShapeRef.new(shape: DirectoryPath, location_name: "LocalPath"))
     DebugHookConfig.add_member(:s3_output_path, Shapes::ShapeRef.new(shape: S3Uri, required: true, location_name: "S3OutputPath"))
     DebugHookConfig.add_member(:hook_parameters, Shapes::ShapeRef.new(shape: HookParameters, location_name: "HookParameters"))
@@ -6102,6 +6172,7 @@ module Aws::SageMaker
     DescribeModelPackageOutput.add_member(:model_package_name, Shapes::ShapeRef.new(shape: EntityName, required: true, location_name: "ModelPackageName"))
     DescribeModelPackageOutput.add_member(:model_package_group_name, Shapes::ShapeRef.new(shape: EntityName, location_name: "ModelPackageGroupName"))
     DescribeModelPackageOutput.add_member(:model_package_version, Shapes::ShapeRef.new(shape: ModelPackageVersion, location_name: "ModelPackageVersion"))
+    DescribeModelPackageOutput.add_member(:model_package_registration_type, Shapes::ShapeRef.new(shape: ModelPackageRegistrationType, location_name: "ModelPackageRegistrationType"))
     DescribeModelPackageOutput.add_member(:model_package_arn, Shapes::ShapeRef.new(shape: ModelPackageArn, required: true, location_name: "ModelPackageArn"))
     DescribeModelPackageOutput.add_member(:model_package_description, Shapes::ShapeRef.new(shape: EntityDescription, location_name: "ModelPackageDescription"))
     DescribeModelPackageOutput.add_member(:creation_time, Shapes::ShapeRef.new(shape: CreationTime, required: true, location_name: "CreationTime"))
@@ -6273,6 +6344,7 @@ module Aws::SageMaker
     DescribePipelineExecutionResponse.add_member(:parallelism_configuration, Shapes::ShapeRef.new(shape: ParallelismConfiguration, location_name: "ParallelismConfiguration"))
     DescribePipelineExecutionResponse.add_member(:selective_execution_config, Shapes::ShapeRef.new(shape: SelectiveExecutionConfig, location_name: "SelectiveExecutionConfig"))
     DescribePipelineExecutionResponse.add_member(:pipeline_version_id, Shapes::ShapeRef.new(shape: PipelineVersionId, location_name: "PipelineVersionId"))
+    DescribePipelineExecutionResponse.add_member(:m_lflow_config, Shapes::ShapeRef.new(shape: MLflowConfiguration, location_name: "MLflowConfig"))
     DescribePipelineExecutionResponse.struct_class = Types::DescribePipelineExecutionResponse
 
     DescribePipelineRequest.add_member(:pipeline_name, Shapes::ShapeRef.new(shape: PipelineNameOrArn, required: true, location_name: "PipelineName"))
@@ -6426,6 +6498,7 @@ module Aws::SageMaker
     DescribeTrainingJobResponse.add_member(:checkpoint_config, Shapes::ShapeRef.new(shape: CheckpointConfig, location_name: "CheckpointConfig"))
     DescribeTrainingJobResponse.add_member(:training_time_in_seconds, Shapes::ShapeRef.new(shape: TrainingTimeInSeconds, location_name: "TrainingTimeInSeconds"))
     DescribeTrainingJobResponse.add_member(:billable_time_in_seconds, Shapes::ShapeRef.new(shape: BillableTimeInSeconds, location_name: "BillableTimeInSeconds"))
+    DescribeTrainingJobResponse.add_member(:billable_token_count, Shapes::ShapeRef.new(shape: BillableTokenCount, location_name: "BillableTokenCount"))
     DescribeTrainingJobResponse.add_member(:debug_hook_config, Shapes::ShapeRef.new(shape: DebugHookConfig, location_name: "DebugHookConfig"))
     DescribeTrainingJobResponse.add_member(:experiment_config, Shapes::ShapeRef.new(shape: ExperimentConfig, location_name: "ExperimentConfig"))
     DescribeTrainingJobResponse.add_member(:debug_rule_configurations, Shapes::ShapeRef.new(shape: DebugRuleConfigurations, location_name: "DebugRuleConfigurations"))
@@ -6439,6 +6512,12 @@ module Aws::SageMaker
     DescribeTrainingJobResponse.add_member(:retry_strategy, Shapes::ShapeRef.new(shape: RetryStrategy, location_name: "RetryStrategy"))
     DescribeTrainingJobResponse.add_member(:remote_debug_config, Shapes::ShapeRef.new(shape: RemoteDebugConfig, location_name: "RemoteDebugConfig"))
     DescribeTrainingJobResponse.add_member(:infra_check_config, Shapes::ShapeRef.new(shape: InfraCheckConfig, location_name: "InfraCheckConfig"))
+    DescribeTrainingJobResponse.add_member(:serverless_job_config, Shapes::ShapeRef.new(shape: ServerlessJobConfig, location_name: "ServerlessJobConfig"))
+    DescribeTrainingJobResponse.add_member(:mlflow_config, Shapes::ShapeRef.new(shape: MlflowConfig, location_name: "MlflowConfig"))
+    DescribeTrainingJobResponse.add_member(:model_package_config, Shapes::ShapeRef.new(shape: ModelPackageConfig, location_name: "ModelPackageConfig"))
+    DescribeTrainingJobResponse.add_member(:mlflow_details, Shapes::ShapeRef.new(shape: MlflowDetails, location_name: "MlflowDetails"))
+    DescribeTrainingJobResponse.add_member(:progress_info, Shapes::ShapeRef.new(shape: TrainingProgressInfo, location_name: "ProgressInfo"))
+    DescribeTrainingJobResponse.add_member(:output_model_package_arn, Shapes::ShapeRef.new(shape: ModelPackageArn, location_name: "OutputModelPackageArn"))
     DescribeTrainingJobResponse.struct_class = Types::DescribeTrainingJobResponse
 
     DescribeTrainingPlanRequest.add_member(:training_plan_name, Shapes::ShapeRef.new(shape: TrainingPlanName, required: true, location_name: "TrainingPlanName"))
@@ -7556,6 +7635,9 @@ module Aws::SageMaker
     InferenceComponentDeploymentConfig.add_member(:auto_rollback_configuration, Shapes::ShapeRef.new(shape: AutoRollbackConfig, location_name: "AutoRollbackConfiguration"))
     InferenceComponentDeploymentConfig.struct_class = Types::InferenceComponentDeploymentConfig
 
+    InferenceComponentMetadata.add_member(:arn, Shapes::ShapeRef.new(shape: String2048, location_name: "Arn"))
+    InferenceComponentMetadata.struct_class = Types::InferenceComponentMetadata
+
     InferenceComponentRollingUpdatePolicy.add_member(:maximum_batch_size, Shapes::ShapeRef.new(shape: InferenceComponentCapacitySize, required: true, location_name: "MaximumBatchSize"))
     InferenceComponentRollingUpdatePolicy.add_member(:wait_interval_in_seconds, Shapes::ShapeRef.new(shape: WaitIntervalInSeconds, required: true, location_name: "WaitIntervalInSeconds"))
     InferenceComponentRollingUpdatePolicy.add_member(:maximum_execution_timeout_in_seconds, Shapes::ShapeRef.new(shape: MaximumExecutionTimeoutInSeconds, location_name: "MaximumExecutionTimeoutInSeconds"))
@@ -7876,6 +7958,12 @@ module Aws::SageMaker
     LineageGroupSummary.add_member(:last_modified_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "LastModifiedTime"))
     LineageGroupSummary.struct_class = Types::LineageGroupSummary
 
+    LineageMetadata.add_member(:action_arns, Shapes::ShapeRef.new(shape: MapString2048, location_name: "ActionArns"))
+    LineageMetadata.add_member(:artifact_arns, Shapes::ShapeRef.new(shape: MapString2048, location_name: "ArtifactArns"))
+    LineageMetadata.add_member(:context_arns, Shapes::ShapeRef.new(shape: MapString2048, location_name: "ContextArns"))
+    LineageMetadata.add_member(:associations, Shapes::ShapeRef.new(shape: AssociationInfoList, location_name: "Associations"))
+    LineageMetadata.struct_class = Types::LineageMetadata
+
     ListActionsRequest.add_member(:source_uri, Shapes::ShapeRef.new(shape: SourceUri, location_name: "SourceUri"))
     ListActionsRequest.add_member(:action_type, Shapes::ShapeRef.new(shape: String256, location_name: "ActionType"))
     ListActionsRequest.add_member(:created_after, Shapes::ShapeRef.new(shape: Timestamp, location_name: "CreatedAfter"))
@@ -9063,6 +9151,13 @@ module Aws::SageMaker
     ListWorkteamsResponse.add_member(:next_token, Shapes::ShapeRef.new(shape: NextToken, location_name: "NextToken"))
     ListWorkteamsResponse.struct_class = Types::ListWorkteamsResponse
 
+    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
+
+    MapString2048.key = Shapes::ShapeRef.new(shape: String2048)
+    MapString2048.value = Shapes::ShapeRef.new(shape: String2048)
+
     MemberDefinition.add_member(:cognito_member_definition, Shapes::ShapeRef.new(shape: CognitoMemberDefinition, location_name: "CognitoMemberDefinition"))
     MemberDefinition.add_member(:oidc_member_definition, Shapes::ShapeRef.new(shape: OidcMemberDefinition, location_name: "OidcMemberDefinition"))
     MemberDefinition.struct_class = Types::MemberDefinition
@@ -9121,6 +9216,15 @@ module Aws::SageMaker
     MlflowAppSummary.add_member(:mlflow_version, Shapes::ShapeRef.new(shape: MlflowVersion, location_name: "MlflowVersion"))
     MlflowAppSummary.struct_class = Types::MlflowAppSummary
 
+    MlflowConfig.add_member(:mlflow_resource_arn, Shapes::ShapeRef.new(shape: MlFlowResourceArn, required: true, location_name: "MlflowResourceArn"))
+    MlflowConfig.add_member(:mlflow_experiment_name, Shapes::ShapeRef.new(shape: MlflowExperimentName, location_name: "MlflowExperimentName"))
+    MlflowConfig.add_member(:mlflow_run_name, Shapes::ShapeRef.new(shape: MlflowRunName, location_name: "MlflowRunName"))
+    MlflowConfig.struct_class = Types::MlflowConfig
+
+    MlflowDetails.add_member(:mlflow_experiment_id, Shapes::ShapeRef.new(shape: MlflowExperimentId, location_name: "MlflowExperimentId"))
+    MlflowDetails.add_member(:mlflow_run_id, Shapes::ShapeRef.new(shape: MlflowRunId, location_name: "MlflowRunId"))
+    MlflowDetails.struct_class = Types::MlflowDetails
+
     Model.add_member(:model_name, Shapes::ShapeRef.new(shape: ModelName, location_name: "ModelName"))
     Model.add_member(:primary_container, Shapes::ShapeRef.new(shape: ContainerDefinition, location_name: "PrimaryContainer"))
     Model.add_member(:containers, Shapes::ShapeRef.new(shape: ContainerDefinitionList, location_name: "Containers"))
@@ -9346,6 +9450,7 @@ module Aws::SageMaker
     ModelPackage.add_member(:model_package_name, Shapes::ShapeRef.new(shape: EntityName, location_name: "ModelPackageName"))
     ModelPackage.add_member(:model_package_group_name, Shapes::ShapeRef.new(shape: EntityName, location_name: "ModelPackageGroupName"))
     ModelPackage.add_member(:model_package_version, Shapes::ShapeRef.new(shape: ModelPackageVersion, location_name: "ModelPackageVersion"))
+    ModelPackage.add_member(:model_package_registration_type, Shapes::ShapeRef.new(shape: ModelPackageRegistrationType, location_name: "ModelPackageRegistrationType"))
     ModelPackage.add_member(:model_package_arn, Shapes::ShapeRef.new(shape: ModelPackageArn, location_name: "ModelPackageArn"))
     ModelPackage.add_member(:model_package_description, Shapes::ShapeRef.new(shape: EntityDescription, location_name: "ModelPackageDescription"))
     ModelPackage.add_member(:creation_time, Shapes::ShapeRef.new(shape: CreationTime, location_name: "CreationTime"))
@@ -9378,6 +9483,10 @@ module Aws::SageMaker
 
     ModelPackageArnList.member = Shapes::ShapeRef.new(shape: ModelPackageArn)
 
+    ModelPackageConfig.add_member(:model_package_group_arn, Shapes::ShapeRef.new(shape: ModelPackageGroupArn, required: true, location_name: "ModelPackageGroupArn"))
+    ModelPackageConfig.add_member(:source_model_package_arn, Shapes::ShapeRef.new(shape: ModelPackageArn, location_name: "SourceModelPackageArn"))
+    ModelPackageConfig.struct_class = Types::ModelPackageConfig
+
     ModelPackageContainerDefinition.add_member(:container_hostname, Shapes::ShapeRef.new(shape: ContainerHostname, location_name: "ContainerHostname"))
     ModelPackageContainerDefinition.add_member(:image, Shapes::ShapeRef.new(shape: ContainerImage, location_name: "Image"))
     ModelPackageContainerDefinition.add_member(:image_digest, Shapes::ShapeRef.new(shape: ImageDigest, location_name: "ImageDigest"))
@@ -9391,6 +9500,8 @@ module Aws::SageMaker
     ModelPackageContainerDefinition.add_member(:nearest_model_name, Shapes::ShapeRef.new(shape: String, location_name: "NearestModelName"))
     ModelPackageContainerDefinition.add_member(:additional_s3_data_source, Shapes::ShapeRef.new(shape: AdditionalS3DataSource, location_name: "AdditionalS3DataSource"))
     ModelPackageContainerDefinition.add_member(:model_data_etag, Shapes::ShapeRef.new(shape: String, location_name: "ModelDataETag"))
+    ModelPackageContainerDefinition.add_member(:is_checkpoint, Shapes::ShapeRef.new(shape: Boolean, location_name: "IsCheckpoint", metadata: {"box" => true}))
+    ModelPackageContainerDefinition.add_member(:base_model, Shapes::ShapeRef.new(shape: BaseModel, location_name: "BaseModel"))
     ModelPackageContainerDefinition.struct_class = Types::ModelPackageContainerDefinition
 
     ModelPackageContainerDefinitionList.member = Shapes::ShapeRef.new(shape: ModelPackageContainerDefinition)
@@ -9443,6 +9554,7 @@ module Aws::SageMaker
     ModelPackageSummary.add_member(:model_package_status, Shapes::ShapeRef.new(shape: ModelPackageStatus, required: true, location_name: "ModelPackageStatus"))
     ModelPackageSummary.add_member(:model_approval_status, Shapes::ShapeRef.new(shape: ModelApprovalStatus, location_name: "ModelApprovalStatus"))
     ModelPackageSummary.add_member(:model_life_cycle, Shapes::ShapeRef.new(shape: ModelLifeCycle, location_name: "ModelLifeCycle"))
+    ModelPackageSummary.add_member(:model_package_registration_type, Shapes::ShapeRef.new(shape: ModelPackageRegistrationType, location_name: "ModelPackageRegistrationType"))
     ModelPackageSummary.struct_class = Types::ModelPackageSummary
 
     ModelPackageSummaryList.member = Shapes::ShapeRef.new(shape: ModelPackageSummary)
@@ -10049,6 +10161,12 @@ module Aws::SageMaker
     PipelineExecutionStepMetadata.add_member(:auto_ml_job, Shapes::ShapeRef.new(shape: AutoMLJobStepMetadata, location_name: "AutoMLJob"))
     PipelineExecutionStepMetadata.add_member(:endpoint, Shapes::ShapeRef.new(shape: EndpointStepMetadata, location_name: "Endpoint"))
     PipelineExecutionStepMetadata.add_member(:endpoint_config, Shapes::ShapeRef.new(shape: EndpointConfigStepMetadata, location_name: "EndpointConfig"))
+    PipelineExecutionStepMetadata.add_member(:bedrock_custom_model, Shapes::ShapeRef.new(shape: BedrockCustomModelMetadata, location_name: "BedrockCustomModel"))
+    PipelineExecutionStepMetadata.add_member(:bedrock_custom_model_deployment, Shapes::ShapeRef.new(shape: BedrockCustomModelDeploymentMetadata, location_name: "BedrockCustomModelDeployment"))
+    PipelineExecutionStepMetadata.add_member(:bedrock_provisioned_model_throughput, Shapes::ShapeRef.new(shape: BedrockProvisionedModelThroughputMetadata, location_name: "BedrockProvisionedModelThroughput"))
+    PipelineExecutionStepMetadata.add_member(:bedrock_model_import, Shapes::ShapeRef.new(shape: BedrockModelImportMetadata, location_name: "BedrockModelImport"))
+    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.struct_class = Types::PipelineExecutionStepMetadata
 
     PipelineExecutionSummary.add_member(:pipeline_execution_arn, Shapes::ShapeRef.new(shape: PipelineExecutionArn, location_name: "PipelineExecutionArn"))
@@ -10850,6 +10968,15 @@ module Aws::SageMaker
     SendPipelineExecutionStepSuccessResponse.add_member(:pipeline_execution_arn, Shapes::ShapeRef.new(shape: PipelineExecutionArn, location_name: "PipelineExecutionArn"))
     SendPipelineExecutionStepSuccessResponse.struct_class = Types::SendPipelineExecutionStepSuccessResponse
 
+    ServerlessJobConfig.add_member(:base_model_arn, Shapes::ShapeRef.new(shape: ServerlessJobBaseModelArn, required: true, location_name: "BaseModelArn"))
+    ServerlessJobConfig.add_member(:accept_eula, Shapes::ShapeRef.new(shape: AcceptEula, location_name: "AcceptEula", metadata: {"box" => true}))
+    ServerlessJobConfig.add_member(:job_type, Shapes::ShapeRef.new(shape: ServerlessJobType, required: true, location_name: "JobType"))
+    ServerlessJobConfig.add_member(:customization_technique, Shapes::ShapeRef.new(shape: CustomizationTechnique, location_name: "CustomizationTechnique"))
+    ServerlessJobConfig.add_member(:peft, Shapes::ShapeRef.new(shape: Peft, location_name: "Peft"))
+    ServerlessJobConfig.add_member(:evaluation_type, Shapes::ShapeRef.new(shape: EvaluationType, location_name: "EvaluationType"))
+    ServerlessJobConfig.add_member(:evaluator_arn, Shapes::ShapeRef.new(shape: EvaluatorArn, location_name: "EvaluatorArn"))
+    ServerlessJobConfig.struct_class = Types::ServerlessJobConfig
+
     ServiceCatalogProvisionedProductDetails.add_member(:provisioned_product_id, Shapes::ShapeRef.new(shape: ServiceCatalogEntityId, location_name: "ProvisionedProductId"))
     ServiceCatalogProvisionedProductDetails.add_member(:provisioned_product_status_message, Shapes::ShapeRef.new(shape: ProvisionedProductStatusMessage, location_name: "ProvisionedProductStatusMessage"))
     ServiceCatalogProvisionedProductDetails.struct_class = Types::ServiceCatalogProvisionedProductDetails
@@ -10987,6 +11114,7 @@ module Aws::SageMaker
     StartPipelineExecutionRequest.add_member(:parallelism_configuration, Shapes::ShapeRef.new(shape: ParallelismConfiguration, location_name: "ParallelismConfiguration"))
     StartPipelineExecutionRequest.add_member(:selective_execution_config, Shapes::ShapeRef.new(shape: SelectiveExecutionConfig, location_name: "SelectiveExecutionConfig"))
     StartPipelineExecutionRequest.add_member(:pipeline_version_id, Shapes::ShapeRef.new(shape: PipelineVersionId, location_name: "PipelineVersionId"))
+    StartPipelineExecutionRequest.add_member(:mlflow_experiment_name, Shapes::ShapeRef.new(shape: MlflowExperimentEntityName, location_name: "MlflowExperimentName"))
     StartPipelineExecutionRequest.struct_class = Types::StartPipelineExecutionRequest
 
     StartPipelineExecutionResponse.add_member(:pipeline_execution_arn, Shapes::ShapeRef.new(shape: PipelineExecutionArn, location_name: "PipelineExecutionArn"))
@@ -11271,6 +11399,8 @@ module Aws::SageMaker
     TrainingJob.add_member(:debug_rule_configurations, Shapes::ShapeRef.new(shape: DebugRuleConfigurations, location_name: "DebugRuleConfigurations"))
     TrainingJob.add_member(:tensor_board_output_config, Shapes::ShapeRef.new(shape: TensorBoardOutputConfig, location_name: "TensorBoardOutputConfig"))
     TrainingJob.add_member(:debug_rule_evaluation_statuses, Shapes::ShapeRef.new(shape: DebugRuleEvaluationStatuses, location_name: "DebugRuleEvaluationStatuses"))
+    TrainingJob.add_member(:output_model_package_arn, Shapes::ShapeRef.new(shape: ModelPackageArn, location_name: "OutputModelPackageArn"))
+    TrainingJob.add_member(:model_package_config, Shapes::ShapeRef.new(shape: ModelPackageConfig, location_name: "ModelPackageConfig"))
     TrainingJob.add_member(:profiler_config, Shapes::ShapeRef.new(shape: ProfilerConfig, location_name: "ProfilerConfig"))
     TrainingJob.add_member(:environment, Shapes::ShapeRef.new(shape: TrainingEnvironmentMap, location_name: "Environment"))
     TrainingJob.add_member(:retry_strategy, Shapes::ShapeRef.new(shape: RetryStrategy, location_name: "RetryStrategy"))
@@ -11349,6 +11479,12 @@ module Aws::SageMaker
     TrainingPlanSummary.add_member(:reserved_capacity_summaries, Shapes::ShapeRef.new(shape: ReservedCapacitySummaries, location_name: "ReservedCapacitySummaries"))
     TrainingPlanSummary.struct_class = Types::TrainingPlanSummary
 
+    TrainingProgressInfo.add_member(:total_step_count_per_epoch, Shapes::ShapeRef.new(shape: TotalStepCountPerEpoch, location_name: "TotalStepCountPerEpoch"))
+    TrainingProgressInfo.add_member(:current_step, Shapes::ShapeRef.new(shape: TrainingStepIndex, location_name: "CurrentStep"))
+    TrainingProgressInfo.add_member(:current_epoch, Shapes::ShapeRef.new(shape: TrainingEpochIndex, location_name: "CurrentEpoch"))
+    TrainingProgressInfo.add_member(:max_epoch, Shapes::ShapeRef.new(shape: TrainingEpochCount, location_name: "MaxEpoch"))
+    TrainingProgressInfo.struct_class = Types::TrainingProgressInfo
+
     TrainingRepositoryAuthConfig.add_member(:training_repository_credentials_provider_arn, Shapes::ShapeRef.new(shape: TrainingRepositoryCredentialsProviderArn, required: true, location_name: "TrainingRepositoryCredentialsProviderArn"))
     TrainingRepositoryAuthConfig.struct_class = Types::TrainingRepositoryAuthConfig
 
@@ -11905,6 +12041,7 @@ module Aws::SageMaker
 
     UpdateModelPackageInput.add_member(:model_package_arn, Shapes::ShapeRef.new(shape: ModelPackageArn, required: true, location_name: "ModelPackageArn"))
     UpdateModelPackageInput.add_member(:model_approval_status, Shapes::ShapeRef.new(shape: ModelApprovalStatus, location_name: "ModelApprovalStatus"))
+    UpdateModelPackageInput.add_member(:model_package_registration_type, Shapes::ShapeRef.new(shape: ModelPackageRegistrationType, location_name: "ModelPackageRegistrationType"))
     UpdateModelPackageInput.add_member(:approval_description, Shapes::ShapeRef.new(shape: ApprovalDescription, location_name: "ApprovalDescription"))
     UpdateModelPackageInput.add_member(:customer_metadata_properties, Shapes::ShapeRef.new(shape: CustomerMetadataMap, location_name: "CustomerMetadataProperties"))
     UpdateModelPackageInput.add_member(:customer_metadata_properties_to_remove, Shapes::ShapeRef.new(shape: CustomerMetadataKeyList, location_name: "CustomerMetadataPropertiesToRemove"))