aws-sdk-sagemakermetrics 1.22.0 → 1.23.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 90741ebb34642ff80a47271add0d8d9a1c651cc03270376e673c437a1399ec69
-  data.tar.gz: 6428da7bd89c37a5e02d82b9a081b89610fd1171cce4a59b7229628b06ec7d26
+  metadata.gz: d2744146186844322a54d402c8d97c535ad46378336b1c0508ecdf27ff200dbb
+  data.tar.gz: ab1231e3c8be2c77138ef4473759bf76a9deb33c61a756112bb7c19370fe0b3c
 SHA512:
-  metadata.gz: bdc56d64b9213eb286d44c4b6cef6faface1967a9b22e1addec93d7e657d73cfeb32a4d606cc1efb32c66e5acdaa71cb1d55f57d3591c4ca2ba4c41304a402bc
-  data.tar.gz: 3e9e55dcb5042723bd8c764914d1677a23facca6be712a50be9c448b2909f91d5cb16fa0a0aa615241a6f7d5a4fbb600e5ff424cefedce0267ec70c019b1d82c
+  metadata.gz: 58606906e0b972188fadec0b372706b59425afc359f3aefa4b302addc4b26901360e4b03da6dd2381f060e0a30add85b641f53e556108452064786e95b37a6a4
+  data.tar.gz: 82d2c4d4f1de52e9dd596ef6ac123b3e36aa0c9eb49552a1ddc0e9936c3584f620b7f048f24fbb3395501fa223985e1b9b8d957bb52fc6b8e55361e1633c5091

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 Unreleased Changes
 ------------------
 
+1.23.0 (2024-09-20)
+------------------
+
+* Feature - This release introduces support for the SageMaker Metrics BatchGetMetrics API.
+
 1.22.0 (2024-09-11)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.22.0
+1.23.0

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

@@ -130,13 +130,15 @@ module Aws::SageMakerMetrics
     #     locations will be searched for credentials:
     #
     #     * `Aws.config[:credentials]`
-    #     * The `:access_key_id`, `:secret_access_key`, and `:session_token` options.
-    #     * ENV['AWS_ACCESS_KEY_ID'], ENV['AWS_SECRET_ACCESS_KEY']
+    #     * The `:access_key_id`, `:secret_access_key`, `:session_token`, and
+    #       `:account_id` options.
+    #     * ENV['AWS_ACCESS_KEY_ID'], ENV['AWS_SECRET_ACCESS_KEY'],
+    #       ENV['AWS_SESSION_TOKEN'], and ENV['AWS_ACCOUNT_ID']
     #     * `~/.aws/credentials`
     #     * `~/.aws/config`
     #     * EC2/ECS IMDS instance profile - When used by default, the timeouts
     #       are very aggressive. Construct and pass an instance of
-    #       `Aws::InstanceProfileCredentails` or `Aws::ECSCredentials` to
+    #       `Aws::InstanceProfileCredentials` or `Aws::ECSCredentials` to
     #       enable retries and extended timeouts. Instance profile credential
     #       fetching can be disabled by setting ENV['AWS_EC2_METADATA_DISABLED']
     #       to true.
@@ -155,6 +157,8 @@ module Aws::SageMakerMetrics
     #
     #   @option options [String] :access_key_id
     #
+    #   @option options [String] :account_id
+    #
     #   @option options [Boolean] :active_endpoint_cache (false)
     #     When set to `true`, a thread polling for endpoints will be running in
     #     the background every 60 secs (default). Defaults to `false`.
@@ -369,7 +373,9 @@ module Aws::SageMakerMetrics
     #     sending the request.
     #
     #   @option options [Aws::SageMakerMetrics::EndpointProvider] :endpoint_provider
-    #     The endpoint provider used to resolve endpoints. Any object that responds to `#resolve_endpoint(parameters)` where `parameters` is a Struct similar to `Aws::SageMakerMetrics::EndpointParameters`
+    #     The endpoint provider used to resolve endpoints. Any object that responds to
+    #     `#resolve_endpoint(parameters)` where `parameters` is a Struct similar to
+    #     `Aws::SageMakerMetrics::EndpointParameters`.
     #
     #   @option options [Float] :http_continue_timeout (1)
     #     The number of seconds to wait for a 100-continue response before sending the
@@ -443,12 +449,56 @@ module Aws::SageMakerMetrics
 
     # @!group API Operations
 
