aws-sdk-sagemaker 1.370.0 → 1.372.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 498cd22927a5354b0f77c1259519c4cd48cc4a5f3de5ba03badfe63f5c246a12
-  data.tar.gz: e28559dee29925ffc3bf4bb11dfc302f6735ec7a64e7f9ddf6ef9392a90a872d
+  metadata.gz: e6399d5cfc5d2d703e98f466dd3d1899884b3f50cd13ec05b5789b0071b1413f
+  data.tar.gz: 48b38958f579fb434e67ec60888f91effcd6df6d7149da6c02f873bcc5ef4470
 SHA512:
-  metadata.gz: 76676db4011ffbae317d8b620e71caa64b95ae55d7e63de710c88198c7c9ddca10a3b5a09fdd80a5e61c513c8a6958dfc8d8c8c16776fa68962fa9ee84684e26
-  data.tar.gz: a40369691b61de8eed6ed50349d46182e10dc2baa99dc67b84ee522894eaa1d6409fe9ff956b81c204dd9d1611c78ac24740df6b65fe5ba345a8e1d9b9931d82
+  metadata.gz: 2f04335adbc52e3257790d17db0b45ae4647872a5c49481ab650d1b83eb6b99409b30e99e750d9b5ff53ca1731f61768d82edf67d8ef490790ac179ab54a4b5a
+  data.tar.gz: 70f6d4de733005162ad7083a3af6e89b236c9b42d4f8f6e7299e3503d37e96cede2dc65204c2f201261b2c75ac066181fe42944a33267bb0524055fd3d7a394e

data/CHANGELOG.md

@@ -1,6 +1,16 @@
 Unreleased Changes
 ------------------
 
+1.372.0 (2026-06-05)
+------------------
+
+* Feature - This release adds support for MLflow experiment tracking in SageMaker inference optimization. CreateAIRecommendationJob and CreateAIBenchmarkJob now accept an optional OutputConfig.MlflowConfig (MLflow App ARN, experiment, run name) to stream benchmark metrics and artifacts to your own MLflow App.
+
+1.371.0 (2026-06-04)
+------------------
+
+* Feature - Adds the IncludedData parameter to DescribeModelCard and DescribeModelPackage. Set it to MetadataOnly to retrieve a model card without decrypt permission on the customer managed AWS KMS key (default AllData returns full content). Adds support for the MTRL Job resource in SageMaker Search.
+
 1.370.0 (2026-06-02)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.370.0
+1.372.0

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

@@ -1231,6 +1231,11 @@ module Aws::SageMaker
     #     },
     #     output_config: { # required
     #       s3_output_location: "S3Uri", # required
+    #       mlflow_config: {
+    #         mlflow_resource_arn: "AIMlflowResourceArn", # required
+    #         mlflow_experiment_name: "AIMlflowExperimentName",
+    #         mlflow_run_name: "AIMlflowRunName",
+    #       },
     #     },
     #     ai_workload_config_identifier: "AIResourceIdentifier", # required
     #     role_arn: "RoleArn", # required
@@ -1328,6 +1333,11 @@ module Aws::SageMaker
     #     output_config: { # required
     #       s3_output_location: "S3Uri",
     #       model_package_group_identifier: "AIResourceIdentifier",
+    #       mlflow_config: {
+    #         mlflow_resource_arn: "AIMlflowResourceArn", # required
+    #         mlflow_experiment_name: "AIMlflowExperimentName",
+    #         mlflow_run_name: "AIMlflowRunName",
+    #       },
     #     },
     #     ai_workload_config_identifier: "AIResourceIdentifier", # required
     #     performance_target: { # required
@@ -13884,6 +13894,9 @@ module Aws::SageMaker
     #   resp.output_config.cloud_watch_logs #=> Array
     #   resp.output_config.cloud_watch_logs[0].log_group_arn #=> String
     #   resp.output_config.cloud_watch_logs[0].log_stream_name #=> String
+    #   resp.output_config.mlflow_config.mlflow_resource_arn #=> String
+    #   resp.output_config.mlflow_config.mlflow_experiment_name #=> String
+    #   resp.output_config.mlflow_config.mlflow_run_name #=> String
     #   resp.ai_workload_config_identifier #=> String
     #   resp.role_arn #=> String
     #   resp.network_config.vpc_config.security_group_ids #=> Array
@@ -13948,6 +13961,9 @@ module Aws::SageMaker
     #   resp.model_source.s3.s3_uri #=> String
     #   resp.output_config.s3_output_location #=> String
     #   resp.output_config.model_package_group_identifier #=> String
+    #   resp.output_config.mlflow_config.mlflow_resource_arn #=> String
+    #   resp.output_config.mlflow_config.mlflow_experiment_name #=> String
+    #   resp.output_config.mlflow_config.mlflow_run_name #=> String
     #   resp.inference_specification.framework #=> String, one of "LMI", "VLLM"
     #   resp.ai_workload_config_identifier #=> String
     #   resp.optimize_model #=> Boolean
@@ -18334,6 +18350,12 @@ module Aws::SageMaker
     # Describes the content, creation time, and security configuration of an
     # Amazon SageMaker Model Card.
     #
+    # To retrieve only metadata about a model card without requiring
+    # `kms:Decrypt` permission on the associated customer-managed Amazon Web
+    # Services KMS key, set `IncludedData` to `MetadataOnly`. The default is
+    # `AllData`, which returns the full model card `Content` and requires
+    # `kms:Decrypt` permission when a customer-managed key is configured.
+    #
     # @option params [required, String] :model_card_name
     #   The name or Amazon Resource Name (ARN) of the model card to describe.
     #
