aws-sdk-sagemaker 1.398.0 → 1.400.0

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.398.0'
+  GEM_VERSION = '1.400.0'
 
 end
 

data/sig/client.rbs

@@ -2845,6 +2845,42 @@ module Aws
                                 ) -> _CreateTrainingPlanResponseSuccess
                               | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _CreateTrainingPlanResponseSuccess
 
+      interface _CreateTrainingSessionResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::CreateTrainingSessionResponse]
+        def training_session_arn: () -> ::String
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#create_training_session-instance_method
+      def create_training_session: (
+                                     training_session_name: ::String,
+                                     ?model_package_group_arn: ::String,
+                                     lora_config: {
+                                       rank: ::Integer,
+                                       alpha: ::Integer
+                                     },
+                                     role_arn: ::String,
+                                     ?vpc_config: {
+                                       security_group_ids: Array[::String],
+                                       subnets: Array[::String]
+                                     },
+                                     model_config: {
+                                       base_model_arn: ::String,
+                                       accept_eula: bool?
+                                     },
+                                     output_config: {
+                                       s3_location: {
+                                         path: ::String,
+                                         kms_key_arn: ::String?
+                                       }
+                                     },
+                                     ?tags: Array[
+                                       {
+                                         key: ::String,
+                                         value: ::String
+                                       }
+                                     ]
+                                   ) -> _CreateTrainingSessionResponseSuccess
+                                 | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _CreateTrainingSessionResponseSuccess
+
       interface _CreateTransformJobResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::CreateTransformJobResponse]
         def transform_job_arn: () -> ::String
@@ -3486,6 +3522,15 @@ module Aws
                                ) -> ::Seahorse::Client::_ResponseSuccess[::Aws::EmptyStructure]
                              | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> ::Seahorse::Client::_ResponseSuccess[::Aws::EmptyStructure]
 
+      interface _DeleteTrainingSessionResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::DeleteTrainingSessionResponse]
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#delete_training_session-instance_method
+      def delete_training_session: (
+                                     training_session_name: ::String
+                                   ) -> _DeleteTrainingSessionResponseSuccess
+                                 | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DeleteTrainingSessionResponseSuccess
+
       interface _DeleteTrialResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::DeleteTrialResponse]
         def trial_arn: () -> ::String
@@ -4877,7 +4922,7 @@ module Aws
         def pipeline_arn: () -> ::String
         def pipeline_execution_arn: () -> ::String
         def pipeline_execution_display_name: () -> ::String
-        def pipeline_execution_status: () -> ("Executing" | "Stopping" | "Stopped" | "Failed" | "Succeeded")
+        def pipeline_execution_status: () -> ("Executing" | "Stopping" | "Stopped" | "Failed" | "Succeeded" | "CompletedWithFailures")
         def pipeline_execution_description: () -> ::String
         def pipeline_experiment_config: () -> Types::PipelineExperimentConfig
         def failure_reason: () -> ::String
@@ -5116,6 +5161,16 @@ module Aws
                                                     ) -> _DescribeTrainingPlanExtensionHistoryResponseSuccess
                                                   | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DescribeTrainingPlanExtensionHistoryResponseSuccess
 
+      interface _DescribeTrainingSessionResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::DescribeTrainingSessionResponse]
+        def training_session_detail: () -> Types::TrainingSessionDetail
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#describe_training_session-instance_method
+      def describe_training_session: (
+                                       training_session_name: ::String
+                                     ) -> _DescribeTrainingSessionResponseSuccess
+                                   | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DescribeTrainingSessionResponseSuccess
+
       interface _DescribeTransformJobResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::DescribeTransformJobResponse]
         def transform_job_name: () -> ::String
@@ -6906,6 +6961,22 @@ module Aws
                                ) -> _ListTrainingPlansResponseSuccess
                              | (?Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _ListTrainingPlansResponseSuccess
 
+      interface _ListTrainingSessionsResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::ListTrainingSessionsResponse]
+        def next_token: () -> ::String
+        def training_session_summaries: () -> ::Array[Types::TrainingSessionSummary]
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#list_training_sessions-instance_method
+      def list_training_sessions: (
+                                    ?next_token: ::String,
+                                    ?max_results: ::Integer,
+                                    ?creation_time_after: ::Time,
+                                    ?creation_time_before: ::Time,
+                                    ?name_contains: ::String,
+                                    ?status_equals: ("Created" | "Blocked" | "Running" | "Failed" | "Scheduled" | "Stopping" | "Stopped" | "Expired" | "Deleting" | "DeletionFailed")
+                                  ) -> _ListTrainingSessionsResponseSuccess
+                                | (?Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _ListTrainingSessionsResponseSuccess
+
       interface _ListTransformJobsResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::ListTransformJobsResponse]
         def transform_job_summaries: () -> ::Array[Types::TransformJobSummary]