+    # Used to retrieve training metrics from SageMaker.
+    #
+    # @option params [required, Array<Types::MetricQuery>] :metric_queries
+    #   Queries made to retrieve training metrics from SageMaker.
+    #
+    # @return [Types::BatchGetMetricsResponse] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
+    #
+    #   * {Types::BatchGetMetricsResponse#metric_query_results #metric_query_results} => Array&lt;Types::MetricQueryResult&gt;
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.batch_get_metrics({
+    #     metric_queries: [ # required
+    #       {
+    #         metric_name: "MetricName", # required
+    #         resource_arn: "SageMakerResourceArn", # required
+    #         metric_stat: "Min", # required, accepts Min, Max, Avg, Count, StdDev, Last
+    #         period: "OneMinute", # required, accepts OneMinute, FiveMinute, OneHour, IterationNumber
+    #         x_axis_type: "IterationNumber", # required, accepts IterationNumber, Timestamp
+    #         start: 1,
+    #         end: 1,
+    #       },
+    #     ],
+    #   })
+    #
+    # @example Response structure
+    #
+    #   resp.metric_query_results #=> Array
+    #   resp.metric_query_results[0].status #=> String, one of "Complete", "Truncated", "InternalError", "ValidationError"
+    #   resp.metric_query_results[0].message #=> String
+    #   resp.metric_query_results[0].x_axis_values #=> Array
+    #   resp.metric_query_results[0].x_axis_values[0] #=> Integer
+    #   resp.metric_query_results[0].metric_values #=> Array
+    #   resp.metric_query_results[0].metric_values[0] #=> Float
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-metrics-2022-09-30/BatchGetMetrics AWS API Documentation
+    #
+    # @overload batch_get_metrics(params = {})
+    # @param [Hash] params ({})
+    def batch_get_metrics(params = {}, options = {})
+      req = build_request(:batch_get_metrics, params)
+      req.send_request(options)
+    end
+
     # Used to ingest training metrics into SageMaker. These metrics can be
-    # visualized in SageMaker Studio and retrieved with the `GetMetrics`
-    # API.
+    # visualized in SageMaker Studio.
     #
     # @option params [required, String] :trial_component_name
-    #   The name of the Trial Component to associate with the metrics.
+    #   The name of the Trial Component to associate with the metrics. The
+    #   Trial Component name must be entirely lowercase.
     #
     # @option params [required, Array<Types::RawMetricData>] :metric_data
     #   A list of raw metric values to put.
@@ -504,7 +554,7 @@ module Aws::SageMakerMetrics
         tracer: tracer
       )
       context[:gem_name] = 'aws-sdk-sagemakermetrics'
-      context[:gem_version] = '1.22.0'
+      context[:gem_version] = '1.23.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -13,6 +13,8 @@ module Aws::SageMakerMetrics
 
     include Seahorse::Model
 
+    BatchGetMetricsRequest = Shapes::StructureShape.new(name: 'BatchGetMetricsRequest')
+    BatchGetMetricsResponse = Shapes::StructureShape.new(name: 'BatchGetMetricsResponse')
     BatchPutMetricsError = Shapes::StructureShape.new(name: 'BatchPutMetricsError')
     BatchPutMetricsErrorList = Shapes::ListShape.new(name: 'BatchPutMetricsErrorList')
     BatchPutMetricsRequest = Shapes::StructureShape.new(name: 'BatchPutMetricsRequest')
@@ -20,12 +22,31 @@ module Aws::SageMakerMetrics
     Double = Shapes::FloatShape.new(name: 'Double')
     ExperimentEntityName = Shapes::StringShape.new(name: 'ExperimentEntityName')
     Integer = Shapes::IntegerShape.new(name: 'Integer')
+    Long = Shapes::IntegerShape.new(name: 'Long')
+    Message = Shapes::StringShape.new(name: 'Message')
     MetricName = Shapes::StringShape.new(name: 'MetricName')
+    MetricQuery = Shapes::StructureShape.new(name: 'MetricQuery')
+    MetricQueryList = Shapes::ListShape.new(name: 'MetricQueryList')
+    MetricQueryResult = Shapes::StructureShape.new(name: 'MetricQueryResult')
+    MetricQueryResultList = Shapes::ListShape.new(name: 'MetricQueryResultList')
+    MetricQueryResultStatus = Shapes::StringShape.new(name: 'MetricQueryResultStatus')
+    MetricStatistic = Shapes::StringShape.new(name: 'MetricStatistic')
+    MetricValues = Shapes::ListShape.new(name: 'MetricValues')
+    Period = Shapes::StringShape.new(name: 'Period')
     PutMetricsErrorCode = Shapes::StringShape.new(name: 'PutMetricsErrorCode')
     RawMetricData = Shapes::StructureShape.new(name: 'RawMetricData')
     RawMetricDataList = Shapes::ListShape.new(name: 'RawMetricDataList')
+    SageMakerResourceArn = Shapes::StringShape.new(name: 'SageMakerResourceArn')
     Step = Shapes::IntegerShape.new(name: 'Step')
     Timestamp = Shapes::TimestampShape.new(name: 'Timestamp')
+    XAxisType = Shapes::StringShape.new(name: 'XAxisType')
+    XAxisValues = Shapes::ListShape.new(name: 'XAxisValues')
+
+    BatchGetMetricsRequest.add_member(:metric_queries, Shapes::ShapeRef.new(shape: MetricQueryList, required: true, location_name: "MetricQueries"))
+    BatchGetMetricsRequest.struct_class = Types::BatchGetMetricsRequest
+
+    BatchGetMetricsResponse.add_member(:metric_query_results, Shapes::ShapeRef.new(shape: MetricQueryResultList, location_name: "MetricQueryResults"))
+    BatchGetMetricsResponse.struct_class = Types::BatchGetMetricsResponse
 
     BatchPutMetricsError.add_member(:code, Shapes::ShapeRef.new(shape: PutMetricsErrorCode, location_name: "Code"))
     BatchPutMetricsError.add_member(:metric_index, Shapes::ShapeRef.new(shape: Integer, location_name: "MetricIndex"))
@@ -40,6 +61,27 @@ module Aws::SageMakerMetrics
     BatchPutMetricsResponse.add_member(:errors, Shapes::ShapeRef.new(shape: BatchPutMetricsErrorList, location_name: "Errors"))
     BatchPutMetricsResponse.struct_class = Types::BatchPutMetricsResponse
 
+    MetricQuery.add_member(:metric_name, Shapes::ShapeRef.new(shape: MetricName, required: true, location_name: "MetricName"))
+    MetricQuery.add_member(:resource_arn, Shapes::ShapeRef.new(shape: SageMakerResourceArn, required: true, location_name: "ResourceArn"))
+    MetricQuery.add_member(:metric_stat, Shapes::ShapeRef.new(shape: MetricStatistic, required: true, location_name: "MetricStat"))
+    MetricQuery.add_member(:period, Shapes::ShapeRef.new(shape: Period, required: true, location_name: "Period"))
+    MetricQuery.add_member(:x_axis_type, Shapes::ShapeRef.new(shape: XAxisType, required: true, location_name: "XAxisType"))
+    MetricQuery.add_member(:start, Shapes::ShapeRef.new(shape: Long, location_name: "Start", metadata: {"box"=>true}))
+    MetricQuery.add_member(:end, Shapes::ShapeRef.new(shape: Long, location_name: "End", metadata: {"box"=>true}))
+    MetricQuery.struct_class = Types::MetricQuery
+
+    MetricQueryList.member = Shapes::ShapeRef.new(shape: MetricQuery)
+
+    MetricQueryResult.add_member(:status, Shapes::ShapeRef.new(shape: MetricQueryResultStatus, required: true, location_name: "Status"))
+    MetricQueryResult.add_member(:message, Shapes::ShapeRef.new(shape: Message, location_name: "Message"))
+    MetricQueryResult.add_member(:x_axis_values, Shapes::ShapeRef.new(shape: XAxisValues, required: true, location_name: "XAxisValues"))
+    MetricQueryResult.add_member(:metric_values, Shapes::ShapeRef.new(shape: MetricValues, required: true, location_name: "MetricValues"))
+    MetricQueryResult.struct_class = Types::MetricQueryResult
+
+    MetricQueryResultList.member = Shapes::ShapeRef.new(shape: MetricQueryResult)
+
+    MetricValues.member = Shapes::ShapeRef.new(shape: Double)
+
     RawMetricData.add_member(:metric_name, Shapes::ShapeRef.new(shape: MetricName, required: true, location_name: "MetricName"))
     RawMetricData.add_member(:timestamp, Shapes::ShapeRef.new(shape: Timestamp, required: true, location_name: "Timestamp"))
     RawMetricData.add_member(:step, Shapes::ShapeRef.new(shape: Step, location_name: "Step"))
@@ -48,6 +90,8 @@ module Aws::SageMakerMetrics
 
     RawMetricDataList.member = Shapes::ShapeRef.new(shape: RawMetricData)
 
+    XAxisValues.member = Shapes::ShapeRef.new(shape: Long)
+
 
     # @api private
     API = Seahorse::Model::Api.new.tap do |api|
@@ -68,6 +112,14 @@ module Aws::SageMakerMetrics
         "uid" => "sagemaker-metrics-2022-09-30",
       }
 
