aws-sdk-sagemaker 1.342.0 → 1.343.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.342.0'
+  GEM_VERSION = '1.343.0'
 
 end
 

data/sig/client.rbs

@@ -344,7 +344,13 @@ module Aws
                                       compression_type: ("None" | "Gzip")?,
                                       etag: ::String?
                                     }?,
-                                    model_data_etag: ::String?
+                                    model_data_etag: ::String?,
+                                    is_checkpoint: bool?,
+                                    base_model: {
+                                      hub_content_name: ::String?,
+                                      hub_content_version: ::String?,
+                                      recipe_name: ::String?
+                                    }?
                                   },
                                 ],
                                 supported_transform_instance_types: Array[("ml.m4.xlarge" | "ml.m4.2xlarge" | "ml.m4.4xlarge" | "ml.m4.10xlarge" | "ml.m4.16xlarge" | "ml.c4.xlarge" | "ml.c4.2xlarge" | "ml.c4.4xlarge" | "ml.c4.8xlarge" | "ml.p2.xlarge" | "ml.p2.8xlarge" | "ml.p2.16xlarge" | "ml.p3.2xlarge" | "ml.p3.8xlarge" | "ml.p3.16xlarge" | "ml.c5.xlarge" | "ml.c5.2xlarge" | "ml.c5.4xlarge" | "ml.c5.9xlarge" | "ml.c5.18xlarge" | "ml.m5.large" | "ml.m5.xlarge" | "ml.m5.2xlarge" | "ml.m5.4xlarge" | "ml.m5.12xlarge" | "ml.m5.24xlarge" | "ml.m6i.large" | "ml.m6i.xlarge" | "ml.m6i.2xlarge" | "ml.m6i.4xlarge" | "ml.m6i.8xlarge" | "ml.m6i.12xlarge" | "ml.m6i.16xlarge" | "ml.m6i.24xlarge" | "ml.m6i.32xlarge" | "ml.c6i.large" | "ml.c6i.xlarge" | "ml.c6i.2xlarge" | "ml.c6i.4xlarge" | "ml.c6i.8xlarge" | "ml.c6i.12xlarge" | "ml.c6i.16xlarge" | "ml.c6i.24xlarge" | "ml.c6i.32xlarge" | "ml.r6i.large" | "ml.r6i.xlarge" | "ml.r6i.2xlarge" | "ml.r6i.4xlarge" | "ml.r6i.8xlarge" | "ml.r6i.12xlarge" | "ml.r6i.16xlarge" | "ml.r6i.24xlarge" | "ml.r6i.32xlarge" | "ml.m7i.large" | "ml.m7i.xlarge" | "ml.m7i.2xlarge" | "ml.m7i.4xlarge" | "ml.m7i.8xlarge" | "ml.m7i.12xlarge" | "ml.m7i.16xlarge" | "ml.m7i.24xlarge" | "ml.m7i.48xlarge" | "ml.c7i.large" | "ml.c7i.xlarge" | "ml.c7i.2xlarge" | "ml.c7i.4xlarge" | "ml.c7i.8xlarge" | "ml.c7i.12xlarge" | "ml.c7i.16xlarge" | "ml.c7i.24xlarge" | "ml.c7i.48xlarge" | "ml.r7i.large" | "ml.r7i.xlarge" | "ml.r7i.2xlarge" | "ml.r7i.4xlarge" | "ml.r7i.8xlarge" | "ml.r7i.12xlarge" | "ml.r7i.16xlarge" | "ml.r7i.24xlarge" | "ml.r7i.48xlarge" | "ml.g4dn.xlarge" | "ml.g4dn.2xlarge" | "ml.g4dn.4xlarge" | "ml.g4dn.8xlarge" | "ml.g4dn.12xlarge" | "ml.g4dn.16xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.trn1.2xlarge" | "ml.trn1.32xlarge" | "ml.inf2.xlarge" | "ml.inf2.8xlarge" | "ml.inf2.24xlarge" | "ml.inf2.48xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.12xlarge" | "ml.g6.16xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge")]?,
@@ -382,6 +388,9 @@ module Aws
                                               file_system_access_mode: ("rw" | "ro"),
                                               file_system_type: ("EFS" | "FSxLustre"),
                                               directory_path: ::String
+                                            }?,
+                                            dataset_source: {
+                                              dataset_arn: ::String
                                             }?
                                           },
                                           content_type: ::String?,
@@ -934,7 +943,7 @@ module Aws
                             ?orchestrator: {
                               eks: {
                                 cluster_arn: ::String
-                              }?
+                              }
                             },
                             ?node_recovery: ("Automatic" | "None"),
                             ?tiered_storage_config: {
@@ -2241,6 +2250,9 @@ module Aws
                                                          file_system_access_mode: ("rw" | "ro"),
                                                          file_system_type: ("EFS" | "FSxLustre"),
                                                          directory_path: ::String
+                                                       }?,
+                                                       dataset_source: {
+                                                         dataset_arn: ::String
                                                        }?
                                                      },
                                                      content_type: ::String?,