@@ -18341,6 +18363,23 @@ module Aws::SageMaker
     #   The version of the model card to describe. If a version is not
     #   provided, then the latest version of the model card is described.
     #
+    # @option params [String] :included_data
+    #   Specifies the level of model card data to include in the response. Use
+    #   this parameter to call `DescribeModelCard` without requiring
+    #   `kms:Decrypt` permission on the customer-managed Amazon Web Services
+    #   KMS key.
+    #
+    #   * `AllData`: Returns the full model card `Content`. This option
+    #     requires `kms:Decrypt` permission on the customer-managed key, if
+    #     one is associated with the model card. This is the default.
+    #
+    #   * `MetadataOnly`: Returns the model card with sanitized `Content` that
+    #     includes only a small set of unencrypted metadata fields. This
+    #     option does not require `kms:Decrypt` permission. For the list of
+    #     fields preserved in the response, see `Content`.
+    #
+    #   If you don't specify a value, SageMaker returns `AllData`.
+    #
     # @return [Types::DescribeModelCardResponse] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
     #
     #   * {Types::DescribeModelCardResponse#model_card_arn #model_card_arn} => String
@@ -18360,6 +18399,7 @@ module Aws::SageMaker
     #   resp = client.describe_model_card({
     #     model_card_name: "ModelCardNameOrArn", # required
     #     model_card_version: 1,
+    #     included_data: "AllData", # accepts AllData, MetadataOnly
     #   })
     #
     # @example Response structure
@@ -18537,7 +18577,10 @@ module Aws::SageMaker
     #
     # If you provided a KMS Key ID when you created your model package, you
     # will see the [KMS Decrypt][1] API call in your CloudTrail logs when
-    # you use this API.
+    # you use this API. To call this operation without requiring
+    # `kms:Decrypt` permission on the customer-managed key, set
+    # `IncludedData` to `MetadataOnly`; the response is returned with the
+    # embedded `ModelCard.ModelCardContent` field sanitized.
     #
     # To create models in SageMaker, buyers can subscribe to model packages
     # listed on Amazon Web Services Marketplace.
@@ -18553,6 +18596,33 @@ module Aws::SageMaker
     #   When you specify a name, the name must have 1 to 63 characters. Valid
     #   characters are a-z, A-Z, 0-9, and - (hyphen).
     #
+    # @option params [String] :included_data
+    #   Specifies the level of model package data to include in the response.
+    #   Use this parameter to call `DescribeModelPackage` on a model package
+    #   that has an associated model card without requiring `kms:Decrypt`
+    #   permission on the customer-managed KMS key associated with the
+    #   embedded model card.
+    #
+    #   * `AllData`: Returns the full model package response, including the
+    #     unredacted `ModelCard.ModelCardContent`. This option requires
+    #     `kms:Decrypt` permission on the customer-managed key, if one is
+    #     associated with the embedded model card. This is the default.
+    #
+    #   * `MetadataOnly`: Returns the full model package response, but with
+    #     the embedded `ModelCard.ModelCardContent` sanitized to include only
+    #     a small set of unencrypted metadata fields. This option does not
+    #     require `kms:Decrypt` permission. All other top-level response
+    #     fields, including `InferenceSpecification`, `ModelMetrics`,
+    #     `DriftCheckBaselines`, and `SecurityConfig`, are returned unchanged.
+    #     For the list of fields preserved within `ModelCardContent`, see
+    #     [ModelCard][1].
+    #
+    #   If you don't specify a value, SageMaker returns `AllData`.
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_DescribeModelPackage.html#sagemaker-DescribeModelPackage-response-ModelCard
+    #
     # @return [Types::DescribeModelPackageOutput] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
     #
     #   * {Types::DescribeModelPackageOutput#model_package_name #model_package_name} => String
@@ -18592,6 +18662,7 @@ module Aws::SageMaker
     #
     #   resp = client.describe_model_package({
     #     model_package_name: "VersionedArnOrName", # required
+    #     included_data: "AllData", # accepts AllData, MetadataOnly
     #   })
     #
     # @example Response structure
@@ -21330,7 +21401,7 @@ module Aws::SageMaker
     # @example Request syntax with placeholder values
     #
     #   resp = client.get_search_suggestions({
-    #     resource: "TrainingJob", # required, accepts TrainingJob, Experiment, ExperimentTrial, ExperimentTrialComponent, Endpoint, Model, ModelPackage, ModelPackageGroup, Pipeline, PipelineExecution, FeatureGroup, FeatureMetadata, Image, ImageVersion, Project, HyperParameterTuningJob, ModelCard, PipelineVersion
+    #     resource: "TrainingJob", # required, accepts TrainingJob, Experiment, ExperimentTrial, ExperimentTrialComponent, Endpoint, Model, ModelPackage, ModelPackageGroup, Pipeline, PipelineExecution, FeatureGroup, FeatureMetadata, Image, ImageVersion, Project, HyperParameterTuningJob, ModelCard, PipelineVersion, Job
     #     suggestion_query: {
     #       property_name_query: {
     #         property_name_hint: "PropertyNameHint", # required
@@ -28846,7 +28917,7 @@ module Aws::SageMaker
     # @example Request syntax with placeholder values
     #
     #   resp = client.search({
