aws-sdk-sagemaker 1.353.0 → 1.354.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a0be4780702f08a2e76c697e94cb3a7c73a60a175a45bff25c0e20ce38af111e
-  data.tar.gz: 5030027d22c467dc811023c55953810b10a89e963682eb1e55ea80f41a6139d1
+  metadata.gz: 3b31ce4da46a108aa4c8c41b2b5f635fa04c56569eff02010ea00a12db0f7421
+  data.tar.gz: c1066466d28b74579d7ea03e819a9c9dea76133624f398449549149730cf2f99
 SHA512:
-  metadata.gz: 17921bcd075e05991c915377c0d50046771d50025e53b0d0121db748453f44f50d30e3ce078a174657e6e38b4eef1378a48f51e27083913c5da563b33fa3c42f
-  data.tar.gz: 26f8522eabb01a3558426efbb1b7f055b7511c23b84cd928cbf23ab5c74584a8f12dd05a53ee1af42a9c1b6aa3fe6d40bdac359dc4c88234ba3f417991519d55
+  metadata.gz: ce840f7b09da1f439afc881dd00dd461468a73b098fcfb181445525beda6b54589eb286095be29de87d93c7d00d837cb601498efbd1f33c6f25f07fec5af0489
+  data.tar.gz: 2c93b24db44e923005c65f309c427cefd94ed281d2d06b131b434521b4554c5a37d1877cb14f5275734a1d951f29116941e5a749303a28e0faeb9fa00f0c552c

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 Unreleased Changes
 ------------------
 
+1.354.0 (2026-03-11)
+------------------
+
+* Feature - SageMaker training plans allow you to extend your existing training plans to avoid workload interruptions without workload reconfiguration. When a training plan is approaching expiration, you can extend it directly through the SageMaker AI console or programmatically using the API or AWS CLI.
+
 1.353.0 (2026-03-05)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.353.0
+1.354.0

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

@@ -19148,6 +19148,60 @@ module Aws::SageMaker
       req.send_request(options)
     end
 