@@ -2388,6 +2400,9 @@ module Aws
                                                            file_system_access_mode: ("rw" | "ro"),
                                                            file_system_type: ("EFS" | "FSxLustre"),
                                                            directory_path: ::String
+                                                         }?,
+                                                         dataset_source: {
+                                                           dataset_arn: ::String
                                                          }?
                                                        },
                                                        content_type: ::String?,
@@ -3220,6 +3235,7 @@ module Aws
                                   ?model_package_name: ::String,
                                   ?model_package_group_name: ::String,
                                   ?model_package_description: ::String,
+                                  ?model_package_registration_type: ("Logged" | "Registered"),
                                   ?inference_specification: {
                                     containers: Array[
                                       {
@@ -3257,7 +3273,13 @@ module Aws
                                           compression_type: ("None" | "Gzip")?,
                                           etag: ::String?
                                         }?,
-                                        model_data_etag: ::String?
+                                        model_data_etag: ::String?,
+                                        is_checkpoint: bool?,
+                                        base_model: {
+                                          hub_content_name: ::String?,
+                                          hub_content_version: ::String?,
+                                          recipe_name: ::String?
+                                        }?
                                       },
                                     ],
                                     supported_transform_instance_types: Array[("ml.m4.xlarge" | "ml.m4.2xlarge" | "ml.m4.4xlarge" | "ml.m4.10xlarge" | "ml.m4.16xlarge" | "ml.c4.xlarge" | "ml.c4.2xlarge" | "ml.c4.4xlarge" | "ml.c4.8xlarge" | "ml.p2.xlarge" | "ml.p2.8xlarge" | "ml.p2.16xlarge" | "ml.p3.2xlarge" | "ml.p3.8xlarge" | "ml.p3.16xlarge" | "ml.c5.xlarge" | "ml.c5.2xlarge" | "ml.c5.4xlarge" | "ml.c5.9xlarge" | "ml.c5.18xlarge" | "ml.m5.large" | "ml.m5.xlarge" | "ml.m5.2xlarge" | "ml.m5.4xlarge" | "ml.m5.12xlarge" | "ml.m5.24xlarge" | "ml.m6i.large" | "ml.m6i.xlarge" | "ml.m6i.2xlarge" | "ml.m6i.4xlarge" | "ml.m6i.8xlarge" | "ml.m6i.12xlarge" | "ml.m6i.16xlarge" | "ml.m6i.24xlarge" | "ml.m6i.32xlarge" | "ml.c6i.large" | "ml.c6i.xlarge" | "ml.c6i.2xlarge" | "ml.c6i.4xlarge" | "ml.c6i.8xlarge" | "ml.c6i.12xlarge" | "ml.c6i.16xlarge" | "ml.c6i.24xlarge" | "ml.c6i.32xlarge" | "ml.r6i.large" | "ml.r6i.xlarge" | "ml.r6i.2xlarge" | "ml.r6i.4xlarge" | "ml.r6i.8xlarge" | "ml.r6i.12xlarge" | "ml.r6i.16xlarge" | "ml.r6i.24xlarge" | "ml.r6i.32xlarge" | "ml.m7i.large" | "ml.m7i.xlarge" | "ml.m7i.2xlarge" | "ml.m7i.4xlarge" | "ml.m7i.8xlarge" | "ml.m7i.12xlarge" | "ml.m7i.16xlarge" | "ml.m7i.24xlarge" | "ml.m7i.48xlarge" | "ml.c7i.large" | "ml.c7i.xlarge" | "ml.c7i.2xlarge" | "ml.c7i.4xlarge" | "ml.c7i.8xlarge" | "ml.c7i.12xlarge" | "ml.c7i.16xlarge" | "ml.c7i.24xlarge" | "ml.c7i.48xlarge" | "ml.r7i.large" | "ml.r7i.xlarge" | "ml.r7i.2xlarge" | "ml.r7i.4xlarge" | "ml.r7i.8xlarge" | "ml.r7i.12xlarge" | "ml.r7i.16xlarge" | "ml.r7i.24xlarge" | "ml.r7i.48xlarge" | "ml.g4dn.xlarge" | "ml.g4dn.2xlarge" | "ml.g4dn.4xlarge" | "ml.g4dn.8xlarge" | "ml.g4dn.12xlarge" | "ml.g4dn.16xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.trn1.2xlarge" | "ml.trn1.32xlarge" | "ml.inf2.xlarge" | "ml.inf2.8xlarge" | "ml.inf2.24xlarge" | "ml.inf2.48xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.12xlarge" | "ml.g6.16xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge")]?,