-    #     resource: "TrainingJob", # required, accepts TrainingJob, Experiment, ExperimentTrial, ExperimentTrialComponent, Endpoint, Model, ModelPackage, ModelPackageGroup, Pipeline, PipelineExecution, FeatureGroup, FeatureMetadata, Image, ImageVersion, Project, HyperParameterTuningJob, ModelCard, PipelineVersion
+    #     resource: "TrainingJob", # required, accepts TrainingJob, Experiment, ExperimentTrial, ExperimentTrialComponent, Endpoint, Model, ModelPackage, ModelPackageGroup, Pipeline, PipelineExecution, FeatureGroup, FeatureMetadata, Image, ImageVersion, Project, HyperParameterTuningJob, ModelCard, PipelineVersion, Job
     #     search_expression: {
     #       filters: [
     #         {
@@ -34308,7 +34379,7 @@ module Aws::SageMaker
         tracer: tracer
       )
       context[:gem_name] = 'aws-sdk-sagemaker'
-      context[:gem_version] = '1.370.0'
+      context[:gem_version] = '1.372.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -35,6 +35,10 @@ module Aws::SageMaker
     AIInferenceSpecificationName = Shapes::StringShape.new(name: 'AIInferenceSpecificationName')
     AIMlReservationArn = Shapes::StringShape.new(name: 'AIMlReservationArn')
     AIMlReservationArnList = Shapes::ListShape.new(name: 'AIMlReservationArnList')
+    AIMlflowConfig = Shapes::StructureShape.new(name: 'AIMlflowConfig')
+    AIMlflowExperimentName = Shapes::StringShape.new(name: 'AIMlflowExperimentName')
+    AIMlflowResourceArn = Shapes::StringShape.new(name: 'AIMlflowResourceArn')
+    AIMlflowRunName = Shapes::StringShape.new(name: 'AIMlflowRunName')
     AIModelSource = Shapes::UnionShape.new(name: 'AIModelSource')
     AIModelSourceS3 = Shapes::StructureShape.new(name: 'AIModelSourceS3')
     AIRecommendation = Shapes::StructureShape.new(name: 'AIRecommendation')
@@ -1443,6 +1447,7 @@ module Aws::SageMaker
     ImportHubContentResponse = Shapes::StructureShape.new(name: 'ImportHubContentResponse')
     InUseInstanceCount = Shapes::IntegerShape.new(name: 'InUseInstanceCount')
     IncludeNodeLogicalIdsBoolean = Shapes::BooleanShape.new(name: 'IncludeNodeLogicalIdsBoolean')
+    IncludedData = Shapes::StringShape.new(name: 'IncludedData')
     InferenceComponentArn = Shapes::StringShape.new(name: 'InferenceComponentArn')
     InferenceComponentAvailabilityZoneBalance = Shapes::StructureShape.new(name: 'InferenceComponentAvailabilityZoneBalance')
     InferenceComponentCapacitySize = Shapes::StructureShape.new(name: 'InferenceComponentCapacitySize')
@@ -1538,6 +1543,7 @@ module Aws::SageMaker
     IotRoleAlias = Shapes::StringShape.new(name: 'IotRoleAlias')
     IsTrackingServerActive = Shapes::StringShape.new(name: 'IsTrackingServerActive')
     ItemIdentifierAttributeName = Shapes::StringShape.new(name: 'ItemIdentifierAttributeName')
+    Job = Shapes::StructureShape.new(name: 'Job')
     JobArn = Shapes::StringShape.new(name: 'JobArn')
     JobCategory = Shapes::StringShape.new(name: 'JobCategory')
     JobConfigDocument = Shapes::StringShape.new(name: 'JobConfigDocument')
@@ -3038,10 +3044,12 @@ module Aws::SageMaker
     AIBenchmarkNetworkConfig.struct_class = Types::AIBenchmarkNetworkConfig
 
     AIBenchmarkOutputConfig.add_member(:s3_output_location, Shapes::ShapeRef.new(shape: S3Uri, required: true, location_name: "S3OutputLocation"))
+    AIBenchmarkOutputConfig.add_member(:mlflow_config, Shapes::ShapeRef.new(shape: AIMlflowConfig, location_name: "MlflowConfig"))
     AIBenchmarkOutputConfig.struct_class = Types::AIBenchmarkOutputConfig
 
     AIBenchmarkOutputResult.add_member(:s3_output_location, Shapes::ShapeRef.new(shape: S3Uri, required: true, location_name: "S3OutputLocation"))
     AIBenchmarkOutputResult.add_member(:cloud_watch_logs, Shapes::ShapeRef.new(shape: AICloudWatchLogsList, location_name: "CloudWatchLogs"))
+    AIBenchmarkOutputResult.add_member(:mlflow_config, Shapes::ShapeRef.new(shape: AIMlflowConfig, location_name: "MlflowConfig"))
     AIBenchmarkOutputResult.struct_class = Types::AIBenchmarkOutputResult
 
     AIBenchmarkTarget.add_member(:endpoint, Shapes::ShapeRef.new(shape: AIBenchmarkEndpoint, location_name: "Endpoint"))
@@ -3068,6 +3076,11 @@ module Aws::SageMaker
 
     AIMlReservationArnList.member = Shapes::ShapeRef.new(shape: AIMlReservationArn)
 
+    AIMlflowConfig.add_member(:mlflow_resource_arn, Shapes::ShapeRef.new(shape: AIMlflowResourceArn, required: true, location_name: "MlflowResourceArn"))
+    AIMlflowConfig.add_member(:mlflow_experiment_name, Shapes::ShapeRef.new(shape: AIMlflowExperimentName, location_name: "MlflowExperimentName"))
+    AIMlflowConfig.add_member(:mlflow_run_name, Shapes::ShapeRef.new(shape: AIMlflowRunName, location_name: "MlflowRunName"))
+    AIMlflowConfig.struct_class = Types::AIMlflowConfig
+
     AIModelSource.add_member(:s3, Shapes::ShapeRef.new(shape: AIModelSourceS3, location_name: "S3"))
     AIModelSource.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
     AIModelSource.add_member_subclass(:s3, Types::AIModelSource::S3)