+    # Retrieves the extension history for a specified training plan. The
+    # response includes details about each extension, such as the offering
+    # ID, start and end dates, status, payment status, and cost information.
+    #
+    # @option params [required, String] :training_plan_arn
+    #   The Amazon Resource Name (ARN); of the training plan to retrieve
+    #   extension history for.
+    #
+    # @option params [String] :next_token
+    #   A token to continue pagination if more results are available.
+    #
+    # @option params [Integer] :max_results
+    #   The maximum number of extensions to return in the response.
+    #
+    # @return [Types::DescribeTrainingPlanExtensionHistoryResponse] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
+    #
+    #   * {Types::DescribeTrainingPlanExtensionHistoryResponse#training_plan_extensions #training_plan_extensions} => Array<Types::TrainingPlanExtension>
+    #   * {Types::DescribeTrainingPlanExtensionHistoryResponse#next_token #next_token} => String
+    #
+    # The returned {Seahorse::Client::Response response} is a pageable response and is Enumerable. For details on usage see {Aws::PageableResponse PageableResponse}.
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.describe_training_plan_extension_history({
+    #     training_plan_arn: "TrainingPlanArn", # required
+    #     next_token: "NextToken",
+    #     max_results: 1,
+    #   })
+    #
+    # @example Response structure
+    #
+    #   resp.training_plan_extensions #=> Array
+    #   resp.training_plan_extensions[0].training_plan_extension_offering_id #=> String
+    #   resp.training_plan_extensions[0].extended_at #=> Time
+    #   resp.training_plan_extensions[0].start_date #=> Time
+    #   resp.training_plan_extensions[0].end_date #=> Time
+    #   resp.training_plan_extensions[0].status #=> String
+    #   resp.training_plan_extensions[0].payment_status #=> String
+    #   resp.training_plan_extensions[0].availability_zone #=> String
+    #   resp.training_plan_extensions[0].availability_zone_id #=> String
+    #   resp.training_plan_extensions[0].duration_hours #=> Integer
+    #   resp.training_plan_extensions[0].upfront_fee #=> String
+    #   resp.training_plan_extensions[0].currency_code #=> String
+    #   resp.next_token #=> String
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeTrainingPlanExtensionHistory AWS API Documentation
+    #
+    # @overload describe_training_plan_extension_history(params = {})
+    # @param [Hash] params ({})
+    def describe_training_plan_extension_history(params = {}, options = {})
+      req = build_request(:describe_training_plan_extension_history, params)
+      req.send_request(options)
+    end
+
     # Returns information about a transform job.
     #
     # @option params [required, String] :transform_job_name
@@ -19809,6 +19863,57 @@ module Aws::SageMaker
       req.send_request(options)
     end
 
+    # Extends an existing training plan by purchasing an extension offering.
+    # This allows you to add additional compute capacity time to your
+    # training plan without creating a new plan or reconfiguring your
+    # workloads.
+    #
+    # To find available extension offerings, use the `
+    # SearchTrainingPlanOfferings ` API with the `TrainingPlanArn`
+    # parameter.
+    #
+    # To view the history of extensions for a training plan, use the `
+    # DescribeTrainingPlanExtensionHistory ` API.
+    #
+    # @option params [required, String] :training_plan_extension_offering_id
+    #   The unique identifier of the extension offering to purchase. You can
+    #   retrieve this ID from the `TrainingPlanExtensionOfferings` in the
+    #   response of the `SearchTrainingPlanOfferings` API.
+    #
+    # @return [Types::ExtendTrainingPlanResponse] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
+    #
+    #   * {Types::ExtendTrainingPlanResponse#training_plan_extensions #training_plan_extensions} => Array<Types::TrainingPlanExtension>
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.extend_training_plan({
+    #     training_plan_extension_offering_id: "TrainingPlanExtensionOfferingId", # required
+    #   })
+    #
+    # @example Response structure
+    #
+    #   resp.training_plan_extensions #=> Array
+    #   resp.training_plan_extensions[0].training_plan_extension_offering_id #=> String
+    #   resp.training_plan_extensions[0].extended_at #=> Time
+    #   resp.training_plan_extensions[0].start_date #=> Time
+    #   resp.training_plan_extensions[0].end_date #=> Time
+    #   resp.training_plan_extensions[0].status #=> String
+    #   resp.training_plan_extensions[0].payment_status #=> String
+    #   resp.training_plan_extensions[0].availability_zone #=> String
+    #   resp.training_plan_extensions[0].availability_zone_id #=> String
+    #   resp.training_plan_extensions[0].duration_hours #=> Integer
+    #   resp.training_plan_extensions[0].upfront_fee #=> String
+    #   resp.training_plan_extensions[0].currency_code #=> String
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ExtendTrainingPlan AWS API Documentation
+    #
+    # @overload extend_training_plan(params = {})
+    # @param [Hash] params ({})
+    def extend_training_plan(params = {}, options = {})
+      req = build_request(:extend_training_plan, params)
+      req.send_request(options)
+    end
+
     # Describes a fleet.
     #
     # @option params [required, String] :device_fleet_name
@@ -27320,9 +27425,15 @@ module Aws::SageMaker
     #     provide compute resources to SageMaker endpoints for model
     #     deployment.
     #
+    # @option params [String] :training_plan_arn
+    #   The Amazon Resource Name (ARN); of an existing training plan to search
+    #   for extension offerings. When specified, the API returns extension
+    #   offerings that can be used to extend the specified training plan.
+    #
     # @return [Types::SearchTrainingPlanOfferingsResponse] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
     #
     #   * {Types::SearchTrainingPlanOfferingsResponse#training_plan_offerings #training_plan_offerings} => Array<Types::TrainingPlanOffering>
+    #   * {Types::SearchTrainingPlanOfferingsResponse#training_plan_extension_offerings #training_plan_extension_offerings} => Array<Types::TrainingPlanExtensionOffering>
     #
     # @example Request syntax with placeholder values
     #
@@ -27335,6 +27446,7 @@ module Aws::SageMaker
     #     end_time_before: Time.now,
     #     duration_hours: 1,
     #     target_resources: ["training-job"], # accepts training-job, hyperpod-cluster, endpoint
+    #     training_plan_arn: "String",
     #   })
     #
     # @example Response structure
@@ -27360,6 +27472,16 @@ module Aws::SageMaker
     #   resp.training_plan_offerings[0].reserved_capacity_offerings[0].duration_minutes #=> Integer
     #   resp.training_plan_offerings[0].reserved_capacity_offerings[0].start_time #=> Time
     #   resp.training_plan_offerings[0].reserved_capacity_offerings[0].end_time #=> Time
+    #   resp.training_plan_offerings[0].reserved_capacity_offerings[0].extension_start_time #=> Time
+    #   resp.training_plan_offerings[0].reserved_capacity_offerings[0].extension_end_time #=> Time
+    #   resp.training_plan_extension_offerings #=> Array
+    #   resp.training_plan_extension_offerings[0].training_plan_extension_offering_id #=> String
+    #   resp.training_plan_extension_offerings[0].availability_zone #=> String
+    #   resp.training_plan_extension_offerings[0].start_date #=> Time
+    #   resp.training_plan_extension_offerings[0].end_date #=> Time
+    #   resp.training_plan_extension_offerings[0].duration_hours #=> Integer
+    #   resp.training_plan_extension_offerings[0].upfront_fee #=> String
+    #   resp.training_plan_extension_offerings[0].currency_code #=> String
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/SearchTrainingPlanOfferings AWS API Documentation
     #
@@ -32373,7 +32495,7 @@ module Aws::SageMaker
         tracer: tracer
       )
       context[:gem_name] = 'aws-sdk-sagemaker'
-      context[:gem_version] = '1.353.0'
+      context[:gem_version] = '1.354.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -194,6 +194,7 @@ module Aws::SageMaker
     Autotune = Shapes::StructureShape.new(name: 'Autotune')
     AutotuneMode = Shapes::StringShape.new(name: 'AutotuneMode')
     AvailabilityZone = Shapes::StringShape.new(name: 'AvailabilityZone')
+    AvailabilityZoneId = Shapes::StringShape.new(name: 'AvailabilityZoneId')
     AvailableInstanceCount = Shapes::IntegerShape.new(name: 'AvailableInstanceCount')
     AvailableSpareInstanceCount = Shapes::IntegerShape.new(name: 'AvailableSpareInstanceCount')
     AvailableUpgrade = Shapes::StructureShape.new(name: 'AvailableUpgrade')
@@ -913,6 +914,8 @@ module Aws::SageMaker
     DescribeSubscribedWorkteamResponse = Shapes::StructureShape.new(name: 'DescribeSubscribedWorkteamResponse')
     DescribeTrainingJobRequest = Shapes::StructureShape.new(name: 'DescribeTrainingJobRequest')
     DescribeTrainingJobResponse = Shapes::StructureShape.new(name: 'DescribeTrainingJobResponse')
+    DescribeTrainingPlanExtensionHistoryRequest = Shapes::StructureShape.new(name: 'DescribeTrainingPlanExtensionHistoryRequest')
+    DescribeTrainingPlanExtensionHistoryResponse = Shapes::StructureShape.new(name: 'DescribeTrainingPlanExtensionHistoryResponse')
     DescribeTrainingPlanRequest = Shapes::StructureShape.new(name: 'DescribeTrainingPlanRequest')
     DescribeTrainingPlanResponse = Shapes::StructureShape.new(name: 'DescribeTrainingPlanResponse')
     DescribeTransformJobRequest = Shapes::StructureShape.new(name: 'DescribeTransformJobRequest')
@@ -1097,6 +1100,8 @@ module Aws::SageMaker
     Explainability = Shapes::StructureShape.new(name: 'Explainability')
     ExplainabilityLocation = Shapes::StringShape.new(name: 'ExplainabilityLocation')
     ExplainerConfig = Shapes::StructureShape.new(name: 'ExplainerConfig')
+    ExtendTrainingPlanRequest = Shapes::StructureShape.new(name: 'ExtendTrainingPlanRequest')
+    ExtendTrainingPlanResponse = Shapes::StructureShape.new(name: 'ExtendTrainingPlanResponse')
     FSxLustreConfig = Shapes::StructureShape.new(name: 'FSxLustreConfig')
     FSxLustreFileSystem = Shapes::StructureShape.new(name: 'FSxLustreFileSystem')
     FSxLustreFileSystemConfig = Shapes::StructureShape.new(name: 'FSxLustreFileSystemConfig')
@@ -2579,6 +2584,12 @@ module Aws::SageMaker
     TrainingPlanDurationHours = Shapes::IntegerShape.new(name: 'TrainingPlanDurationHours')
     TrainingPlanDurationHoursInput = Shapes::IntegerShape.new(name: 'TrainingPlanDurationHoursInput')
     TrainingPlanDurationMinutes = Shapes::IntegerShape.new(name: 'TrainingPlanDurationMinutes')
+    TrainingPlanExtension = Shapes::StructureShape.new(name: 'TrainingPlanExtension')
+    TrainingPlanExtensionDurationHours = Shapes::IntegerShape.new(name: 'TrainingPlanExtensionDurationHours')
+    TrainingPlanExtensionOffering = Shapes::StructureShape.new(name: 'TrainingPlanExtensionOffering')
+    TrainingPlanExtensionOfferingId = Shapes::StringShape.new(name: 'TrainingPlanExtensionOfferingId')
+    TrainingPlanExtensionOfferings = Shapes::ListShape.new(name: 'TrainingPlanExtensionOfferings')
+    TrainingPlanExtensions = Shapes::ListShape.new(name: 'TrainingPlanExtensions')
     TrainingPlanFilter = Shapes::StructureShape.new(name: 'TrainingPlanFilter')
     TrainingPlanFilterName = Shapes::StringShape.new(name: 'TrainingPlanFilterName')
     TrainingPlanFilters = Shapes::ListShape.new(name: 'TrainingPlanFilters')
@@ -6572,6 +6583,15 @@ module Aws::SageMaker
     DescribeTrainingJobResponse.add_member(:output_model_package_arn, Shapes::ShapeRef.new(shape: ModelPackageArn, location_name: "OutputModelPackageArn"))
     DescribeTrainingJobResponse.struct_class = Types::DescribeTrainingJobResponse
 
+    DescribeTrainingPlanExtensionHistoryRequest.add_member(:training_plan_arn, Shapes::ShapeRef.new(shape: TrainingPlanArn, required: true, location_name: "TrainingPlanArn"))
+    DescribeTrainingPlanExtensionHistoryRequest.add_member(:next_token, Shapes::ShapeRef.new(shape: NextToken, location_name: "NextToken"))
+    DescribeTrainingPlanExtensionHistoryRequest.add_member(:max_results, Shapes::ShapeRef.new(shape: MaxResults, location_name: "MaxResults"))
+    DescribeTrainingPlanExtensionHistoryRequest.struct_class = Types::DescribeTrainingPlanExtensionHistoryRequest
+
+    DescribeTrainingPlanExtensionHistoryResponse.add_member(:training_plan_extensions, Shapes::ShapeRef.new(shape: TrainingPlanExtensions, required: true, location_name: "TrainingPlanExtensions"))
+    DescribeTrainingPlanExtensionHistoryResponse.add_member(:next_token, Shapes::ShapeRef.new(shape: NextToken, location_name: "NextToken"))
+    DescribeTrainingPlanExtensionHistoryResponse.struct_class = Types::DescribeTrainingPlanExtensionHistoryResponse
+
     DescribeTrainingPlanRequest.add_member(:training_plan_name, Shapes::ShapeRef.new(shape: TrainingPlanName, required: true, location_name: "TrainingPlanName"))
     DescribeTrainingPlanRequest.struct_class = Types::DescribeTrainingPlanRequest
 
@@ -7134,6 +7154,12 @@ module Aws::SageMaker
     ExplainerConfig.add_member(:clarify_explainer_config, Shapes::ShapeRef.new(shape: ClarifyExplainerConfig, location_name: "ClarifyExplainerConfig"))
     ExplainerConfig.struct_class = Types::ExplainerConfig
 
+    ExtendTrainingPlanRequest.add_member(:training_plan_extension_offering_id, Shapes::ShapeRef.new(shape: TrainingPlanExtensionOfferingId, required: true, location_name: "TrainingPlanExtensionOfferingId"))
+    ExtendTrainingPlanRequest.struct_class = Types::ExtendTrainingPlanRequest
+
+    ExtendTrainingPlanResponse.add_member(:training_plan_extensions, Shapes::ShapeRef.new(shape: TrainingPlanExtensions, required: true, location_name: "TrainingPlanExtensions"))
+    ExtendTrainingPlanResponse.struct_class = Types::ExtendTrainingPlanResponse
+
     FSxLustreConfig.add_member(:size_in_gi_b, Shapes::ShapeRef.new(shape: FSxLustreSizeInGiB, required: true, location_name: "SizeInGiB"))
     FSxLustreConfig.add_member(:per_unit_storage_throughput, Shapes::ShapeRef.new(shape: FSxLustrePerUnitStorageThroughput, required: true, location_name: "PerUnitStorageThroughput"))
     FSxLustreConfig.struct_class = Types::FSxLustreConfig
@@ -10753,6 +10779,8 @@ module Aws::SageMaker
     ReservedCapacityOffering.add_member(:duration_minutes, Shapes::ShapeRef.new(shape: ReservedCapacityDurationMinutes, location_name: "DurationMinutes"))
     ReservedCapacityOffering.add_member(:start_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "StartTime"))
     ReservedCapacityOffering.add_member(:end_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "EndTime"))
+    ReservedCapacityOffering.add_member(:extension_start_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "ExtensionStartTime"))
+    ReservedCapacityOffering.add_member(:extension_end_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "ExtensionEndTime"))
     ReservedCapacityOffering.struct_class = Types::ReservedCapacityOffering
 
     ReservedCapacityOfferings.member = Shapes::ShapeRef.new(shape: ReservedCapacityOffering)