@@ -3491,7 +3513,13 @@ module Aws
                                             compression_type: ("None" | "Gzip")?,
                                             etag: ::String?
                                           }?,
-                                          model_data_etag: ::String?
+                                          model_data_etag: ::String?,
+                                          is_checkpoint: bool?,
+                                          base_model: {
+                                            hub_content_name: ::String?,
+                                            hub_content_version: ::String?,
+                                            recipe_name: ::String?
+                                          }?
                                         },
                                       ],
                                       supported_transform_instance_types: Array[("ml.m4.xlarge" | "ml.m4.2xlarge" | "ml.m4.4xlarge" | "ml.m4.10xlarge" | "ml.m4.16xlarge" | "ml.c4.xlarge" | "ml.c4.2xlarge" | "ml.c4.4xlarge" | "ml.c4.8xlarge" | "ml.p2.xlarge" | "ml.p2.8xlarge" | "ml.p2.16xlarge" | "ml.p3.2xlarge" | "ml.p3.8xlarge" | "ml.p3.16xlarge" | "ml.c5.xlarge" | "ml.c5.2xlarge" | "ml.c5.4xlarge" | "ml.c5.9xlarge" | "ml.c5.18xlarge" | "ml.m5.large" | "ml.m5.xlarge" | "ml.m5.2xlarge" | "ml.m5.4xlarge" | "ml.m5.12xlarge" | "ml.m5.24xlarge" | "ml.m6i.large" | "ml.m6i.xlarge" | "ml.m6i.2xlarge" | "ml.m6i.4xlarge" | "ml.m6i.8xlarge" | "ml.m6i.12xlarge" | "ml.m6i.16xlarge" | "ml.m6i.24xlarge" | "ml.m6i.32xlarge" | "ml.c6i.large" | "ml.c6i.xlarge" | "ml.c6i.2xlarge" | "ml.c6i.4xlarge" | "ml.c6i.8xlarge" | "ml.c6i.12xlarge" | "ml.c6i.16xlarge" | "ml.c6i.24xlarge" | "ml.c6i.32xlarge" | "ml.r6i.large" | "ml.r6i.xlarge" | "ml.r6i.2xlarge" | "ml.r6i.4xlarge" | "ml.r6i.8xlarge" | "ml.r6i.12xlarge" | "ml.r6i.16xlarge" | "ml.r6i.24xlarge" | "ml.r6i.32xlarge" | "ml.m7i.large" | "ml.m7i.xlarge" | "ml.m7i.2xlarge" | "ml.m7i.4xlarge" | "ml.m7i.8xlarge" | "ml.m7i.12xlarge" | "ml.m7i.16xlarge" | "ml.m7i.24xlarge" | "ml.m7i.48xlarge" | "ml.c7i.large" | "ml.c7i.xlarge" | "ml.c7i.2xlarge" | "ml.c7i.4xlarge" | "ml.c7i.8xlarge" | "ml.c7i.12xlarge" | "ml.c7i.16xlarge" | "ml.c7i.24xlarge" | "ml.c7i.48xlarge" | "ml.r7i.large" | "ml.r7i.xlarge" | "ml.r7i.2xlarge" | "ml.r7i.4xlarge" | "ml.r7i.8xlarge" | "ml.r7i.12xlarge" | "ml.r7i.16xlarge" | "ml.r7i.24xlarge" | "ml.r7i.48xlarge" | "ml.g4dn.xlarge" | "ml.g4dn.2xlarge" | "ml.g4dn.4xlarge" | "ml.g4dn.8xlarge" | "ml.g4dn.12xlarge" | "ml.g4dn.16xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.trn1.2xlarge" | "ml.trn1.32xlarge" | "ml.inf2.xlarge" | "ml.inf2.8xlarge" | "ml.inf2.24xlarge" | "ml.inf2.48xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.12xlarge" | "ml.g6.16xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge")]?,
@@ -4342,6 +4370,9 @@ module Aws
                                          file_system_access_mode: ("rw" | "ro"),
                                          file_system_type: ("EFS" | "FSxLustre"),
                                          directory_path: ::String
+                                       }?,
+                                       dataset_source: {
+                                         dataset_arn: ::String
                                        }?
                                      },
                                      content_type: ::String?,