@@ -3147,10 +3160,12 @@ module Aws::SageMaker
 
     AIRecommendationOutputConfig.add_member(:s3_output_location, Shapes::ShapeRef.new(shape: S3Uri, location_name: "S3OutputLocation"))
     AIRecommendationOutputConfig.add_member(:model_package_group_identifier, Shapes::ShapeRef.new(shape: AIResourceIdentifier, location_name: "ModelPackageGroupIdentifier"))
+    AIRecommendationOutputConfig.add_member(:mlflow_config, Shapes::ShapeRef.new(shape: AIMlflowConfig, location_name: "MlflowConfig"))
     AIRecommendationOutputConfig.struct_class = Types::AIRecommendationOutputConfig
 
     AIRecommendationOutputResult.add_member(:s3_output_location, Shapes::ShapeRef.new(shape: S3Uri, required: true, location_name: "S3OutputLocation"))
     AIRecommendationOutputResult.add_member(:model_package_group_identifier, Shapes::ShapeRef.new(shape: AIResourceIdentifier, location_name: "ModelPackageGroupIdentifier"))
+    AIRecommendationOutputResult.add_member(:mlflow_config, Shapes::ShapeRef.new(shape: AIMlflowConfig, location_name: "MlflowConfig"))
     AIRecommendationOutputResult.struct_class = Types::AIRecommendationOutputResult
 
     AIRecommendationPerformanceMetric.add_member(:metric, Shapes::ShapeRef.new(shape: String, required: true, location_name: "Metric"))
@@ -6733,6 +6748,7 @@ module Aws::SageMaker
 
     DescribeModelCardRequest.add_member(:model_card_name, Shapes::ShapeRef.new(shape: ModelCardNameOrArn, required: true, location_name: "ModelCardName"))
     DescribeModelCardRequest.add_member(:model_card_version, Shapes::ShapeRef.new(shape: Integer, location_name: "ModelCardVersion", metadata: {"box" => true}))
+    DescribeModelCardRequest.add_member(:included_data, Shapes::ShapeRef.new(shape: IncludedData, location_name: "IncludedData"))
     DescribeModelCardRequest.struct_class = Types::DescribeModelCardRequest
 
     DescribeModelCardResponse.add_member(:model_card_arn, Shapes::ShapeRef.new(shape: ModelCardArn, required: true, location_name: "ModelCardArn"))
@@ -6792,6 +6808,7 @@ module Aws::SageMaker
     DescribeModelPackageGroupOutput.struct_class = Types::DescribeModelPackageGroupOutput
 
     DescribeModelPackageInput.add_member(:model_package_name, Shapes::ShapeRef.new(shape: VersionedArnOrName, required: true, location_name: "ModelPackageName"))
+    DescribeModelPackageInput.add_member(:included_data, Shapes::ShapeRef.new(shape: IncludedData, location_name: "IncludedData"))
     DescribeModelPackageInput.struct_class = Types::DescribeModelPackageInput
 
     DescribeModelPackageOutput.add_member(:model_package_name, Shapes::ShapeRef.new(shape: EntityName, required: true, location_name: "ModelPackageName"))
@@ -8514,6 +8531,22 @@ module Aws::SageMaker
 
     IntegerParameterRanges.member = Shapes::ShapeRef.new(shape: IntegerParameterRange)
 
+    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"))
+    Job.add_member(:job_category, Shapes::ShapeRef.new(shape: JobCategory, location_name: "JobCategory"))
+    Job.add_member(:job_config_schema_version, Shapes::ShapeRef.new(shape: JobSchemaVersion, location_name: "JobConfigSchemaVersion"))
+    Job.add_member(:job_config_document, Shapes::ShapeRef.new(shape: JobConfigDocument, location_name: "JobConfigDocument"))
+    Job.add_member(:creation_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "CreationTime"))
+    Job.add_member(:last_modified_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "LastModifiedTime"))
+    Job.add_member(:end_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "EndTime"))
+    Job.add_member(:job_status, Shapes::ShapeRef.new(shape: JobStatus, location_name: "JobStatus"))
+    Job.add_member(:secondary_status, Shapes::ShapeRef.new(shape: JobSecondaryStatus, location_name: "SecondaryStatus"))
+    Job.add_member(:secondary_status_transitions, Shapes::ShapeRef.new(shape: JobSecondaryStatusTransitions, location_name: "SecondaryStatusTransitions"))
+    Job.add_member(:failure_reason, Shapes::ShapeRef.new(shape: FailureReason, location_name: "FailureReason"))
+    Job.add_member(:tags, Shapes::ShapeRef.new(shape: TagList, location_name: "Tags"))
+    Job.struct_class = Types::Job
+
     JobConfigSchemaVersionSummary.add_member(:job_config_schema_version, Shapes::ShapeRef.new(shape: JobSchemaVersion, required: true, location_name: "JobConfigSchemaVersion"))
     JobConfigSchemaVersionSummary.struct_class = Types::JobConfigSchemaVersionSummary
 
@@ -11710,6 +11743,7 @@ module Aws::SageMaker
     SearchRecord.add_member(:hyper_parameter_tuning_job, Shapes::ShapeRef.new(shape: HyperParameterTuningJobSearchEntity, location_name: "HyperParameterTuningJob"))
     SearchRecord.add_member(:model_card, Shapes::ShapeRef.new(shape: ModelCard, location_name: "ModelCard"))
     SearchRecord.add_member(:model, Shapes::ShapeRef.new(shape: ModelDashboardModel, location_name: "Model"))