+      api.add_operation(:batch_get_metrics, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "BatchGetMetrics"
+        o.http_method = "POST"
+        o.http_request_uri = "/BatchGetMetrics"
+        o.input = Shapes::ShapeRef.new(shape: BatchGetMetricsRequest)
+        o.output = Shapes::ShapeRef.new(shape: BatchGetMetricsResponse)
+      end)
+
       api.add_operation(:batch_put_metrics, Seahorse::Model::Operation.new.tap do |o|
         o.name = "BatchPutMetrics"
         o.http_method = "PUT"

data/lib/aws-sdk-sagemakermetrics/endpoint_provider.rb

@@ -32,7 +32,7 @@ module Aws::SageMakerMetrics
             raise ArgumentError, "FIPS and DualStack are enabled, but this partition does not support one or both"
           end
           if Aws::Endpoints::Matchers.boolean_equals?(use_fips, true)
-            if Aws::Endpoints::Matchers.boolean_equals?(true, Aws::Endpoints::Matchers.attr(partition_result, "supportsFIPS"))
+            if Aws::Endpoints::Matchers.boolean_equals?(Aws::Endpoints::Matchers.attr(partition_result, "supportsFIPS"), true)
               return Aws::Endpoints::Endpoint.new(url: "https://metrics.sagemaker-fips.#{region}.#{partition_result['dnsSuffix']}", headers: {}, properties: {})
             end
             raise ArgumentError, "FIPS is enabled but this partition does not support FIPS"

data/lib/aws-sdk-sagemakermetrics/endpoints.rb

@@ -12,6 +12,17 @@ module Aws::SageMakerMetrics
   # @api private
   module Endpoints
 
+    class BatchGetMetrics
+      def self.build(context)
+        Aws::SageMakerMetrics::EndpointParameters.new(
+          region: context.config.region,
+          use_dual_stack: context.config.use_dualstack_endpoint,
+          use_fips: context.config.use_fips_endpoint,
+          endpoint: context.config.regional_endpoint ? nil : context.config.endpoint.to_s,
+        )
+      end
+    end
+
     class BatchPutMetrics
       def self.build(context)
         Aws::SageMakerMetrics::EndpointParameters.new(

data/lib/aws-sdk-sagemakermetrics/plugins/endpoints.rb

@@ -15,11 +15,11 @@ module Aws::SageMakerMetrics
         :endpoint_provider,
         doc_type: 'Aws::SageMakerMetrics::EndpointProvider',
         rbs_type: 'untyped',
-        docstring: 'The endpoint provider used to resolve endpoints. Any '\
-                   'object that responds to `#resolve_endpoint(parameters)` '\
-                   'where `parameters` is a Struct similar to '\
-                   '`Aws::SageMakerMetrics::EndpointParameters`'
-      ) do |cfg|
+        docstring: <<~DOCS) do |_cfg|
+The endpoint provider used to resolve endpoints. Any object that responds to
+`#resolve_endpoint(parameters)` where `parameters` is a Struct similar to
+`Aws::SageMakerMetrics::EndpointParameters`.
+        DOCS
         Aws::SageMakerMetrics::EndpointProvider.new
       end
 