@@ -4465,6 +4496,24 @@ module Aws
                                  },
                                  ?session_chaining_config: {
                                    enable_session_tag_chaining: bool?
+                                 },
+                                 ?serverless_job_config: {
+                                   base_model_arn: ::String,
+                                   accept_eula: bool?,
+                                   job_type: ("FineTuning" | "Evaluation"),
+                                   customization_technique: ("SFT" | "DPO" | "RLVR" | "RLAIF")?,
+                                   peft: ("LORA")?,
+                                   evaluation_type: ("LLMAJEvaluation" | "CustomScorerEvaluation" | "BenchmarkEvaluation")?,
+                                   evaluator_arn: ::String?
+                                 },
+                                 ?mlflow_config: {
+                                   mlflow_resource_arn: ::String,
+                                   mlflow_experiment_name: ::String?,
+                                   mlflow_run_name: ::String?
+                                 },
+                                 ?model_package_config: {
+                                   model_package_group_arn: ::String,
+                                   source_model_package_arn: ::String?
                                  }
                                ) -> _CreateTrainingJobResponseSuccess
                              | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _CreateTrainingJobResponseSuccess
@@ -6374,6 +6423,7 @@ module Aws
         def model_package_name: () -> ::String
         def model_package_group_name: () -> ::String
         def model_package_version: () -> ::Integer
+        def model_package_registration_type: () -> ("Logged" | "Registered")
         def model_package_arn: () -> ::String
         def model_package_description: () -> ::String
         def creation_time: () -> ::Time
@@ -6616,6 +6666,7 @@ module Aws
         def parallelism_configuration: () -> Types::ParallelismConfiguration
         def selective_execution_config: () -> Types::SelectiveExecutionConfig
         def pipeline_version_id: () -> ::Integer
+        def m_lflow_config: () -> Types::MLflowConfiguration
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#describe_pipeline_execution-instance_method
       def describe_pipeline_execution: (
@@ -6776,6 +6827,7 @@ module Aws
         def checkpoint_config: () -> Types::CheckpointConfig
         def training_time_in_seconds: () -> ::Integer
         def billable_time_in_seconds: () -> ::Integer
+        def billable_token_count: () -> ::Integer
         def debug_hook_config: () -> Types::DebugHookConfig
         def experiment_config: () -> Types::ExperimentConfig
         def debug_rule_configurations: () -> ::Array[Types::DebugRuleConfiguration]
@@ -6789,6 +6841,12 @@ module Aws
         def retry_strategy: () -> Types::RetryStrategy
         def remote_debug_config: () -> Types::RemoteDebugConfig
         def infra_check_config: () -> Types::InfraCheckConfig
+        def serverless_job_config: () -> Types::ServerlessJobConfig
+        def mlflow_config: () -> Types::MlflowConfig
+        def model_package_config: () -> Types::ModelPackageConfig
+        def mlflow_details: () -> Types::MlflowDetails
+        def progress_info: () -> Types::TrainingProgressInfo
+        def output_model_package_arn: () -> ::String
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMaker/Client.html#describe_training_job-instance_method
       def describe_training_job: (
@@ -8880,7 +8938,8 @@ module Aws
                                           },
                                         ]
                                       },
-                                      ?pipeline_version_id: ::Integer
+                                      ?pipeline_version_id: ::Integer,
+                                      ?mlflow_experiment_name: ::String
                                     ) -> _StartPipelineExecutionResponseSuccess
                                   | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _StartPipelineExecutionResponseSuccess
 
@@ -10121,6 +10180,7 @@ module Aws
       def update_model_package: (
                                   model_package_arn: ::String,
                                   ?model_approval_status: ("Approved" | "Rejected" | "PendingManualApproval"),
+                                  ?model_package_registration_type: ("Logged" | "Registered"),
                                   ?approval_description: ::String,
                                   ?customer_metadata_properties: Hash[::String, ::String],
                                   ?customer_metadata_properties_to_remove: Array[::String],
@@ -10164,7 +10224,13 @@ module Aws
                                             compression_type: ("None" | "Gzip")?,
                                             etag: ::String?
                                           }?,