+    SearchRecord.add_member(:job, Shapes::ShapeRef.new(shape: Job, location_name: "Job"))
     SearchRecord.struct_class = Types::SearchRecord
 
     SearchRequest.add_member(:resource, Shapes::ShapeRef.new(shape: ResourceType, required: true, location_name: "Resource"))

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

@@ -109,10 +109,16 @@ module Aws::SageMaker
     #   The Amazon S3 URI where benchmark results are stored.
     #   @return [String]
     #
+    # @!attribute [rw] mlflow_config
+    #   The MLflow tracking configuration for the job. If you don't specify
+    #   this parameter, MLflow tracking is disabled.
+    #   @return [Types::AIMlflowConfig]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/AIBenchmarkOutputConfig AWS API Documentation
     #
     class AIBenchmarkOutputConfig < Struct.new(
-      :s3_output_location)
+      :s3_output_location,
+      :mlflow_config)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -128,11 +134,16 @@ module Aws::SageMaker
     #   The CloudWatch log information for the benchmark job.
     #   @return [Array<Types::AICloudWatchLogs>]
     #
+    # @!attribute [rw] mlflow_config
+    #   The MLflow tracking configuration for the job.
+    #   @return [Types::AIMlflowConfig]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/AIBenchmarkOutputResult AWS API Documentation
     #
     class AIBenchmarkOutputResult < Struct.new(
       :s3_output_location,
-      :cloud_watch_logs)
+      :cloud_watch_logs,
+      :mlflow_config)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -224,6 +235,32 @@ module Aws::SageMaker
       class Unknown < AIDatasetConfig; end
     end
 
+    # The MLflow tracking configuration for logging metrics and parameters
+    # to a SageMaker managed MLflow tracking server.
+    #
+    # @!attribute [rw] mlflow_resource_arn
+    #   The Amazon Resource Name (ARN) of the SageMaker managed MLflow
+    #   resource.
+    #   @return [String]
+    #
+    # @!attribute [rw] mlflow_experiment_name
+    #   The MLflow experiment name used for tracking.
+    #   @return [String]
+    #
+    # @!attribute [rw] mlflow_run_name
+    #   The MLflow run name used for tracking.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/AIMlflowConfig AWS API Documentation
+    #
+    class AIMlflowConfig < Struct.new(
+      :mlflow_resource_arn,
+      :mlflow_experiment_name,
+      :mlflow_run_name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The source of the model for an AI recommendation job. This is a union
     # type.
     #
@@ -527,11 +564,17 @@ module Aws::SageMaker
     #   version.
     #   @return [String]
     #
+    # @!attribute [rw] mlflow_config
+    #   The MLflow tracking configuration for the job. If you don't specify
+    #   this parameter, MLflow tracking is disabled.
+    #   @return [Types::AIMlflowConfig]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/AIRecommendationOutputConfig AWS API Documentation
     #
     class AIRecommendationOutputConfig < Struct.new(
       :s3_output_location,
-      :model_package_group_identifier)
+      :model_package_group_identifier,
+      :mlflow_config)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -549,11 +592,16 @@ module Aws::SageMaker
     #   where deployment-ready model packages are registered.
     #   @return [String]
     #
+    # @!attribute [rw] mlflow_config
+    #   The MLflow tracking configuration for the job.
+    #   @return [Types::AIMlflowConfig]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/AIRecommendationOutputResult AWS API Documentation
     #
     class AIRecommendationOutputResult < Struct.new(
       :s3_output_location,
-      :model_package_group_identifier)
+      :model_package_group_identifier,
+      :mlflow_config)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -21331,11 +21379,30 @@ module Aws::SageMaker
     #   provided, then the latest version of the model card is described.
     #   @return [Integer]
     #
+    # @!attribute [rw] included_data
+    #   Specifies the level of model card data to include in the response.
+    #   Use this parameter to call `DescribeModelCard` without requiring
+    #   `kms:Decrypt` permission on the customer-managed Amazon Web Services
+    #   KMS key.
+    #
+    #   * `AllData`: Returns the full model card `Content`. This option
+    #     requires `kms:Decrypt` permission on the customer-managed key, if
+    #     one is associated with the model card. This is the default.
+    #
+    #   * `MetadataOnly`: Returns the model card with sanitized `Content`
+    #     that includes only a small set of unencrypted metadata fields.
+    #     This option does not require `kms:Decrypt` permission. For the
+    #     list of fields preserved in the response, see `Content`.
+    #
+    #   If you don't specify a value, SageMaker returns `AllData`.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeModelCardRequest AWS API Documentation
     #
     class DescribeModelCardRequest < Struct.new(
       :model_card_name,
-      :model_card_version)
+      :model_card_version,
+      :included_data)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -21353,7 +21420,32 @@ module Aws::SageMaker
     #   @return [Integer]
     #
     # @!attribute [rw] content
-    #   The content of the model card.
+    #   The content of the model card. Content is provided as a string in
+    #   the [model card JSON schema][1].
+    #
+    #   When you set `IncludedData` to `MetadataOnly` in the request,
+    #   SageMaker returns a sanitized version of `Content` that includes
+    #   only the following JSON paths, when present in the model card:
+    #
+    #   * `model_overview.model_id`
+    #
+    #   * `model_overview.model_name`
+    #
+    #   * `intended_uses.risk_rating`
+    #
+    #   * `model_package_details.model_package_group_name`
+    #
+    #   * `model_package_details.model_package_arn`
+    #
+    #   All other fields are removed from `Content` when `IncludedData` is
+    #   `MetadataOnly`, including model description, training details,
+    #   evaluation details, business details, and additional information. To
+    #   retrieve the complete `Content`, set `IncludedData` to `AllData` or
+    #   omit the parameter.
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/sagemaker/latest/dg/model-cards.html#model-cards-json-schema
     #   @return [String]
     #
     # @!attribute [rw] model_card_status