@@ -10979,9 +11007,11 @@ module Aws::SageMaker
     SearchTrainingPlanOfferingsRequest.add_member(:end_time_before, Shapes::ShapeRef.new(shape: Timestamp, location_name: "EndTimeBefore"))
     SearchTrainingPlanOfferingsRequest.add_member(:duration_hours, Shapes::ShapeRef.new(shape: TrainingPlanDurationHoursInput, location_name: "DurationHours"))
     SearchTrainingPlanOfferingsRequest.add_member(:target_resources, Shapes::ShapeRef.new(shape: SageMakerResourceNames, location_name: "TargetResources"))
+    SearchTrainingPlanOfferingsRequest.add_member(:training_plan_arn, Shapes::ShapeRef.new(shape: String, location_name: "TrainingPlanArn"))
     SearchTrainingPlanOfferingsRequest.struct_class = Types::SearchTrainingPlanOfferingsRequest
 
     SearchTrainingPlanOfferingsResponse.add_member(:training_plan_offerings, Shapes::ShapeRef.new(shape: TrainingPlanOfferings, required: true, location_name: "TrainingPlanOfferings"))
+    SearchTrainingPlanOfferingsResponse.add_member(:training_plan_extension_offerings, Shapes::ShapeRef.new(shape: TrainingPlanExtensionOfferings, location_name: "TrainingPlanExtensionOfferings"))
     SearchTrainingPlanOfferingsResponse.struct_class = Types::SearchTrainingPlanOfferingsResponse
 
     SecondaryStatusTransition.add_member(:status, Shapes::ShapeRef.new(shape: SecondaryStatus, required: true, location_name: "Status"))