-                                          model_data_etag: ::String?
+                                          model_data_etag: ::String?,
+                                          is_checkpoint: bool?,
+                                          base_model: {
+                                            hub_content_name: ::String?,
+                                            hub_content_version: ::String?,
+                                            recipe_name: ::String?
+                                          }?
                                         },
                                       ],
                                       supported_transform_instance_types: Array[("ml.m4.xlarge" | "ml.m4.2xlarge" | "ml.m4.4xlarge" | "ml.m4.10xlarge" | "ml.m4.16xlarge" | "ml.c4.xlarge" | "ml.c4.2xlarge" | "ml.c4.4xlarge" | "ml.c4.8xlarge" | "ml.p2.xlarge" | "ml.p2.8xlarge" | "ml.p2.16xlarge" | "ml.p3.2xlarge" | "ml.p3.8xlarge" | "ml.p3.16xlarge" | "ml.c5.xlarge" | "ml.c5.2xlarge" | "ml.c5.4xlarge" | "ml.c5.9xlarge" | "ml.c5.18xlarge" | "ml.m5.large" | "ml.m5.xlarge" | "ml.m5.2xlarge" | "ml.m5.4xlarge" | "ml.m5.12xlarge" | "ml.m5.24xlarge" | "ml.m6i.large" | "ml.m6i.xlarge" | "ml.m6i.2xlarge" | "ml.m6i.4xlarge" | "ml.m6i.8xlarge" | "ml.m6i.12xlarge" | "ml.m6i.16xlarge" | "ml.m6i.24xlarge" | "ml.m6i.32xlarge" | "ml.c6i.large" | "ml.c6i.xlarge" | "ml.c6i.2xlarge" | "ml.c6i.4xlarge" | "ml.c6i.8xlarge" | "ml.c6i.12xlarge" | "ml.c6i.16xlarge" | "ml.c6i.24xlarge" | "ml.c6i.32xlarge" | "ml.r6i.large" | "ml.r6i.xlarge" | "ml.r6i.2xlarge" | "ml.r6i.4xlarge" | "ml.r6i.8xlarge" | "ml.r6i.12xlarge" | "ml.r6i.16xlarge" | "ml.r6i.24xlarge" | "ml.r6i.32xlarge" | "ml.m7i.large" | "ml.m7i.xlarge" | "ml.m7i.2xlarge" | "ml.m7i.4xlarge" | "ml.m7i.8xlarge" | "ml.m7i.12xlarge" | "ml.m7i.16xlarge" | "ml.m7i.24xlarge" | "ml.m7i.48xlarge" | "ml.c7i.large" | "ml.c7i.xlarge" | "ml.c7i.2xlarge" | "ml.c7i.4xlarge" | "ml.c7i.8xlarge" | "ml.c7i.12xlarge" | "ml.c7i.16xlarge" | "ml.c7i.24xlarge" | "ml.c7i.48xlarge" | "ml.r7i.large" | "ml.r7i.xlarge" | "ml.r7i.2xlarge" | "ml.r7i.4xlarge" | "ml.r7i.8xlarge" | "ml.r7i.12xlarge" | "ml.r7i.16xlarge" | "ml.r7i.24xlarge" | "ml.r7i.48xlarge" | "ml.g4dn.xlarge" | "ml.g4dn.2xlarge" | "ml.g4dn.4xlarge" | "ml.g4dn.8xlarge" | "ml.g4dn.12xlarge" | "ml.g4dn.16xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.trn1.2xlarge" | "ml.trn1.32xlarge" | "ml.inf2.xlarge" | "ml.inf2.8xlarge" | "ml.inf2.24xlarge" | "ml.inf2.48xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.12xlarge" | "ml.g6.16xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge")]?,
@@ -10210,7 +10276,13 @@ module Aws
                                           compression_type: ("None" | "Gzip")?,
                                           etag: ::String?
                                         }?,
-                                        model_data_etag: ::String?
+                                        model_data_etag: ::String?,
+                                        is_checkpoint: bool?,
+                                        base_model: {
+                                          hub_content_name: ::String?,
+                                          hub_content_version: ::String?,
+                                          recipe_name: ::String?
+                                        }?
                                       },
                                     ],
                                     supported_transform_instance_types: Array[("ml.m4.xlarge" | "ml.m4.2xlarge" | "ml.m4.4xlarge" | "ml.m4.10xlarge" | "ml.m4.16xlarge" | "ml.c4.xlarge" | "ml.c4.2xlarge" | "ml.c4.4xlarge" | "ml.c4.8xlarge" | "ml.p2.xlarge" | "ml.p2.8xlarge" | "ml.p2.16xlarge" | "ml.p3.2xlarge" | "ml.p3.8xlarge" | "ml.p3.16xlarge" | "ml.c5.xlarge" | "ml.c5.2xlarge" | "ml.c5.4xlarge" | "ml.c5.9xlarge" | "ml.c5.18xlarge" | "ml.m5.large" | "ml.m5.xlarge" | "ml.m5.2xlarge" | "ml.m5.4xlarge" | "ml.m5.12xlarge" | "ml.m5.24xlarge" | "ml.m6i.large" | "ml.m6i.xlarge" | "ml.m6i.2xlarge" | "ml.m6i.4xlarge" | "ml.m6i.8xlarge" | "ml.m6i.12xlarge" | "ml.m6i.16xlarge" | "ml.m6i.24xlarge" | "ml.m6i.32xlarge" | "ml.c6i.large" | "ml.c6i.xlarge" | "ml.c6i.2xlarge" | "ml.c6i.4xlarge" | "ml.c6i.8xlarge" | "ml.c6i.12xlarge" | "ml.c6i.16xlarge" | "ml.c6i.24xlarge" | "ml.c6i.32xlarge" | "ml.r6i.large" | "ml.r6i.xlarge" | "ml.r6i.2xlarge" | "ml.r6i.4xlarge" | "ml.r6i.8xlarge" | "ml.r6i.12xlarge" | "ml.r6i.16xlarge" | "ml.r6i.24xlarge" | "ml.r6i.32xlarge" | "ml.m7i.large" | "ml.m7i.xlarge" | "ml.m7i.2xlarge" | "ml.m7i.4xlarge" | "ml.m7i.8xlarge" | "ml.m7i.12xlarge" | "ml.m7i.16xlarge" | "ml.m7i.24xlarge" | "ml.m7i.48xlarge" | "ml.c7i.large" | "ml.c7i.xlarge" | "ml.c7i.2xlarge" | "ml.c7i.4xlarge" | "ml.c7i.8xlarge" | "ml.c7i.12xlarge" | "ml.c7i.16xlarge" | "ml.c7i.24xlarge" | "ml.c7i.48xlarge" | "ml.r7i.large" | "ml.r7i.xlarge" | "ml.r7i.2xlarge" | "ml.r7i.4xlarge" | "ml.r7i.8xlarge" | "ml.r7i.12xlarge" | "ml.r7i.16xlarge" | "ml.r7i.24xlarge" | "ml.r7i.48xlarge" | "ml.g4dn.xlarge" | "ml.g4dn.2xlarge" | "ml.g4dn.4xlarge" | "ml.g4dn.8xlarge" | "ml.g4dn.12xlarge" | "ml.g4dn.16xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.trn1.2xlarge" | "ml.trn1.32xlarge" | "ml.inf2.xlarge" | "ml.inf2.8xlarge" | "ml.inf2.24xlarge" | "ml.inf2.48xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.12xlarge" | "ml.g6.16xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge")]?,