@@ -51,6 +51,9 @@ module Aws::SageMakerMetrics
           if context[:auth_scheme] && context[:auth_scheme]['name'] == 'sigv4a'
             metrics << 'SIGV4A_SIGNING'
           end
+          if context.config.credentials&.credentials&.account_id
+            metrics << 'RESOLVED_ACCOUNT_ID'
+          end
           Aws::Plugins::UserAgent.metric(*metrics, &block)
         end
 
@@ -67,6 +70,8 @@ module Aws::SageMakerMetrics
 
         def parameters_for_operation(context)
           case context.operation_name
+          when :batch_get_metrics
+            Aws::SageMakerMetrics::Endpoints::BatchGetMetrics.build(context)
           when :batch_put_metrics
             Aws::SageMakerMetrics::Endpoints::BatchPutMetrics.build(context)
           end

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

@@ -10,6 +10,30 @@
 module Aws::SageMakerMetrics
   module Types
 
+    # @!attribute [rw] metric_queries
+    #   Queries made to retrieve training metrics from SageMaker.
+    #   @return [Array<Types::MetricQuery>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-metrics-2022-09-30/BatchGetMetricsRequest AWS API Documentation
+    #
+    class BatchGetMetricsRequest < Struct.new(
+      :metric_queries)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] metric_query_results
+    #   The results of a query to retrieve training metrics from SageMaker.
+    #   @return [Array<Types::MetricQueryResult>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-metrics-2022-09-30/BatchGetMetricsResponse AWS API Documentation
+    #
+    class BatchGetMetricsResponse < Struct.new(
+      :metric_query_results)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # An error that occured when putting the metric data.
     #
     # @!attribute [rw] code