@@ -11497,6 +11527,32 @@ module Aws::SageMaker
 
     TrainingPlanArns.member = Shapes::ShapeRef.new(shape: TrainingPlanArn)
 
+    TrainingPlanExtension.add_member(:training_plan_extension_offering_id, Shapes::ShapeRef.new(shape: TrainingPlanExtensionOfferingId, required: true, location_name: "TrainingPlanExtensionOfferingId"))
+    TrainingPlanExtension.add_member(:extended_at, Shapes::ShapeRef.new(shape: Timestamp, location_name: "ExtendedAt"))
+    TrainingPlanExtension.add_member(:start_date, Shapes::ShapeRef.new(shape: Timestamp, location_name: "StartDate"))
+    TrainingPlanExtension.add_member(:end_date, Shapes::ShapeRef.new(shape: Timestamp, location_name: "EndDate"))
+    TrainingPlanExtension.add_member(:status, Shapes::ShapeRef.new(shape: String256, location_name: "Status"))
+    TrainingPlanExtension.add_member(:payment_status, Shapes::ShapeRef.new(shape: String256, location_name: "PaymentStatus"))
+    TrainingPlanExtension.add_member(:availability_zone, Shapes::ShapeRef.new(shape: String256, location_name: "AvailabilityZone"))
+    TrainingPlanExtension.add_member(:availability_zone_id, Shapes::ShapeRef.new(shape: AvailabilityZoneId, location_name: "AvailabilityZoneId"))
+    TrainingPlanExtension.add_member(:duration_hours, Shapes::ShapeRef.new(shape: TrainingPlanExtensionDurationHours, location_name: "DurationHours"))
+    TrainingPlanExtension.add_member(:upfront_fee, Shapes::ShapeRef.new(shape: String256, location_name: "UpfrontFee"))
+    TrainingPlanExtension.add_member(:currency_code, Shapes::ShapeRef.new(shape: CurrencyCode, location_name: "CurrencyCode"))
+    TrainingPlanExtension.struct_class = Types::TrainingPlanExtension
+
+    TrainingPlanExtensionOffering.add_member(:training_plan_extension_offering_id, Shapes::ShapeRef.new(shape: TrainingPlanExtensionOfferingId, required: true, location_name: "TrainingPlanExtensionOfferingId"))
+    TrainingPlanExtensionOffering.add_member(:availability_zone, Shapes::ShapeRef.new(shape: String256, location_name: "AvailabilityZone"))
+    TrainingPlanExtensionOffering.add_member(:start_date, Shapes::ShapeRef.new(shape: Timestamp, location_name: "StartDate"))
+    TrainingPlanExtensionOffering.add_member(:end_date, Shapes::ShapeRef.new(shape: Timestamp, location_name: "EndDate"))
+    TrainingPlanExtensionOffering.add_member(:duration_hours, Shapes::ShapeRef.new(shape: TrainingPlanExtensionDurationHours, location_name: "DurationHours"))
+    TrainingPlanExtensionOffering.add_member(:upfront_fee, Shapes::ShapeRef.new(shape: String256, location_name: "UpfrontFee"))
+    TrainingPlanExtensionOffering.add_member(:currency_code, Shapes::ShapeRef.new(shape: CurrencyCode, location_name: "CurrencyCode"))
+    TrainingPlanExtensionOffering.struct_class = Types::TrainingPlanExtensionOffering
+
+    TrainingPlanExtensionOfferings.member = Shapes::ShapeRef.new(shape: TrainingPlanExtensionOffering)
+
+    TrainingPlanExtensions.member = Shapes::ShapeRef.new(shape: TrainingPlanExtension)
+
     TrainingPlanFilter.add_member(:name, Shapes::ShapeRef.new(shape: TrainingPlanFilterName, required: true, location_name: "Name"))
     TrainingPlanFilter.add_member(:value, Shapes::ShapeRef.new(shape: String64, required: true, location_name: "Value"))
     TrainingPlanFilter.struct_class = Types::TrainingPlanFilter