@@ -21656,10 +21748,39 @@ module Aws::SageMaker
     #   Valid characters are a-z, A-Z, 0-9, and - (hyphen).
     #   @return [String]
     #
+    # @!attribute [rw] included_data
+    #   Specifies the level of model package data to include in the
+    #   response. Use this parameter to call `DescribeModelPackage` on a
+    #   model package that has an associated model card without requiring
+    #   `kms:Decrypt` permission on the customer-managed KMS key associated
+    #   with the embedded model card.
+    #
+    #   * `AllData`: Returns the full model package response, including the
+    #     unredacted `ModelCard.ModelCardContent`. This option requires
+    #     `kms:Decrypt` permission on the customer-managed key, if one is
+    #     associated with the embedded model card. This is the default.
+    #
+    #   * `MetadataOnly`: Returns the full model package response, but with
+    #     the embedded `ModelCard.ModelCardContent` sanitized to include
+    #     only a small set of unencrypted metadata fields. This option does
+    #     not require `kms:Decrypt` permission. All other top-level response
+    #     fields, including `InferenceSpecification`, `ModelMetrics`,
+    #     `DriftCheckBaselines`, and `SecurityConfig`, are returned
+    #     unchanged. For the list of fields preserved within
+    #     `ModelCardContent`, see [ModelCard][1].
+    #
+    #   If you don't specify a value, SageMaker returns `AllData`.
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_DescribeModelPackage.html#sagemaker-DescribeModelPackage-response-ModelCard
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeModelPackageInput AWS API Documentation
     #
     class DescribeModelPackageInput < Struct.new(
-      :model_package_name)
+      :model_package_name,
+      :included_data)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -21817,6 +21938,29 @@ module Aws::SageMaker
     #   card associated with the model package, see [View the Details of a
     #   Model Version][2].
     #
+    #   When you set `IncludedData` to `MetadataOnly` in the request,
+    #   `ModelCardStatus` is preserved and `ModelCardContent` is sanitized
+    #   to include only the following JSON paths, when present in the model
+    #   card:
+    #
+    #   * `model_overview.model_id`
+    #
+    #   * `model_overview.model_name`
+    #
+    #   * `intended_uses.risk_rating`
+    #
+    #   * `model_package_details.model_package_group_name`
+    #
+    #   * `model_package_details.model_package_arn`
+    #
+    #   Because the `ModelPackageModelCard` schema does not include
+    #   `model_package_details` and limits `model_overview` to
+    #   `model_creator` and `model_artifact`, the sanitized
+    #   `ModelCardContent` for a model package typically contains only
+    #   `intended_uses.risk_rating` if it was provided when the model card
+    #   was created. To retrieve the complete `ModelCardContent`, set
+    #   `IncludedData` to `AllData` or omit the parameter.
+    #
     #
     #
     #   [1]: https://docs.aws.amazon.com/sagemaker/latest/dg/model-registry-details.html#model-card-schema
@@ -32179,6 +32323,88 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Search shape for Job. Mirrors DescribeJobResponse fields. If you
+    # update DescribeJobResponse, update this structure as well.
+    #
+    # @!attribute [rw] job_name
+    #   The name of the job.
+    #   @return [String]
+    #
+    # @!attribute [rw] job_arn
+    #   The Amazon Resource Name (ARN) of the job.
+    #   @return [String]
+    #
+    # @!attribute [rw] role_arn
+    #   The ARN of the IAM role associated with the job.
+    #   @return [String]
+    #
+    # @!attribute [rw] job_category
+    #   The category of the job.
+    #   @return [String]
+    #
+    # @!attribute [rw] job_config_schema_version
+    #   The schema version used for the job configuration document.
+    #   @return [String]
+    #
+    # @!attribute [rw] job_config_document
+    #   The JSON configuration document for the job.
+    #   @return [String]
+    #
+    # @!attribute [rw] creation_time
+    #   The date and time that the job was created.
+    #   @return [Time]
+    #
+    # @!attribute [rw] last_modified_time
+    #   The date and time that the job was last modified.
+    #   @return [Time]
+    #
+    # @!attribute [rw] end_time
+    #   The date and time that the job ended.
+    #   @return [Time]
+    #
+    # @!attribute [rw] job_status
+    #   The current status of the job.
+    #   @return [String]
+    #
+    # @!attribute [rw] secondary_status
+    #   The detailed secondary status of the job, providing more granular
+    #   information about the job's progress.
+    #   @return [String]
+    #
+    # @!attribute [rw] secondary_status_transitions
+    #   A list of secondary status transitions for the job, with timestamps
+    #   and optional status messages.
+    #   @return [Array<Types::JobSecondaryStatusTransition>]
+    #
+    # @!attribute [rw] failure_reason
+    #   If the job failed, the reason it failed.
+    #   @return [String]
+    #
+    # @!attribute [rw] tags
+    #   The tags associated with the job.
+    #   @return [Array<Types::Tag>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/Job AWS API Documentation
+    #
+    class Job < Struct.new(
+      :job_name,
+      :job_arn,
+      :role_arn,
+      :job_category,
+      :job_config_schema_version,
+      :job_config_document,
+      :creation_time,
+      :last_modified_time,
+      :end_time,
+      :job_status,
+      :secondary_status,
+      :secondary_status_transitions,
+      :failure_reason,
+      :tags)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Provides summary information about a job configuration schema version.
     #
     # @!attribute [rw] job_config_schema_version