data/sig/types.rbs

@@ -235,6 +235,12 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class AssociationInfo
+      attr_accessor source_arn: ::String
+      attr_accessor destination_arn: ::String
+      SENSITIVE: []
+    end
+
     class AssociationSummary
       attr_accessor source_arn: ::String
       attr_accessor destination_arn: ::String
@@ -518,6 +524,13 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class BaseModel
+      attr_accessor hub_content_name: ::String
+      attr_accessor hub_content_version: ::String
+      attr_accessor recipe_name: ::String
+      SENSITIVE: []
+    end
+
     class BatchAddClusterNodesError
       attr_accessor instance_group_name: ::String
       attr_accessor error_code: ("InstanceGroupNotFound" | "InvalidInstanceGroupStatus")
@@ -601,6 +614,7 @@ module Aws::SageMaker
       attr_accessor inference_specification: Types::InferenceSpecification
       attr_accessor model_package_status: ("Pending" | "InProgress" | "Completed" | "Failed" | "Deleting")
       attr_accessor model_approval_status: ("Approved" | "Rejected" | "PendingManualApproval")
+      attr_accessor model_package_registration_type: ("Logged" | "Registered")
       SENSITIVE: []
     end
 
@@ -678,6 +692,26 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class BedrockCustomModelDeploymentMetadata
+      attr_accessor arn: ::String
+      SENSITIVE: []
+    end
+
+    class BedrockCustomModelMetadata
+      attr_accessor arn: ::String
+      SENSITIVE: []
+    end
+
+    class BedrockModelImportMetadata
+      attr_accessor arn: ::String
+      SENSITIVE: []
+    end
+
+    class BedrockProvisionedModelThroughputMetadata
+      attr_accessor arn: ::String
+      SENSITIVE: []
+    end
+
     class BestObjectiveNotImproving
       attr_accessor max_number_of_training_jobs_not_improving: ::Integer
       SENSITIVE: []
@@ -2066,6 +2100,7 @@ module Aws::SageMaker
       attr_accessor model_package_name: ::String
       attr_accessor model_package_group_name: ::String
       attr_accessor model_package_description: ::String
+      attr_accessor model_package_registration_type: ("Logged" | "Registered")
       attr_accessor inference_specification: Types::InferenceSpecification
       attr_accessor validation_specification: Types::ModelPackageValidationSpecification
       attr_accessor source_algorithm_specification: Types::SourceAlgorithmSpecification
@@ -2380,6 +2415,9 @@ module Aws::SageMaker
       attr_accessor remote_debug_config: Types::RemoteDebugConfig
       attr_accessor infra_check_config: Types::InfraCheckConfig
       attr_accessor session_chaining_config: Types::SessionChainingConfig
+      attr_accessor serverless_job_config: Types::ServerlessJobConfig
+      attr_accessor mlflow_config: Types::MlflowConfig
+      attr_accessor model_package_config: Types::ModelPackageConfig
       SENSITIVE: []
     end
 
@@ -2617,6 +2655,7 @@ module Aws::SageMaker
     class DataSource
       attr_accessor s3_data_source: Types::S3DataSource
       attr_accessor file_system_data_source: Types::FileSystemDataSource