@@ -14280,6 +14336,21 @@ module Aws::SageMaker
         o.errors << Shapes::ShapeRef.new(shape: ResourceNotFound)
       end)
 
+      api.add_operation(:describe_training_plan_extension_history, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "DescribeTrainingPlanExtensionHistory"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: DescribeTrainingPlanExtensionHistoryRequest)
+        o.output = Shapes::ShapeRef.new(shape: DescribeTrainingPlanExtensionHistoryResponse)
+        o.errors << Shapes::ShapeRef.new(shape: ResourceNotFound)
+        o[:pager] = Aws::Pager.new(
+          limit_key: "max_results",
+          tokens: {
+            "next_token" => "next_token"
+          }
+        )
+      end)
+
       api.add_operation(:describe_transform_job, Seahorse::Model::Operation.new.tap do |o|
         o.name = "DescribeTransformJob"
         o.http_method = "POST"
@@ -14367,6 +14438,15 @@ module Aws::SageMaker
         o.output = Shapes::ShapeRef.new(shape: EnableSagemakerServicecatalogPortfolioOutput)
       end)
 
+      api.add_operation(:extend_training_plan, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "ExtendTrainingPlan"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: ExtendTrainingPlanRequest)
+        o.output = Shapes::ShapeRef.new(shape: ExtendTrainingPlanResponse)
+        o.errors << Shapes::ShapeRef.new(shape: ResourceNotFound)
+      end)
+
       api.add_operation(:get_device_fleet_report, Seahorse::Model::Operation.new.tap do |o|
         o.name = "GetDeviceFleetReport"
         o.http_method = "POST"

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

@@ -21962,6 +21962,46 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] training_plan_arn
+    #   The Amazon Resource Name (ARN); of the training plan to retrieve
+    #   extension history for.
+    #   @return [String]
+    #
+    # @!attribute [rw] next_token
+    #   A token to continue pagination if more results are available.
+    #   @return [String]
+    #
+    # @!attribute [rw] max_results
+    #   The maximum number of extensions to return in the response.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeTrainingPlanExtensionHistoryRequest AWS API Documentation
+    #
+    class DescribeTrainingPlanExtensionHistoryRequest < Struct.new(
+      :training_plan_arn,
+      :next_token,
+      :max_results)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] training_plan_extensions
+    #   A list of extensions for the specified training plan.
+    #   @return [Array<Types::TrainingPlanExtension>]
+    #
+    # @!attribute [rw] next_token
+    #   A token to continue pagination if more results are available.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeTrainingPlanExtensionHistoryResponse AWS API Documentation
+    #
+    class DescribeTrainingPlanExtensionHistoryResponse < Struct.new(
+      :training_plan_extensions,
+      :next_token)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] training_plan_name
     #   The name of the training plan to describe.
     #   @return [String]