@@ -49884,6 +50110,11 @@ module Aws::SageMaker
     #   A model displayed in the Amazon SageMaker Model Dashboard.
     #   @return [Types::ModelDashboardModel]
     #
+    # @!attribute [rw] job
+    #   Search shape for Job. Mirrors DescribeJobResponse fields. If you
+    #   update DescribeJobResponse, update this structure as well.
+    #   @return [Types::Job]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/SearchRecord AWS API Documentation
     #
     class SearchRecord < Struct.new(
@@ -49902,7 +50133,8 @@ module Aws::SageMaker
       :project,
       :hyper_parameter_tuning_job,
       :model_card,
-      :model)
+      :model,
+      :job)
       SENSITIVE = []
       include Aws::Structure
     end

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.370.0'
+  GEM_VERSION = '1.372.0'
 
 end
 

data/sig/client.rbs

@@ -232,7 +232,12 @@ module Aws
                                        }?
                                      },
                                      output_config: {
-                                       s3_output_location: ::String
+                                       s3_output_location: ::String,
+                                       mlflow_config: {
+                                         mlflow_resource_arn: ::String,
+                                         mlflow_experiment_name: ::String?,
+                                         mlflow_run_name: ::String?
+                                       }?
                                      },
                                      ai_workload_config_identifier: ::String,
                                      role_arn: ::String,