@@ -7422,6 +7493,15 @@ module Aws
                              ) -> ::Seahorse::Client::_ResponseSuccess[::Aws::EmptyStructure]
                            | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> ::Seahorse::Client::_ResponseSuccess[::Aws::EmptyStructure]
 
+      interface _StopTrainingSessionResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::StopTrainingSessionResponse]
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#stop_training_session-instance_method
+      def stop_training_session: (
+                                   training_session_name: ::String
+                                 ) -> _StopTrainingSessionResponseSuccess
+                               | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _StopTrainingSessionResponseSuccess
+
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#stop_transform_job-instance_method
       def stop_transform_job: (
                                 transform_job_name: ::String

data/sig/types.rbs

@@ -551,6 +551,12 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class AthenaStepMetadata
+      attr_accessor query_execution_id: ::String
+      attr_accessor s3_output_location: ::String
+      SENSITIVE: []
+    end
+
     class AttachClusterNodeNetworkInterfaceRequest
       attr_accessor cluster_name: ::String
       attr_accessor node_id: ::String
@@ -1000,6 +1006,13 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class BillableTokenUsage
+      attr_accessor prefill_token_count: ::Integer
+      attr_accessor sample_token_count: ::Integer
+      attr_accessor training_token_count: ::Integer
+      SENSITIVE: []
+    end
+
     class BlueGreenUpdatePolicy
       attr_accessor traffic_routing_configuration: Types::TrafficRoutingConfig
       attr_accessor termination_wait_in_seconds: ::Integer
@@ -1227,6 +1240,20 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class CleanupFailure
+      attr_accessor step_name: ::String
+      attr_accessor error_code: ("ClientError" | "InternalFailure")
+      attr_accessor error_message: ::String
+      SENSITIVE: []
+    end
+
+    class CleanupStepMetadata
+      attr_accessor cleaned_steps: ::Array[::String]
+      attr_accessor failed_steps: ::Array[Types::CleanupFailure]
+      attr_accessor skipped_steps: ::Array[::String]
+      SENSITIVE: []
+    end
+
     class ClusterAccountingDatabase
       attr_accessor endpoint: ::String
       attr_accessor port: ::Integer
@@ -2941,6 +2968,23 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class CreateTrainingSessionRequest
+      attr_accessor training_session_name: ::String
+      attr_accessor model_package_group_arn: ::String
+      attr_accessor lora_config: Types::LoraConfig
+      attr_accessor role_arn: ::String
+      attr_accessor vpc_config: Types::VpcConfig
+      attr_accessor model_config: Types::TrainingSessionModelConfig
+      attr_accessor output_config: Types::TrainingSessionOutputConfig
+      attr_accessor tags: ::Array[Types::Tag]
+      SENSITIVE: []
+    end
+
+    class CreateTrainingSessionResponse
+      attr_accessor training_session_arn: ::String
+      SENSITIVE: []
+    end
+
     class CreateTransformJobRequest
       attr_accessor transform_job_name: ::String
       attr_accessor model_name: ::String
@@ -3627,6 +3671,14 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class DeleteTrainingSessionRequest
+      attr_accessor training_session_name: ::String
+      SENSITIVE: []
+    end
+
+    class DeleteTrainingSessionResponse < Aws::EmptyStructure
+    end
+
     class DeleteTrialComponentRequest
       attr_accessor trial_component_name: ::String
       SENSITIVE: []
@@ -5042,7 +5094,7 @@ module Aws::SageMaker
       attr_accessor pipeline_arn: ::String
       attr_accessor pipeline_execution_arn: ::String
       attr_accessor pipeline_execution_display_name: ::String
-      attr_accessor pipeline_execution_status: ("Executing" | "Stopping" | "Stopped" | "Failed" | "Succeeded")
+      attr_accessor pipeline_execution_status: ("Executing" | "Stopping" | "Stopped" | "Failed" | "Succeeded" | "CompletedWithFailures")
       attr_accessor pipeline_execution_description: ::String
       attr_accessor pipeline_experiment_config: Types::PipelineExperimentConfig
       attr_accessor failure_reason: ::String
@@ -5302,6 +5354,16 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class DescribeTrainingSessionRequest
+      attr_accessor training_session_name: ::String
+      SENSITIVE: []
+    end
+
+    class DescribeTrainingSessionResponse
+      attr_accessor training_session_detail: Types::TrainingSessionDetail
+      SENSITIVE: []
+    end
+
     class DescribeTransformJobRequest
       attr_accessor transform_job_name: ::String
       SENSITIVE: []
@@ -5619,6 +5681,14 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class EMRServerlessStepMetadata
+      attr_accessor application_id: ::String
+      attr_accessor job_run_id: ::String
+      attr_accessor job_run_name: ::String
+      attr_accessor log_file_path: ::String
+      SENSITIVE: []
+    end
+
     class EMRStepMetadata
       attr_accessor cluster_id: ::String
       attr_accessor step_id: ::String
@@ -6102,6 +6172,23 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class ForEachSelectorResult
+      attr_accessor step_name: ::String
+      attr_accessor index: ::Integer
+      SENSITIVE: []
+    end
+
+    class ForEachStepMetadata
+      attr_accessor status: ("Succeeded" | "Failed" | "Stopped")
+      attr_accessor input_item_count: ::Integer
+      attr_accessor total_step_count: ::Integer
+      attr_accessor succeeded_step_count: ::Integer
+      attr_accessor failed_step_count: ::Integer
+      attr_accessor stopped_step_count: ::Integer
+      attr_accessor selector: Types::ForEachSelectorResult
+      SENSITIVE: []
+    end
+
     class GenerativeAiSettings
       attr_accessor amazon_bedrock_role_arn: ::String
       SENSITIVE: []
@@ -6904,6 +6991,13 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class IterationContext
+      attr_accessor parent_step_name: ::String
+      attr_accessor index: ::Integer
+      attr_accessor item: ::String
+      SENSITIVE: []
+    end
+
     class Job
       attr_accessor job_name: ::String
       attr_accessor job_arn: ::String
@@ -8624,6 +8718,22 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class ListTrainingSessionsRequest
+      attr_accessor next_token: ::String
+      attr_accessor max_results: ::Integer
+      attr_accessor creation_time_after: ::Time
+      attr_accessor creation_time_before: ::Time
+      attr_accessor name_contains: ::String
+      attr_accessor status_equals: ("Created" | "Blocked" | "Running" | "Failed" | "Scheduled" | "Stopping" | "Stopped" | "Expired" | "Deleting" | "DeletionFailed")
+      SENSITIVE: []
+    end
+
+    class ListTrainingSessionsResponse
+      attr_accessor next_token: ::String
+      attr_accessor training_session_summaries: ::Array[Types::TrainingSessionSummary]
+      SENSITIVE: []
+    end
+
     class ListTransformJobsRequest
       attr_accessor creation_time_after: ::Time
       attr_accessor creation_time_before: ::Time
@@ -8740,6 +8850,12 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class LoraConfig
+      attr_accessor rank: ::Integer
+      attr_accessor alpha: ::Integer
+      SENSITIVE: []
+    end
+
     class MLflowConfiguration
       attr_accessor mlflow_resource_arn: ::String
       attr_accessor mlflow_experiment_name: ::String
@@ -9900,7 +10016,7 @@ module Aws::SageMaker
       attr_accessor pipeline_arn: ::String
       attr_accessor pipeline_execution_arn: ::String
       attr_accessor pipeline_execution_display_name: ::String
-      attr_accessor pipeline_execution_status: ("Executing" | "Stopping" | "Stopped" | "Failed" | "Succeeded")
+      attr_accessor pipeline_execution_status: ("Executing" | "Stopping" | "Stopped" | "Failed" | "Succeeded" | "CompletedWithFailures")
       attr_accessor pipeline_execution_description: ::String
       attr_accessor pipeline_experiment_config: Types::PipelineExperimentConfig
       attr_accessor failure_reason: ::String
@@ -9928,6 +10044,7 @@ module Aws::SageMaker
       attr_accessor metadata: Types::PipelineExecutionStepMetadata
       attr_accessor attempt_count: ::Integer
       attr_accessor selective_execution_result: Types::SelectiveExecutionResult
+      attr_accessor iteration_context: Types::IterationContext
       SENSITIVE: []
     end
 
@@ -9942,6 +10059,7 @@ module Aws::SageMaker
       attr_accessor callback: Types::CallbackStepMetadata
       attr_accessor lambda: Types::LambdaStepMetadata
       attr_accessor emr: Types::EMRStepMetadata
+      attr_accessor emr_serverless: Types::EMRServerlessStepMetadata
       attr_accessor quality_check: Types::QualityCheckStepMetadata
       attr_accessor clarify_check: Types::ClarifyCheckStepMetadata
       attr_accessor fail: Types::FailStepMetadata
@@ -9955,13 +10073,16 @@ module Aws::SageMaker
       attr_accessor inference_component: Types::InferenceComponentMetadata
       attr_accessor lineage: Types::LineageMetadata
       attr_accessor job: Types::JobStepMetadata
+      attr_accessor athena: Types::AthenaStepMetadata
+      attr_accessor cleanup: Types::CleanupStepMetadata
+      attr_accessor for_each: Types::ForEachStepMetadata
       SENSITIVE: []
     end
 
     class PipelineExecutionSummary
       attr_accessor pipeline_execution_arn: ::String
       attr_accessor start_time: ::Time
-      attr_accessor pipeline_execution_status: ("Executing" | "Stopping" | "Stopped" | "Failed" | "Succeeded")
+      attr_accessor pipeline_execution_status: ("Executing" | "Stopping" | "Stopped" | "Failed" | "Succeeded" | "CompletedWithFailures")
       attr_accessor pipeline_execution_description: ::String
       attr_accessor pipeline_execution_display_name: ::String
       attr_accessor pipeline_execution_failure_reason: ::String
@@ -9997,7 +10118,7 @@ module Aws::SageMaker
       attr_accessor last_modified_by: Types::UserContext
       attr_accessor last_executed_pipeline_execution_arn: ::String
       attr_accessor last_executed_pipeline_execution_display_name: ::String
-      attr_accessor last_executed_pipeline_execution_status: ("Executing" | "Stopping" | "Stopped" | "Failed" | "Succeeded")
+      attr_accessor last_executed_pipeline_execution_status: ("Executing" | "Stopping" | "Stopped" | "Failed" | "Succeeded" | "CompletedWithFailures")
       SENSITIVE: []
     end
 
@@ -11306,6 +11427,14 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class StopTrainingSessionRequest
+      attr_accessor training_session_name: ::String
+      SENSITIVE: []
+    end
+
+    class StopTrainingSessionResponse < Aws::EmptyStructure
+    end
+
     class StopTransformJobRequest
       attr_accessor transform_job_name: ::String
       SENSITIVE: []
@@ -11674,6 +11803,51 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class TrainingSessionDetail
+      attr_accessor training_session_arn: ::String
+      attr_accessor training_session_name: ::String
+      attr_accessor training_session_status: ("Created" | "Blocked" | "Running" | "Failed" | "Scheduled" | "Stopping" | "Stopped" | "Expired" | "Deleting" | "DeletionFailed")
+      attr_accessor model_package_group_arn: ::String
+      attr_accessor creation_time: ::Time
+      attr_accessor last_modified_time: ::Time
+      attr_accessor end_time: ::Time
+      attr_accessor error_details: ::String
+      attr_accessor last_received_sequence_number: ::Integer
+      attr_accessor billable_token_usage: Types::BillableTokenUsage
+      attr_accessor vpc_config: Types::VpcConfig
+      attr_accessor role_arn: ::String
+      attr_accessor lora_config: Types::LoraConfig
+      attr_accessor model_config: Types::TrainingSessionModelConfig
+      attr_accessor output_config: Types::TrainingSessionOutputConfig
+      attr_accessor tags: ::Array[Types::Tag]
+      SENSITIVE: []
+    end
+
+    class TrainingSessionModelConfig
+      attr_accessor base_model_arn: ::String
+      attr_accessor accept_eula: bool
+      SENSITIVE: []
+    end
+
+    class TrainingSessionOutputConfig
+      attr_accessor s3_location: Types::TrainingSessionS3Location
+      SENSITIVE: []
+    end
+
+    class TrainingSessionS3Location
+      attr_accessor path: ::String
+      attr_accessor kms_key_arn: ::String
+      SENSITIVE: []
+    end
+
+    class TrainingSessionSummary
+      attr_accessor training_session_arn: ::String
+      attr_accessor training_session_status: ("Created" | "Blocked" | "Running" | "Failed" | "Scheduled" | "Stopping" | "Stopped" | "Expired" | "Deleting" | "DeletionFailed")
+      attr_accessor creation_time: ::Time
+      attr_accessor last_modified_time: ::Time
+      SENSITIVE: []
+    end
+
     class TrainingSpecification
       attr_accessor training_image: ::String
       attr_accessor training_image_digest: ::String

metadata

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