@@ -41,7 +65,8 @@ module Aws::SageMakerMetrics
     end
 
     # @!attribute [rw] trial_component_name
-    #   The name of the Trial Component to associate with the metrics.
+    #   The name of the Trial Component to associate with the metrics. The
+    #   Trial Component name must be entirely lowercase.
     #   @return [String]
     #
     # @!attribute [rw] metric_data
@@ -69,6 +94,79 @@ module Aws::SageMakerMetrics
       include Aws::Structure
     end
 
+    # Specifies a query to retrieve training metrics from SageMaker.
+    #
+    # @!attribute [rw] metric_name
+    #   The name of the metric to retrieve.
+    #   @return [String]
+    #
+    # @!attribute [rw] resource_arn
+    #   The ARN of the SageMaker resource to retrieve metrics for.
+    #   @return [String]
+    #
+    # @!attribute [rw] metric_stat
+    #   The metrics stat type of metrics to retrieve.
+    #   @return [String]
+    #
+    # @!attribute [rw] period
+    #   The time period of metrics to retrieve.
+    #   @return [String]
+    #
+    # @!attribute [rw] x_axis_type
+    #   The x-axis type of metrics to retrieve.
+    #   @return [String]
+    #
+    # @!attribute [rw] start
+    #   The start time of metrics to retrieve.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] end
+    #   The end time of metrics to retrieve.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-metrics-2022-09-30/MetricQuery AWS API Documentation
+    #
+    class MetricQuery < Struct.new(
+      :metric_name,
+      :resource_arn,
+      :metric_stat,
+      :period,
+      :x_axis_type,
+      :start,
+      :end)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The result of a query to retrieve training metrics from SageMaker.
+    #
+    # @!attribute [rw] status
+    #   The status of the metric query.
+    #   @return [String]
+    #
+    # @!attribute [rw] message
+    #   A message describing the status of the metric query.
+    #   @return [String]
+    #
+    # @!attribute [rw] x_axis_values
+    #   The values for the x-axis of the metrics.
+    #   @return [Array<Integer>]
+    #
+    # @!attribute [rw] metric_values
+    #   The metric values retrieved by the query.
+    #   @return [Array<Float>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-metrics-2022-09-30/MetricQueryResult AWS API Documentation
+    #
+    class MetricQueryResult < Struct.new(
+      :status,
+      :message,
+      :x_axis_values,
+      :metric_values)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The raw metric data to associate with the resource.
     #
     # @!attribute [rw] metric_name

data/lib/aws-sdk-sagemakermetrics.rb

@@ -32,7 +32,7 @@ require_relative 'aws-sdk-sagemakermetrics/customizations'
 # structure.
 #
 #     sage_maker_metrics = Aws::SageMakerMetrics::Client.new
-#     resp = sage_maker_metrics.batch_put_metrics(params)
+#     resp = sage_maker_metrics.batch_get_metrics(params)
 #
 # See {Client} for more information.
 #
@@ -52,6 +52,6 @@ require_relative 'aws-sdk-sagemakermetrics/customizations'
 # @!group service
 module Aws::SageMakerMetrics
 