+      attr_accessor dataset_source: Types::DatasetSource
       SENSITIVE: []
     end
 
@@ -2629,6 +2668,11 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class DatasetSource
+      attr_accessor dataset_arn: ::String
+      SENSITIVE: []
+    end
+
     class DebugHookConfig
       attr_accessor local_path: ::String
       attr_accessor s3_output_path: ::String
@@ -4143,6 +4187,7 @@ module Aws::SageMaker
       attr_accessor model_package_name: ::String
       attr_accessor model_package_group_name: ::String
       attr_accessor model_package_version: ::Integer
+      attr_accessor model_package_registration_type: ("Logged" | "Registered")
       attr_accessor model_package_arn: ::String
       attr_accessor model_package_description: ::String
       attr_accessor creation_time: ::Time
@@ -4346,6 +4391,7 @@ module Aws::SageMaker
       attr_accessor parallelism_configuration: Types::ParallelismConfiguration
       attr_accessor selective_execution_config: Types::SelectiveExecutionConfig
       attr_accessor pipeline_version_id: ::Integer
+      attr_accessor m_lflow_config: Types::MLflowConfiguration
       SENSITIVE: []
     end
 
@@ -4531,6 +4577,7 @@ module Aws::SageMaker
       attr_accessor checkpoint_config: Types::CheckpointConfig
       attr_accessor training_time_in_seconds: ::Integer
       attr_accessor billable_time_in_seconds: ::Integer
+      attr_accessor billable_token_count: ::Integer
       attr_accessor debug_hook_config: Types::DebugHookConfig
       attr_accessor experiment_config: Types::ExperimentConfig
       attr_accessor debug_rule_configurations: ::Array[Types::DebugRuleConfiguration]
@@ -4544,6 +4591,12 @@ module Aws::SageMaker
       attr_accessor retry_strategy: Types::RetryStrategy
       attr_accessor remote_debug_config: Types::RemoteDebugConfig
       attr_accessor infra_check_config: Types::InfraCheckConfig
+      attr_accessor serverless_job_config: Types::ServerlessJobConfig
+      attr_accessor mlflow_config: Types::MlflowConfig
+      attr_accessor model_package_config: Types::ModelPackageConfig
+      attr_accessor mlflow_details: Types::MlflowDetails
+      attr_accessor progress_info: Types::TrainingProgressInfo
+      attr_accessor output_model_package_arn: ::String
       SENSITIVE: []
     end
 
@@ -5843,6 +5896,11 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class InferenceComponentMetadata
+      attr_accessor arn: ::String
+      SENSITIVE: []
+    end
+
     class InferenceComponentRollingUpdatePolicy
       attr_accessor maximum_batch_size: Types::InferenceComponentCapacitySize
       attr_accessor wait_interval_in_seconds: ::Integer
@@ -6230,6 +6288,14 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class LineageMetadata
+      attr_accessor action_arns: ::Hash[::String, ::String]
+      attr_accessor artifact_arns: ::Hash[::String, ::String]
+      attr_accessor context_arns: ::Hash[::String, ::String]
+      attr_accessor associations: ::Array[Types::AssociationInfo]
+      SENSITIVE: []
+    end
+
     class ListActionsRequest
       attr_accessor source_uri: ::String
       attr_accessor action_type: ::String
@@ -7753,6 +7819,12 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class MLflowConfiguration
+      attr_accessor mlflow_resource_arn: ::String
+      attr_accessor mlflow_experiment_name: ::String
+      SENSITIVE: []
+    end
+
     class MemberDefinition
       attr_accessor cognito_member_definition: Types::CognitoMemberDefinition
       attr_accessor oidc_member_definition: Types::OidcMemberDefinition
@@ -7825,6 +7897,19 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class MlflowConfig
+      attr_accessor mlflow_resource_arn: ::String
+      attr_accessor mlflow_experiment_name: ::String
+      attr_accessor mlflow_run_name: ::String
+      SENSITIVE: []
+    end
+
+    class MlflowDetails
+      attr_accessor mlflow_experiment_id: ::String
+      attr_accessor mlflow_run_id: ::String
+      SENSITIVE: []
+    end
+
     class Model
       attr_accessor model_name: ::String
       attr_accessor primary_container: Types::ContainerDefinition
@@ -8109,6 +8194,7 @@ module Aws::SageMaker
       attr_accessor model_package_name: ::String
       attr_accessor model_package_group_name: ::String
       attr_accessor model_package_version: ::Integer
+      attr_accessor model_package_registration_type: ("Logged" | "Registered")
       attr_accessor model_package_arn: ::String
       attr_accessor model_package_description: ::String
       attr_accessor creation_time: ::Time
@@ -8140,6 +8226,12 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class ModelPackageConfig
+      attr_accessor model_package_group_arn: ::String
+      attr_accessor source_model_package_arn: ::String
+      SENSITIVE: []
+    end
+
     class ModelPackageContainerDefinition
       attr_accessor container_hostname: ::String
       attr_accessor image: ::String