@@ -24767,6 +24807,33 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] training_plan_extension_offering_id
+    #   The unique identifier of the extension offering to purchase. You can
+    #   retrieve this ID from the `TrainingPlanExtensionOfferings` in the
+    #   response of the `SearchTrainingPlanOfferings` API.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ExtendTrainingPlanRequest AWS API Documentation
+    #
+    class ExtendTrainingPlanRequest < Struct.new(
+      :training_plan_extension_offering_id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] training_plan_extensions
+    #   The list of extensions for the training plan, including the newly
+    #   created extension.
+    #   @return [Array<Types::TrainingPlanExtension>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ExtendTrainingPlanResponse AWS API Documentation
+    #
+    class ExtendTrainingPlanResponse < Struct.new(
+      :training_plan_extensions)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Configuration settings for an Amazon FSx for Lustre file system to be
     # used with the cluster.
     #
@@ -45812,6 +45879,14 @@ module Aws::SageMaker
     #   The end time of the reserved capacity offering.
     #   @return [Time]
     #
+    # @!attribute [rw] extension_start_time
+    #   The start time of the extension for the reserved capacity offering.
+    #   @return [Time]
+    #
+    # @!attribute [rw] extension_end_time
+    #   The end time of the extension for the reserved capacity offering.
+    #   @return [Time]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ReservedCapacityOffering AWS API Documentation
     #
     class ReservedCapacityOffering < Struct.new(
@@ -45824,7 +45899,9 @@ module Aws::SageMaker
       :duration_hours,
       :duration_minutes,
       :start_time,
-      :end_time)
+      :end_time,
+      :extension_start_time,
+      :extension_end_time)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -47339,6 +47416,13 @@ module Aws::SageMaker
     #     deployment.
     #   @return [Array<String>]
     #
+    # @!attribute [rw] training_plan_arn
+    #   The Amazon Resource Name (ARN); of an existing training plan to
+    #   search for extension offerings. When specified, the API returns
+    #   extension offerings that can be used to extend the specified
+    #   training plan.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/SearchTrainingPlanOfferingsRequest AWS API Documentation
     #
     class SearchTrainingPlanOfferingsRequest < Struct.new(
@@ -47349,7 +47433,8 @@ module Aws::SageMaker
       :start_time_after,
       :end_time_before,
       :duration_hours,
-      :target_resources)
+      :target_resources,
+      :training_plan_arn)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -47358,10 +47443,17 @@ module Aws::SageMaker
     #   A list of training plan offerings that match the search criteria.
     #   @return [Array<Types::TrainingPlanOffering>]
     #