-  GEM_VERSION = '1.22.0'
+  GEM_VERSION = '1.23.0'
 
 end

data/sig/client.rbs

@@ -15,6 +15,7 @@ module Aws
                       ?credentials: untyped,
                       ?region: String,
                       ?access_key_id: String,
+                      ?account_id: String,
                       ?active_endpoint_cache: bool,
                       ?adaptive_retry_wait_to_fill: bool,
                       ?client_side_monitoring: bool,
@@ -74,6 +75,26 @@ module Aws
                   | (?Hash[Symbol, untyped]) -> instance
 
 
+      interface _BatchGetMetricsResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::BatchGetMetricsResponse]
+        def metric_query_results: () -> ::Array[Types::MetricQueryResult]
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMakerMetrics/Client.html#batch_get_metrics-instance_method
+      def batch_get_metrics: (
+                               metric_queries: Array[
+                                 {
+                                   metric_name: ::String,
+                                   resource_arn: ::String,
+                                   metric_stat: ("Min" | "Max" | "Avg" | "Count" | "StdDev" | "Last"),
+                                   period: ("OneMinute" | "FiveMinute" | "OneHour" | "IterationNumber"),
+                                   x_axis_type: ("IterationNumber" | "Timestamp"),
+                                   start: ::Integer?,
+                                   end: ::Integer?
+                                 },
+                               ]
+                             ) -> _BatchGetMetricsResponseSuccess
+                           | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _BatchGetMetricsResponseSuccess
+
       interface _BatchPutMetricsResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::BatchPutMetricsResponse]
         def errors: () -> ::Array[Types::BatchPutMetricsError]

data/sig/resource.rbs

@@ -15,6 +15,7 @@ module Aws
                         ?credentials: untyped,
                         ?region: String,
                         ?access_key_id: String,
+                        ?account_id: String,
                         ?active_endpoint_cache: bool,
                         ?adaptive_retry_wait_to_fill: bool,
                         ?client_side_monitoring: bool,

data/sig/types.rbs

@@ -8,6 +8,16 @@
 module Aws::SageMakerMetrics
   module Types
 
+    class BatchGetMetricsRequest
+      attr_accessor metric_queries: ::Array[Types::MetricQuery]
+      SENSITIVE: []
+    end
+
+    class BatchGetMetricsResponse
+      attr_accessor metric_query_results: ::Array[Types::MetricQueryResult]
+      SENSITIVE: []
+    end
+
     class BatchPutMetricsError
       attr_accessor code: ("METRIC_LIMIT_EXCEEDED" | "INTERNAL_ERROR" | "VALIDATION_ERROR" | "CONFLICT_ERROR")
       attr_accessor metric_index: ::Integer
@@ -25,6 +35,25 @@ module Aws::SageMakerMetrics
       SENSITIVE: []
     end
 
+    class MetricQuery
+      attr_accessor metric_name: ::String
+      attr_accessor resource_arn: ::String
+      attr_accessor metric_stat: ("Min" | "Max" | "Avg" | "Count" | "StdDev" | "Last")
+      attr_accessor period: ("OneMinute" | "FiveMinute" | "OneHour" | "IterationNumber")
+      attr_accessor x_axis_type: ("IterationNumber" | "Timestamp")
+      attr_accessor start: ::Integer
+      attr_accessor end: ::Integer
+      SENSITIVE: []
+    end
+
+    class MetricQueryResult
+      attr_accessor status: ("Complete" | "Truncated" | "InternalError" | "ValidationError")
+      attr_accessor message: ::String
+      attr_accessor x_axis_values: ::Array[::Integer]
+      attr_accessor metric_values: ::Array[::Float]
+      SENSITIVE: []
+    end
+
     class RawMetricData
       attr_accessor metric_name: ::String
       attr_accessor timestamp: ::Time

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-sagemakermetrics
 version: !ruby/object:Gem::Version
-  version: 1.22.0
+  version: 1.23.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-11 00:00:00.000000000 Z
+date: 2024-09-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core
@@ -19,7 +19,7 @@ dependencies:
         version: '3'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 3.205.0
+        version: 3.207.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -29,7 +29,7 @@ dependencies:
         version: '3'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 3.205.0
+        version: 3.207.0
 - !ruby/object:Gem::Dependency
   name: aws-sigv4
   requirement: !ruby/object:Gem::Requirement