aws-sdk-sagemaker 1.371.0 → 1.372.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bbb9f56b330c8e6b214a2f5db6f1f423a322de62d3a1f6c89b5fd9a3c5c556ce
-  data.tar.gz: 294be139ced80cc72549555b70163f39751f0d19f61d093e4eb495a782fd5fda
+  metadata.gz: e6399d5cfc5d2d703e98f466dd3d1899884b3f50cd13ec05b5789b0071b1413f
+  data.tar.gz: 48b38958f579fb434e67ec60888f91effcd6df6d7149da6c02f873bcc5ef4470
 SHA512:
-  metadata.gz: '0588d9267f26569baf148d8ea4f8680559f09c54f1a3fbb68cb9b7ca50d7e61c03ba0835370097f73fd8b80adae847cab6801981410465e3d57bef32fc2807c9'
-  data.tar.gz: 7618d5e45560fe2e4bd951a31ce7a7f1e76165824096b0a373ba0ff5a7cd33cc0378aa55d014808442fae1af264c04bfc8c9ca5ee14826fdc7f1289c883e7f9e
+  metadata.gz: 2f04335adbc52e3257790d17db0b45ae4647872a5c49481ab650d1b83eb6b99409b30e99e750d9b5ff53ca1731f61768d82edf67d8ef490790ac179ab54a4b5a
+  data.tar.gz: 70f6d4de733005162ad7083a3af6e89b236c9b42d4f8f6e7299e3503d37e96cede2dc65204c2f201261b2c75ac066181fe42944a33267bb0524055fd3d7a394e

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 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)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.371.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
@@ -34363,7 +34379,7 @@ module Aws::SageMaker
         tracer: tracer
       )
       context[:gem_name] = 'aws-sdk-sagemaker'
-      context[:gem_version] = '1.371.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')
@@ -3040,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"))
@@ -3070,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)
@@ -3149,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"))

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

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.371.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: {

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
 

metadata

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