@@ -265,7 +270,12 @@ module Aws
                                           },
                                           output_config: {
                                             s3_output_location: ::String?,
-                                            model_package_group_identifier: ::String?
+                                            model_package_group_identifier: ::String?,
+                                            mlflow_config: {
+                                              mlflow_resource_arn: ::String,
+                                              mlflow_experiment_name: ::String?,
+                                              mlflow_run_name: ::String?
+                                            }?
                                           },
                                           ai_workload_config_identifier: ::String,
                                           performance_target: {
@@ -4535,7 +4545,8 @@ module Aws
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#describe_model_card-instance_method
       def describe_model_card: (
                                  model_card_name: ::String,
-                                 ?model_card_version: ::Integer
+                                 ?model_card_version: ::Integer,
+                                 ?included_data: ("AllData" | "MetadataOnly")
                                ) -> _DescribeModelCardResponseSuccess
                              | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DescribeModelCardResponseSuccess
 
@@ -4615,7 +4626,8 @@ module Aws
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#describe_model_package-instance_method
       def describe_model_package: (
-                                    model_package_name: ::String
+                                    model_package_name: ::String,
+                                    ?included_data: ("AllData" | "MetadataOnly")
                                   ) -> _DescribeModelPackageResponseSuccess
                                 | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DescribeModelPackageResponseSuccess
 
@@ -5304,7 +5316,7 @@ module Aws
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#get_search_suggestions-instance_method
       def get_search_suggestions: (
-                                    resource: ("TrainingJob" | "Experiment" | "ExperimentTrial" | "ExperimentTrialComponent" | "Endpoint" | "Model" | "ModelPackage" | "ModelPackageGroup" | "Pipeline" | "PipelineExecution" | "FeatureGroup" | "FeatureMetadata" | "Image" | "ImageVersion" | "Project" | "HyperParameterTuningJob" | "ModelCard" | "PipelineVersion"),
+                                    resource: ("TrainingJob" | "Experiment" | "ExperimentTrial" | "ExperimentTrialComponent" | "Endpoint" | "Model" | "ModelPackage" | "ModelPackageGroup" | "Pipeline" | "PipelineExecution" | "FeatureGroup" | "FeatureMetadata" | "Image" | "ImageVersion" | "Project" | "HyperParameterTuningJob" | "ModelCard" | "PipelineVersion" | "Job"),
                                     ?suggestion_query: {
                                       property_name_query: {
                                         property_name_hint: ::String
@@ -7058,7 +7070,7 @@ module Aws
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#search-instance_method
       def search: (
-                    resource: ("TrainingJob" | "Experiment" | "ExperimentTrial" | "ExperimentTrialComponent" | "Endpoint" | "Model" | "ModelPackage" | "ModelPackageGroup" | "Pipeline" | "PipelineExecution" | "FeatureGroup" | "FeatureMetadata" | "Image" | "ImageVersion" | "Project" | "HyperParameterTuningJob" | "ModelCard" | "PipelineVersion"),
+                    resource: ("TrainingJob" | "Experiment" | "ExperimentTrial" | "ExperimentTrialComponent" | "Endpoint" | "Model" | "ModelPackage" | "ModelPackageGroup" | "Pipeline" | "PipelineExecution" | "FeatureGroup" | "FeatureMetadata" | "Image" | "ImageVersion" | "Project" | "HyperParameterTuningJob" | "ModelCard" | "PipelineVersion" | "Job"),
                     ?search_expression: Params::search_expression,
                     ?sort_by: ::String,
                     ?sort_order: ("Ascending" | "Descending"),

data/sig/types.rbs

@@ -37,12 +37,14 @@ module Aws::SageMaker
 
     class AIBenchmarkOutputConfig
       attr_accessor s3_output_location: ::String
+      attr_accessor mlflow_config: Types::AIMlflowConfig
       SENSITIVE: []
     end
 
     class AIBenchmarkOutputResult
       attr_accessor s3_output_location: ::String
       attr_accessor cloud_watch_logs: ::Array[Types::AICloudWatchLogs]
+      attr_accessor mlflow_config: Types::AIMlflowConfig
       SENSITIVE: []
     end
 
@@ -80,6 +82,13 @@ module Aws::SageMaker
       end
     end
 
+    class AIMlflowConfig
+      attr_accessor mlflow_resource_arn: ::String
+      attr_accessor mlflow_experiment_name: ::String
+      attr_accessor mlflow_run_name: ::String
+      SENSITIVE: []
+    end
+
     class AIModelSource
       attr_accessor s3: Types::AIModelSourceS3
       attr_accessor unknown: untyped
@@ -170,12 +179,14 @@ module Aws::SageMaker
     class AIRecommendationOutputConfig
       attr_accessor s3_output_location: ::String
       attr_accessor model_package_group_identifier: ::String
+      attr_accessor mlflow_config: Types::AIMlflowConfig
       SENSITIVE: []
     end
 
     class AIRecommendationOutputResult
       attr_accessor s3_output_location: ::String
       attr_accessor model_package_group_identifier: ::String
+      attr_accessor mlflow_config: Types::AIMlflowConfig
       SENSITIVE: []
     end
 
@@ -4640,6 +4651,7 @@ module Aws::SageMaker
     class DescribeModelCardRequest
       attr_accessor model_card_name: ::String
       attr_accessor model_card_version: ::Integer
+      attr_accessor included_data: ("AllData" | "MetadataOnly")
       SENSITIVE: []
     end
 
@@ -4715,6 +4727,7 @@ module Aws::SageMaker
 
     class DescribeModelPackageInput
       attr_accessor model_package_name: ::String
+      attr_accessor included_data: ("AllData" | "MetadataOnly")
       SENSITIVE: []
     end
 
@@ -6042,7 +6055,7 @@ module Aws::SageMaker
     end
 
     class GetSearchSuggestionsRequest
-      attr_accessor resource: ("TrainingJob" | "Experiment" | "ExperimentTrial" | "ExperimentTrialComponent" | "Endpoint" | "Model" | "ModelPackage" | "ModelPackageGroup" | "Pipeline" | "PipelineExecution" | "FeatureGroup" | "FeatureMetadata" | "Image" | "ImageVersion" | "Project" | "HyperParameterTuningJob" | "ModelCard" | "PipelineVersion")
+      attr_accessor resource: ("TrainingJob" | "Experiment" | "ExperimentTrial" | "ExperimentTrialComponent" | "Endpoint" | "Model" | "ModelPackage" | "ModelPackageGroup" | "Pipeline" | "PipelineExecution" | "FeatureGroup" | "FeatureMetadata" | "Image" | "ImageVersion" | "Project" | "HyperParameterTuningJob" | "ModelCard" | "PipelineVersion" | "Job")
       attr_accessor suggestion_query: Types::SuggestionQuery
       SENSITIVE: []
     end
@@ -6727,6 +6740,24 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class Job
+      attr_accessor job_name: ::String
+      attr_accessor job_arn: ::String
+      attr_accessor role_arn: ::String
+      attr_accessor job_category: ("AgentRFT" | "AgentRFTEvaluation")
+      attr_accessor job_config_schema_version: ::String
+      attr_accessor job_config_document: ::String
+      attr_accessor creation_time: ::Time
+      attr_accessor last_modified_time: ::Time
+      attr_accessor end_time: ::Time
+      attr_accessor job_status: ("InProgress" | "Completed" | "Failed" | "Stopping" | "Stopped" | "Deleting" | "DeleteFailed")
+      attr_accessor secondary_status: ("Starting" | "Downloading" | "Training" | "Uploading" | "Stopping" | "Stopped" | "MaxRuntimeExceeded" | "Interrupted" | "Failed" | "Completed" | "Restarting" | "Pending" | "Evaluating" | "Deleting" | "DeleteFailed")
+      attr_accessor secondary_status_transitions: ::Array[Types::JobSecondaryStatusTransition]
+      attr_accessor failure_reason: ::String
+      attr_accessor tags: ::Array[Types::Tag]
+      SENSITIVE: []
+    end
+
     class JobConfigSchemaVersionSummary
       attr_accessor job_config_schema_version: ::String
       SENSITIVE: []
@@ -10627,11 +10658,12 @@ module Aws::SageMaker
       attr_accessor hyper_parameter_tuning_job: Types::HyperParameterTuningJobSearchEntity
       attr_accessor model_card: Types::ModelCard
       attr_accessor model: Types::ModelDashboardModel
+      attr_accessor job: Types::Job
       SENSITIVE: []
     end
 
     class SearchRequest
-      attr_accessor resource: ("TrainingJob" | "Experiment" | "ExperimentTrial" | "ExperimentTrialComponent" | "Endpoint" | "Model" | "ModelPackage" | "ModelPackageGroup" | "Pipeline" | "PipelineExecution" | "FeatureGroup" | "FeatureMetadata" | "Image" | "ImageVersion" | "Project" | "HyperParameterTuningJob" | "ModelCard" | "PipelineVersion")
+      attr_accessor resource: ("TrainingJob" | "Experiment" | "ExperimentTrial" | "ExperimentTrialComponent" | "Endpoint" | "Model" | "ModelPackage" | "ModelPackageGroup" | "Pipeline" | "PipelineExecution" | "FeatureGroup" | "FeatureMetadata" | "Image" | "ImageVersion" | "Project" | "HyperParameterTuningJob" | "ModelCard" | "PipelineVersion" | "Job")
       attr_accessor search_expression: Types::SearchExpression
       attr_accessor sort_by: ::String
       attr_accessor sort_order: ("Ascending" | "Descending")

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-sagemaker
 version: !ruby/object:Gem::Version
-  version: 1.370.0
+  version: 1.372.0
 platform: ruby
 authors:
 - Amazon Web Services