+    # @!attribute [rw] training_plan_extension_offerings
+    #   A list of extension offerings available for the specified training
+    #   plan. These offerings can be used with the ` ExtendTrainingPlan `
+    #   API to extend an existing training plan.
+    #   @return [Array<Types::TrainingPlanExtensionOffering>]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/SearchTrainingPlanOfferingsResponse AWS API Documentation
     #
     class SearchTrainingPlanOfferingsResponse < Struct.new(
-      :training_plan_offerings)
+      :training_plan_offerings,
+      :training_plan_extension_offerings)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -50615,6 +50707,119 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Details about an extension to a training plan, including the offering
+    # ID, dates, status, and cost information.
+    #
+    # @!attribute [rw] training_plan_extension_offering_id
+    #   The unique identifier of the extension offering that was used to
+    #   create this extension.
+    #   @return [String]
+    #
+    # @!attribute [rw] extended_at
+    #   The timestamp when the extension was created.
+    #   @return [Time]
+    #
+    # @!attribute [rw] start_date
+    #   The start date of the extension period.
+    #   @return [Time]
+    #
+    # @!attribute [rw] end_date
+    #   The end date of the extension period.
+    #   @return [Time]
+    #
+    # @!attribute [rw] status
+    #   The current status of the extension (e.g., Pending, Active,
+    #   Scheduled, Failed, Expired).
+    #   @return [String]
+    #
+    # @!attribute [rw] payment_status
+    #   The payment processing status of the extension.
+    #   @return [String]
+    #
+    # @!attribute [rw] availability_zone
+    #   The Availability Zone of the extension.
+    #   @return [String]
+    #
+    # @!attribute [rw] availability_zone_id
+    #   The Availability Zone ID of the extension.
+    #   @return [String]
+    #
+    # @!attribute [rw] duration_hours
+    #   The duration of the extension in hours.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] upfront_fee
+    #   The upfront fee for the extension.
+    #   @return [String]
+    #
+    # @!attribute [rw] currency_code
+    #   The currency code for the upfront fee (e.g., USD).
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingPlanExtension AWS API Documentation
+    #
+    class TrainingPlanExtension < Struct.new(
+      :training_plan_extension_offering_id,
+      :extended_at,
+      :start_date,
+      :end_date,
+      :status,
+      :payment_status,
+      :availability_zone,
+      :availability_zone_id,
+      :duration_hours,
+      :upfront_fee,
+      :currency_code)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Details about an available extension offering for a training plan. Use
+    # the offering ID with the ` ExtendTrainingPlan ` API to extend a
+    # training plan.
+    #
+    # @!attribute [rw] training_plan_extension_offering_id
+    #   The unique identifier for this extension offering.
+    #   @return [String]
+    #
+    # @!attribute [rw] availability_zone
+    #   The Availability Zone for this extension offering.
+    #   @return [String]
+    #
+    # @!attribute [rw] start_date
+    #   The start date of this extension offering.
+    #   @return [Time]
+    #
+    # @!attribute [rw] end_date
+    #   The end date of this extension offering.
+    #   @return [Time]
+    #
+    # @!attribute [rw] duration_hours
+    #   The duration of this extension offering in hours.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] upfront_fee
+    #   The upfront fee for this extension offering.
+    #   @return [String]
+    #
+    # @!attribute [rw] currency_code
+    #   The currency code for the upfront fee (e.g., USD).
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingPlanExtensionOffering AWS API Documentation
+    #
+    class TrainingPlanExtensionOffering < Struct.new(
+      :training_plan_extension_offering_id,
+      :availability_zone,
+      :start_date,
+      :end_date,
+      :duration_hours,
+      :upfront_fee,
+      :currency_code)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # A filter to apply when listing or searching for training plans.
     #
     # For more information about how to reserve GPU capacity for your

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.353.0'
+  GEM_VERSION = '1.354.0'
 
 end
 

data/sig/client.rbs

@@ -6925,6 +6925,19 @@ module Aws
                                   ) -> _DescribeTrainingPlanResponseSuccess
                                 | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DescribeTrainingPlanResponseSuccess
 
+      interface _DescribeTrainingPlanExtensionHistoryResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::DescribeTrainingPlanExtensionHistoryResponse]
+        def training_plan_extensions: () -> ::Array[Types::TrainingPlanExtension]
+        def next_token: () -> ::String
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#describe_training_plan_extension_history-instance_method
+      def describe_training_plan_extension_history: (
+                                                      training_plan_arn: ::String,
+                                                      ?next_token: ::String,
+                                                      ?max_results: ::Integer
+                                                    ) -> _DescribeTrainingPlanExtensionHistoryResponseSuccess
+                                                  | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DescribeTrainingPlanExtensionHistoryResponseSuccess
+
       interface _DescribeTransformJobResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::DescribeTransformJobResponse]
         def transform_job_name: () -> ::String