@@ -8154,6 +8246,8 @@ module Aws::SageMaker
       attr_accessor nearest_model_name: ::String
       attr_accessor additional_s3_data_source: Types::AdditionalS3DataSource
       attr_accessor model_data_etag: ::String
+      attr_accessor is_checkpoint: bool
+      attr_accessor base_model: Types::BaseModel
       SENSITIVE: []
     end
 
@@ -8211,6 +8305,7 @@ module Aws::SageMaker
       attr_accessor model_package_status: ("Pending" | "InProgress" | "Completed" | "Failed" | "Deleting")
       attr_accessor model_approval_status: ("Approved" | "Rejected" | "PendingManualApproval")
       attr_accessor model_life_cycle: Types::ModelLifeCycle
+      attr_accessor model_package_registration_type: ("Logged" | "Registered")
       SENSITIVE: []
     end
 
@@ -8914,6 +9009,12 @@ module Aws::SageMaker
       attr_accessor auto_ml_job: Types::AutoMLJobStepMetadata
       attr_accessor endpoint: Types::EndpointStepMetadata
       attr_accessor endpoint_config: Types::EndpointConfigStepMetadata
+      attr_accessor bedrock_custom_model: Types::BedrockCustomModelMetadata
+      attr_accessor bedrock_custom_model_deployment: Types::BedrockCustomModelDeploymentMetadata
+      attr_accessor bedrock_provisioned_model_throughput: Types::BedrockProvisionedModelThroughputMetadata
+      attr_accessor bedrock_model_import: Types::BedrockModelImportMetadata
+      attr_accessor inference_component: Types::InferenceComponentMetadata
+      attr_accessor lineage: Types::LineageMetadata
       SENSITIVE: []
     end
 
@@ -9859,6 +9960,17 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class ServerlessJobConfig
+      attr_accessor base_model_arn: ::String
+      attr_accessor accept_eula: bool
+      attr_accessor job_type: ("FineTuning" | "Evaluation")
+      attr_accessor customization_technique: ("SFT" | "DPO" | "RLVR" | "RLAIF")
+      attr_accessor peft: ("LORA")
+      attr_accessor evaluation_type: ("LLMAJEvaluation" | "CustomScorerEvaluation" | "BenchmarkEvaluation")
+      attr_accessor evaluator_arn: ::String
+      SENSITIVE: []
+    end
+
     class ServiceCatalogProvisionedProductDetails
       attr_accessor provisioned_product_id: ::String
       attr_accessor provisioned_product_status_message: ::String
@@ -10049,6 +10161,7 @@ module Aws::SageMaker
       attr_accessor parallelism_configuration: Types::ParallelismConfiguration
       attr_accessor selective_execution_config: Types::SelectiveExecutionConfig
       attr_accessor pipeline_version_id: ::Integer
+      attr_accessor mlflow_experiment_name: ::String
       SENSITIVE: []
     end
 
@@ -10407,6 +10520,8 @@ module Aws::SageMaker
       attr_accessor debug_rule_configurations: ::Array[Types::DebugRuleConfiguration]
       attr_accessor tensor_board_output_config: Types::TensorBoardOutputConfig
       attr_accessor debug_rule_evaluation_statuses: ::Array[Types::DebugRuleEvaluationStatus]
+      attr_accessor output_model_package_arn: ::String
+      attr_accessor model_package_config: Types::ModelPackageConfig
       attr_accessor profiler_config: Types::ProfilerConfig
       attr_accessor environment: ::Hash[::String, ::String]
       attr_accessor retry_strategy: Types::RetryStrategy
@@ -10490,6 +10605,14 @@ module Aws::SageMaker
       SENSITIVE: []
     end
 
+    class TrainingProgressInfo
+      attr_accessor total_step_count_per_epoch: ::Integer
+      attr_accessor current_step: ::Integer
+      attr_accessor current_epoch: ::Integer
+      attr_accessor max_epoch: ::Integer
+      SENSITIVE: []
+    end
+
     class TrainingRepositoryAuthConfig
       attr_accessor training_repository_credentials_provider_arn: ::String
       SENSITIVE: []
@@ -11198,6 +11321,7 @@ module Aws::SageMaker
     class UpdateModelPackageInput
       attr_accessor model_package_arn: ::String
       attr_accessor model_approval_status: ("Approved" | "Rejected" | "PendingManualApproval")
+      attr_accessor model_package_registration_type: ("Logged" | "Registered")
       attr_accessor approval_description: ::String
       attr_accessor customer_metadata_properties: ::Hash[::String, ::String]
       attr_accessor customer_metadata_properties_to_remove: ::Array[::String]

metadata

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