@@ -7087,6 +7100,16 @@ module Aws
                                                      ) -> _EnableSagemakerServicecatalogPortfolioResponseSuccess
                                                    | (?Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _EnableSagemakerServicecatalogPortfolioResponseSuccess
 
+      interface _ExtendTrainingPlanResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::ExtendTrainingPlanResponse]
+        def training_plan_extensions: () -> ::Array[Types::TrainingPlanExtension]
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#extend_training_plan-instance_method
+      def extend_training_plan: (
+                                  training_plan_extension_offering_id: ::String
+                                ) -> _ExtendTrainingPlanResponseSuccess
+                              | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _ExtendTrainingPlanResponseSuccess
+
       interface _GetDeviceFleetReportResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::GetDeviceFleetReportResponse]
         def device_fleet_arn: () -> ::String
@@ -8873,6 +8896,7 @@ module Aws
       interface _SearchTrainingPlanOfferingsResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::SearchTrainingPlanOfferingsResponse]
         def training_plan_offerings: () -> ::Array[Types::TrainingPlanOffering]
+        def training_plan_extension_offerings: () -> ::Array[Types::TrainingPlanExtensionOffering]
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#search_training_plan_offerings-instance_method
       def search_training_plan_offerings: (
@@ -8883,7 +8907,8 @@ module Aws
                                             ?start_time_after: ::Time,
                                             ?end_time_before: ::Time,
                                             ?duration_hours: ::Integer,
-                                            ?target_resources: Array[("training-job" | "hyperpod-cluster" | "endpoint")]
+                                            ?target_resources: Array[("training-job" | "hyperpod-cluster" | "endpoint")],
+                                            ?training_plan_arn: ::String
                                           ) -> _SearchTrainingPlanOfferingsResponseSuccess
                                         | (?Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _SearchTrainingPlanOfferingsResponseSuccess
 

data/sig/types.rbs

@@ -4644,6 +4644,19 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class DescribeTrainingPlanExtensionHistoryRequest
+      attr_accessor training_plan_arn: ::String
+      attr_accessor next_token: ::String
+      attr_accessor max_results: ::Integer
+      SENSITIVE: []
+    end
+
+    class DescribeTrainingPlanExtensionHistoryResponse
+      attr_accessor training_plan_extensions: ::Array[Types::TrainingPlanExtension]
+      attr_accessor next_token: ::String
+      SENSITIVE: []
+    end
+
     class DescribeTrainingPlanRequest
       attr_accessor training_plan_name: ::String
       SENSITIVE: []
@@ -5319,6 +5332,16 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class ExtendTrainingPlanRequest
+      attr_accessor training_plan_extension_offering_id: ::String
+      SENSITIVE: []
+    end
+
+    class ExtendTrainingPlanResponse
+      attr_accessor training_plan_extensions: ::Array[Types::TrainingPlanExtension]
+      SENSITIVE: []
+    end
+
     class FSxLustreConfig
       attr_accessor size_in_gi_b: ::Integer
       attr_accessor per_unit_storage_throughput: ::Integer
@@ -9674,6 +9697,8 @@ module Aws::SageMaker
       attr_accessor duration_minutes: ::Integer
       attr_accessor start_time: ::Time
       attr_accessor end_time: ::Time
+      attr_accessor extension_start_time: ::Time
+      attr_accessor extension_end_time: ::Time
       SENSITIVE: []
     end
 
@@ -9950,11 +9975,13 @@ module Aws::SageMaker
       attr_accessor end_time_before: ::Time
       attr_accessor duration_hours: ::Integer
       attr_accessor target_resources: ::Array[("training-job" | "hyperpod-cluster" | "endpoint")]
+      attr_accessor training_plan_arn: ::String
       SENSITIVE: []
     end
 
     class SearchTrainingPlanOfferingsResponse
       attr_accessor training_plan_offerings: ::Array[Types::TrainingPlanOffering]
+      attr_accessor training_plan_extension_offerings: ::Array[Types::TrainingPlanExtensionOffering]
       SENSITIVE: []
     end
 
@@ -10612,6 +10639,32 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class TrainingPlanExtension
+      attr_accessor training_plan_extension_offering_id: ::String
+      attr_accessor extended_at: ::Time
+      attr_accessor start_date: ::Time
+      attr_accessor end_date: ::Time
+      attr_accessor status: ::String
+      attr_accessor payment_status: ::String
+      attr_accessor availability_zone: ::String
+      attr_accessor availability_zone_id: ::String
+      attr_accessor duration_hours: ::Integer
+      attr_accessor upfront_fee: ::String
+      attr_accessor currency_code: ::String
+      SENSITIVE: []
+    end
+
+    class TrainingPlanExtensionOffering
+      attr_accessor training_plan_extension_offering_id: ::String
+      attr_accessor availability_zone: ::String
+      attr_accessor start_date: ::Time
+      attr_accessor end_date: ::Time
+      attr_accessor duration_hours: ::Integer
+      attr_accessor upfront_fee: ::String
+      attr_accessor currency_code: ::String
+      SENSITIVE: []
+    end
+
     class TrainingPlanFilter
       attr_accessor name: ("Status")
       attr_accessor value: ::String

metadata

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