@aws-sdk/client-sagemaker 3.699.0 → 3.709.0

package/dist-es/commands/UpdateClusterSchedulerConfigCommand.js

@@ -0,0 +1,22 @@
+import { getEndpointPlugin } from "@smithy/middleware-endpoint";
+import { getSerdePlugin } from "@smithy/middleware-serde";
+import { Command as $Command } from "@smithy/smithy-client";
+import { commonParams } from "../endpoint/EndpointParameters";
+import { de_UpdateClusterSchedulerConfigCommand, se_UpdateClusterSchedulerConfigCommand, } from "../protocols/Aws_json1_1";
+export { $Command };
+export class UpdateClusterSchedulerConfigCommand extends $Command
+    .classBuilder()
+    .ep(commonParams)
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+        getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
+    ];
+})
+    .s("SageMaker", "UpdateClusterSchedulerConfig", {})
+    .n("SageMakerClient", "UpdateClusterSchedulerConfigCommand")
+    .f(void 0, void 0)
+    .ser(se_UpdateClusterSchedulerConfigCommand)
+    .de(de_UpdateClusterSchedulerConfigCommand)
+    .build() {
+}

package/dist-es/commands/UpdateComputeQuotaCommand.js

@@ -0,0 +1,22 @@
+import { getEndpointPlugin } from "@smithy/middleware-endpoint";
+import { getSerdePlugin } from "@smithy/middleware-serde";
+import { Command as $Command } from "@smithy/smithy-client";
+import { commonParams } from "../endpoint/EndpointParameters";
+import { de_UpdateComputeQuotaCommand, se_UpdateComputeQuotaCommand } from "../protocols/Aws_json1_1";
+export { $Command };
+export class UpdateComputeQuotaCommand extends $Command
+    .classBuilder()
+    .ep(commonParams)
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+        getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
+    ];
+})
+    .s("SageMaker", "UpdateComputeQuota", {})
+    .n("SageMakerClient", "UpdateComputeQuotaCommand")
+    .f(void 0, void 0)
+    .ser(se_UpdateComputeQuotaCommand)
+    .de(de_UpdateComputeQuotaCommand)
+    .build() {
+}

package/dist-es/commands/UpdateModelCardCommand.js

@@ -2,7 +2,7 @@ import { getEndpointPlugin } from "@smithy/middleware-endpoint";
 import { getSerdePlugin } from "@smithy/middleware-serde";
 import { Command as $Command } from "@smithy/smithy-client";
 import { commonParams } from "../endpoint/EndpointParameters";
-import { UpdateModelCardRequestFilterSensitiveLog, } from "../models/models_4";
+import { UpdateModelCardRequestFilterSensitiveLog, } from "../models/models_5";
 import { de_UpdateModelCardCommand, se_UpdateModelCardCommand } from "../protocols/Aws_json1_1";
 export { $Command };
 export class UpdateModelCardCommand extends $Command

package/dist-es/commands/UpdateModelPackageCommand.js

@@ -2,7 +2,7 @@ import { getEndpointPlugin } from "@smithy/middleware-endpoint";
 import { getSerdePlugin } from "@smithy/middleware-serde";
 import { Command as $Command } from "@smithy/smithy-client";
 import { commonParams } from "../endpoint/EndpointParameters";
-import { UpdateModelPackageInputFilterSensitiveLog, } from "../models/models_4";
+import { UpdateModelPackageInputFilterSensitiveLog, } from "../models/models_5";
 import { de_UpdateModelPackageCommand, se_UpdateModelPackageCommand } from "../protocols/Aws_json1_1";
 export { $Command };
 export class UpdateModelPackageCommand extends $Command

package/dist-es/commands/UpdatePartnerAppCommand.js

@@ -0,0 +1,22 @@
+import { getEndpointPlugin } from "@smithy/middleware-endpoint";
+import { getSerdePlugin } from "@smithy/middleware-serde";
+import { Command as $Command } from "@smithy/smithy-client";
+import { commonParams } from "../endpoint/EndpointParameters";
+import { de_UpdatePartnerAppCommand, se_UpdatePartnerAppCommand } from "../protocols/Aws_json1_1";
+export { $Command };
+export class UpdatePartnerAppCommand extends $Command
+    .classBuilder()
+    .ep(commonParams)
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+        getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
+    ];
+})
+    .s("SageMaker", "UpdatePartnerApp", {})
+    .n("SageMakerClient", "UpdatePartnerAppCommand")
+    .f(void 0, void 0)
+    .ser(se_UpdatePartnerAppCommand)
+    .de(de_UpdatePartnerAppCommand)
+    .build() {
+}

package/dist-es/commands/index.js

@@ -11,8 +11,10 @@ export * from "./CreateArtifactCommand";
 export * from "./CreateAutoMLJobCommand";
 export * from "./CreateAutoMLJobV2Command";
 export * from "./CreateClusterCommand";
+export * from "./CreateClusterSchedulerConfigCommand";
 export * from "./CreateCodeRepositoryCommand";
 export * from "./CreateCompilationJobCommand";
+export * from "./CreateComputeQuotaCommand";
 export * from "./CreateContextCommand";
 export * from "./CreateDataQualityJobDefinitionCommand";
 export * from "./CreateDeviceFleetCommand";
@@ -48,6 +50,8 @@ export * from "./CreateMonitoringScheduleCommand";
 export * from "./CreateNotebookInstanceCommand";
 export * from "./CreateNotebookInstanceLifecycleConfigCommand";
 export * from "./CreateOptimizationJobCommand";
+export * from "./CreatePartnerAppCommand";
+export * from "./CreatePartnerAppPresignedUrlCommand";
 export * from "./CreatePipelineCommand";
 export * from "./CreatePresignedDomainUrlCommand";
 export * from "./CreatePresignedMlflowTrackingServerUrlCommand";
@@ -57,6 +61,7 @@ export * from "./CreateProjectCommand";
 export * from "./CreateSpaceCommand";
 export * from "./CreateStudioLifecycleConfigCommand";
 export * from "./CreateTrainingJobCommand";
+export * from "./CreateTrainingPlanCommand";
 export * from "./CreateTransformJobCommand";
 export * from "./CreateTrialCommand";
 export * from "./CreateTrialComponentCommand";
@@ -70,8 +75,10 @@ export * from "./DeleteAppImageConfigCommand";
 export * from "./DeleteArtifactCommand";
 export * from "./DeleteAssociationCommand";
 export * from "./DeleteClusterCommand";
+export * from "./DeleteClusterSchedulerConfigCommand";
 export * from "./DeleteCodeRepositoryCommand";
 export * from "./DeleteCompilationJobCommand";
+export * from "./DeleteComputeQuotaCommand";
 export * from "./DeleteContextCommand";
 export * from "./DeleteDataQualityJobDefinitionCommand";
 export * from "./DeleteDeviceFleetCommand";
@@ -105,6 +112,7 @@ export * from "./DeleteMonitoringScheduleCommand";
 export * from "./DeleteNotebookInstanceCommand";
 export * from "./DeleteNotebookInstanceLifecycleConfigCommand";
 export * from "./DeleteOptimizationJobCommand";
+export * from "./DeletePartnerAppCommand";
 export * from "./DeletePipelineCommand";
 export * from "./DeleteProjectCommand";
 export * from "./DeleteSpaceCommand";
@@ -125,8 +133,10 @@ export * from "./DescribeAutoMLJobCommand";
 export * from "./DescribeAutoMLJobV2Command";
 export * from "./DescribeClusterCommand";
 export * from "./DescribeClusterNodeCommand";
+export * from "./DescribeClusterSchedulerConfigCommand";
 export * from "./DescribeCodeRepositoryCommand";
 export * from "./DescribeCompilationJobCommand";
+export * from "./DescribeComputeQuotaCommand";
 export * from "./DescribeContextCommand";
 export * from "./DescribeDataQualityJobDefinitionCommand";
 export * from "./DescribeDeviceCommand";
@@ -164,6 +174,7 @@ export * from "./DescribeMonitoringScheduleCommand";
 export * from "./DescribeNotebookInstanceCommand";
 export * from "./DescribeNotebookInstanceLifecycleConfigCommand";
 export * from "./DescribeOptimizationJobCommand";
+export * from "./DescribePartnerAppCommand";
 export * from "./DescribePipelineCommand";
 export * from "./DescribePipelineDefinitionForExecutionCommand";
 export * from "./DescribePipelineExecutionCommand";
@@ -173,6 +184,7 @@ export * from "./DescribeSpaceCommand";
 export * from "./DescribeStudioLifecycleConfigCommand";
 export * from "./DescribeSubscribedWorkteamCommand";
 export * from "./DescribeTrainingJobCommand";
+export * from "./DescribeTrainingPlanCommand";
 export * from "./DescribeTransformJobCommand";
 export * from "./DescribeTrialCommand";
 export * from "./DescribeTrialComponentCommand";
@@ -199,9 +211,11 @@ export * from "./ListAssociationsCommand";
 export * from "./ListAutoMLJobsCommand";
 export * from "./ListCandidatesForAutoMLJobCommand";
 export * from "./ListClusterNodesCommand";
+export * from "./ListClusterSchedulerConfigsCommand";
 export * from "./ListClustersCommand";
 export * from "./ListCodeRepositoriesCommand";
 export * from "./ListCompilationJobsCommand";
+export * from "./ListComputeQuotasCommand";
 export * from "./ListContextsCommand";
 export * from "./ListDataQualityJobDefinitionsCommand";
 export * from "./ListDeviceFleetsCommand";
@@ -246,6 +260,7 @@ export * from "./ListMonitoringSchedulesCommand";
 export * from "./ListNotebookInstanceLifecycleConfigsCommand";
 export * from "./ListNotebookInstancesCommand";
 export * from "./ListOptimizationJobsCommand";
+export * from "./ListPartnerAppsCommand";
 export * from "./ListPipelineExecutionStepsCommand";
 export * from "./ListPipelineExecutionsCommand";
 export * from "./ListPipelineParametersForExecutionCommand";
@@ -260,6 +275,7 @@ export * from "./ListSubscribedWorkteamsCommand";
 export * from "./ListTagsCommand";
 export * from "./ListTrainingJobsCommand";
 export * from "./ListTrainingJobsForHyperParameterTuningJobCommand";
+export * from "./ListTrainingPlansCommand";
 export * from "./ListTransformJobsCommand";
 export * from "./ListTrialComponentsCommand";
 export * from "./ListTrialsCommand";
@@ -272,6 +288,7 @@ export * from "./RegisterDevicesCommand";
 export * from "./RenderUiTemplateCommand";
 export * from "./RetryPipelineExecutionCommand";
 export * from "./SearchCommand";
+export * from "./SearchTrainingPlanOfferingsCommand";
 export * from "./SendPipelineExecutionStepFailureCommand";
 export * from "./SendPipelineExecutionStepSuccessCommand";
 export * from "./StartEdgeDeploymentStageCommand";
@@ -300,8 +317,10 @@ export * from "./UpdateActionCommand";
 export * from "./UpdateAppImageConfigCommand";
 export * from "./UpdateArtifactCommand";
 export * from "./UpdateClusterCommand";
+export * from "./UpdateClusterSchedulerConfigCommand";
 export * from "./UpdateClusterSoftwareCommand";
 export * from "./UpdateCodeRepositoryCommand";
+export * from "./UpdateComputeQuotaCommand";
 export * from "./UpdateContextCommand";
 export * from "./UpdateDeviceFleetCommand";
 export * from "./UpdateDevicesCommand";
@@ -324,6 +343,7 @@ export * from "./UpdateMonitoringAlertCommand";
 export * from "./UpdateMonitoringScheduleCommand";
 export * from "./UpdateNotebookInstanceCommand";
 export * from "./UpdateNotebookInstanceLifecycleConfigCommand";
+export * from "./UpdatePartnerAppCommand";
 export * from "./UpdatePipelineCommand";
 export * from "./UpdatePipelineExecutionCommand";
 export * from "./UpdateProjectCommand";

package/dist-es/models/models_0.js

@@ -7,6 +7,10 @@ export const ActionStatus = {
     STOPPING: "Stopping",
     UNKNOWN: "Unknown",
 };
+export const ActivationState = {
+    DISABLED: "Disabled",
+    ENABLED: "Enabled",
+};
 export const AssociationEdgeType = {
     ASSOCIATED_WITH: "AssociatedWith",
     CONTRIBUTED_TO: "ContributedTo",
@@ -511,6 +515,7 @@ export const TrainingInstanceType = {
     ML_P3_8XLARGE: "ml.p3.8xlarge",
     ML_P4DE_24XLARGE: "ml.p4de.24xlarge",
     ML_P4D_24XLARGE: "ml.p4d.24xlarge",
+    ML_P5EN_48XLARGE: "ml.p5en.48xlarge",
     ML_P5E_48XLARGE: "ml.p5e.48xlarge",
     ML_P5_48XLARGE: "ml.p5.48xlarge",
     ML_R5D_12XLARGE: "ml.r5d.12xlarge",
@@ -536,6 +541,7 @@ export const TrainingInstanceType = {
     ML_TRN1N_32XLARGE: "ml.trn1n.32xlarge",
     ML_TRN1_2XLARGE: "ml.trn1.2xlarge",
     ML_TRN1_32XLARGE: "ml.trn1.32xlarge",
+    ML_TRN2_48XLARGE: "ml.trn2.48xlarge",
 };
 export const BatchStrategy = {
     MULTI_RECORD: "MultiRecord",
@@ -1173,6 +1179,7 @@ export const ClusterInstanceType = {
     ML_M5_XLARGE: "ml.m5.xlarge",
     ML_P4DE_24XLARGE: "ml.p4de.24xlarge",
     ML_P4D_24XLARGE: "ml.p4d.24xlarge",
+    ML_P5EN_48XLARGE: "ml.p5en.48xlarge",
     ML_P5E_48XLARGE: "ml.p5e.48xlarge",
     ML_P5_48XLARGE: "ml.p5.48xlarge",
     ML_T3_2XLARGE: "ml.t3.2xlarge",
@@ -1181,11 +1188,21 @@ export const ClusterInstanceType = {
     ML_T3_XLARGE: "ml.t3.xlarge",
     ML_TRN1N_32XLARGE: "ml.trn1n.32xlarge",
     ML_TRN1_32XLARGE: "ml.trn1.32xlarge",
+    ML_TRN2_48XLARGE: "ml.trn2.48xlarge",
 };
 export const DeepHealthCheckType = {
     INSTANCE_CONNECTIVITY: "InstanceConnectivity",
     INSTANCE_STRESS: "InstanceStress",
 };
+export const InstanceGroupStatus = {
+    CREATING: "Creating",
+    DEGRADED: "Degraded",
+    DELETING: "Deleting",
+    FAILED: "Failed",
+    INSERVICE: "InService",
+    SYSTEMUPDATING: "SystemUpdating",
+    UPDATING: "Updating",
+};
 export const ClusterInstanceStatus = {
     DEEP_HEALTH_CHECK_IN_PROGRESS: "DeepHealthCheckInProgress",
     FAILURE: "Failure",
@@ -1198,6 +1215,20 @@ export const ClusterNodeRecovery = {
     AUTOMATIC: "Automatic",
     NONE: "None",
 };
+export const SchedulerResourceStatus = {
+    CREATED: "Created",
+    CREATE_FAILED: "CreateFailed",
+    CREATE_ROLLBACK_FAILED: "CreateRollbackFailed",
+    CREATING: "Creating",
+    DELETED: "Deleted",
+    DELETE_FAILED: "DeleteFailed",
+    DELETE_ROLLBACK_FAILED: "DeleteRollbackFailed",
+    DELETING: "Deleting",
+    UPDATED: "Updated",
+    UPDATE_FAILED: "UpdateFailed",
+    UPDATE_ROLLBACK_FAILED: "UpdateRollbackFailed",
+    UPDATING: "Updating",
+};
 export const ClusterSortBy = {
     CREATION_TIME: "CREATION_TIME",
     NAME: "NAME",
@@ -1301,6 +1332,15 @@ export const CompleteOnConvergence = {
     DISABLED: "Disabled",
     ENABLED: "Enabled",
 };
+export const PreemptTeamTasks = {
+    LOWERPRIORITY: "LowerPriority",
+    NEVER: "Never",
+};
+export const ResourceSharingStrategy = {
+    DONTLEND: "DontLend",
+    LEND: "Lend",
+    LENDANDBORROW: "LendAndBorrow",
+};
 export const ConditionOutcome = {
     FALSE: "False",
     TRUE: "True",
@@ -1340,13 +1380,3 @@ export const HyperParameterScalingType = {
     LOGARITHMIC: "Logarithmic",
     REVERSE_LOGARITHMIC: "ReverseLogarithmic",
 };
-export const ParameterType = {
-    CATEGORICAL: "Categorical",
-    CONTINUOUS: "Continuous",
-    FREE_TEXT: "FreeText",
-    INTEGER: "Integer",
-};
-export const HyperParameterTuningJobObjectiveType = {
-    MAXIMIZE: "Maximize",
-    MINIMIZE: "Minimize",
-};

package/dist-es/models/models_1.js

@@ -1,5 +1,15 @@
 import { SENSITIVE_STRING, } from "@smithy/smithy-client";
 import { SageMakerServiceException as __BaseException } from "./SageMakerServiceException";
+export const ParameterType = {
+    CATEGORICAL: "Categorical",
+    CONTINUOUS: "Continuous",
+    FREE_TEXT: "FreeText",
+    INTEGER: "Integer",
+};
+export const HyperParameterTuningJobObjectiveType = {
+    MAXIMIZE: "Maximize",
+    MINIMIZE: "Minimize",
+};
 export class ResourceInUse extends __BaseException {
     constructor(opts) {
         super({
@@ -13,6 +23,10 @@ export class ResourceInUse extends __BaseException {
         this.Message = opts.Message;
     }
 }
+export const FairShare = {
+    DISABLED: "Disabled",
+    ENABLED: "Enabled",
+};
 export const Framework = {
     DARKNET: "DARKNET",
     KERAS: "KERAS",
@@ -98,6 +112,8 @@ export var CustomFileSystemConfig;
     CustomFileSystemConfig.visit = (value, visitor) => {
         if (value.EFSFileSystemConfig !== undefined)
             return visitor.EFSFileSystemConfig(value.EFSFileSystemConfig);
+        if (value.FSxLustreFileSystemConfig !== undefined)
+            return visitor.FSxLustreFileSystemConfig(value.FSxLustreFileSystemConfig);
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(CustomFileSystemConfig || (CustomFileSystemConfig = {}));
@@ -119,14 +135,19 @@ export const StudioWebPortal = {
 };
 export const MlTools = {
     AUTO_ML: "AutoMl",
+    COMET: "Comet",
     DATA_WRANGLER: "DataWrangler",
+    DEEPCHECKS_LLM_EVALUATION: "DeepchecksLLMEvaluation",
     EMR_CLUSTERS: "EmrClusters",
     ENDPOINTS: "Endpoints",
     EXPERIMENTS: "Experiments",
     FEATURE_STORE: "FeatureStore",
+    FIDDLER: "Fiddler",
+    HYPER_POD_CLUSTERS: "HyperPodClusters",
     INFERENCE_OPTIMIZATION: "InferenceOptimization",
     INFERENCE_RECOMMENDER: "InferenceRecommender",
     JUMP_START: "JumpStart",
+    LAKERA_GUARD: "LakeraGuard",
     MODELS: "Models",
     MODEL_EVALUATION: "ModelEvaluation",
     PERFORMANCE_EVALUATION: "PerformanceEvaluation",
@@ -456,62 +477,6 @@ export const MonitoringType = {
     MODEL_EXPLAINABILITY: "ModelExplainability",
     MODEL_QUALITY: "ModelQuality",
 };
-export const NotebookInstanceAcceleratorType = {
-    ML_EIA1_LARGE: "ml.eia1.large",
-    ML_EIA1_MEDIUM: "ml.eia1.medium",
-    ML_EIA1_XLARGE: "ml.eia1.xlarge",
-    ML_EIA2_LARGE: "ml.eia2.large",
-    ML_EIA2_MEDIUM: "ml.eia2.medium",
-    ML_EIA2_XLARGE: "ml.eia2.xlarge",
-};
-export const DirectInternetAccess = {
-    DISABLED: "Disabled",
-    ENABLED: "Enabled",
-};
-export const RootAccess = {
-    DISABLED: "Disabled",
-    ENABLED: "Enabled",
-};
-export const OptimizationJobDeploymentInstanceType = {
-    ML_G5_12XLARGE: "ml.g5.12xlarge",
-    ML_G5_16XLARGE: "ml.g5.16xlarge",
-    ML_G5_24XLARGE: "ml.g5.24xlarge",
-    ML_G5_2XLARGE: "ml.g5.2xlarge",
-    ML_G5_48XLARGE: "ml.g5.48xlarge",
-    ML_G5_4XLARGE: "ml.g5.4xlarge",
-    ML_G5_8XLARGE: "ml.g5.8xlarge",
-    ML_G5_XLARGE: "ml.g5.xlarge",
-    ML_G6_12XLARGE: "ml.g6.12xlarge",
-    ML_G6_16XLARGE: "ml.g6.16xlarge",
-    ML_G6_24XLARGE: "ml.g6.24xlarge",
-    ML_G6_2XLARGE: "ml.g6.2xlarge",
-    ML_G6_48XLARGE: "ml.g6.48xlarge",
-    ML_G6_4XLARGE: "ml.g6.4xlarge",
-    ML_G6_8XLARGE: "ml.g6.8xlarge",
-    ML_G6_XLARGE: "ml.g6.xlarge",
-    ML_INF2_24XLARGE: "ml.inf2.24xlarge",
-    ML_INF2_48XLARGE: "ml.inf2.48xlarge",
-    ML_INF2_8XLARGE: "ml.inf2.8xlarge",
-    ML_INF2_XLARGE: "ml.inf2.xlarge",
-    ML_P4DE_24XLARGE: "ml.p4de.24xlarge",
-    ML_P4D_24XLARGE: "ml.p4d.24xlarge",
-    ML_P5_48XLARGE: "ml.p5.48xlarge",
-    ML_TRN1N_32XLARGE: "ml.trn1n.32xlarge",
-    ML_TRN1_2XLARGE: "ml.trn1.2xlarge",
-    ML_TRN1_32XLARGE: "ml.trn1.32xlarge",
-};
-export var OptimizationConfig;
-(function (OptimizationConfig) {
-    OptimizationConfig.visit = (value, visitor) => {
-        if (value.ModelQuantizationConfig !== undefined)
-            return visitor.ModelQuantizationConfig(value.ModelQuantizationConfig);
-        if (value.ModelCompilationConfig !== undefined)
-            return visitor.ModelCompilationConfig(value.ModelCompilationConfig);
-        if (value.ModelShardingConfig !== undefined)
-            return visitor.ModelShardingConfig(value.ModelShardingConfig);
-        return visitor._(value.$unknown[0], value.$unknown[1]);
-    };
-})(OptimizationConfig || (OptimizationConfig = {}));
 export const CreateModelCardRequestFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.Content && { Content: SENSITIVE_STRING }),

package/dist-es/models/models_2.js

@@ -1,4 +1,69 @@
 import { SENSITIVE_STRING } from "@smithy/smithy-client";
+export const NotebookInstanceAcceleratorType = {
+    ML_EIA1_LARGE: "ml.eia1.large",
+    ML_EIA1_MEDIUM: "ml.eia1.medium",
+    ML_EIA1_XLARGE: "ml.eia1.xlarge",
+    ML_EIA2_LARGE: "ml.eia2.large",
+    ML_EIA2_MEDIUM: "ml.eia2.medium",
+    ML_EIA2_XLARGE: "ml.eia2.xlarge",
+};
+export const DirectInternetAccess = {
+    DISABLED: "Disabled",
+    ENABLED: "Enabled",
+};
+export const RootAccess = {
+    DISABLED: "Disabled",
+    ENABLED: "Enabled",
+};
+export const OptimizationJobDeploymentInstanceType = {
+    ML_G5_12XLARGE: "ml.g5.12xlarge",
+    ML_G5_16XLARGE: "ml.g5.16xlarge",
+    ML_G5_24XLARGE: "ml.g5.24xlarge",
+    ML_G5_2XLARGE: "ml.g5.2xlarge",
+    ML_G5_48XLARGE: "ml.g5.48xlarge",
+    ML_G5_4XLARGE: "ml.g5.4xlarge",
+    ML_G5_8XLARGE: "ml.g5.8xlarge",
+    ML_G5_XLARGE: "ml.g5.xlarge",
+    ML_G6_12XLARGE: "ml.g6.12xlarge",
+    ML_G6_16XLARGE: "ml.g6.16xlarge",
+    ML_G6_24XLARGE: "ml.g6.24xlarge",
+    ML_G6_2XLARGE: "ml.g6.2xlarge",
+    ML_G6_48XLARGE: "ml.g6.48xlarge",
+    ML_G6_4XLARGE: "ml.g6.4xlarge",
+    ML_G6_8XLARGE: "ml.g6.8xlarge",
+    ML_G6_XLARGE: "ml.g6.xlarge",
+    ML_INF2_24XLARGE: "ml.inf2.24xlarge",
+    ML_INF2_48XLARGE: "ml.inf2.48xlarge",
+    ML_INF2_8XLARGE: "ml.inf2.8xlarge",
+    ML_INF2_XLARGE: "ml.inf2.xlarge",
+    ML_P4DE_24XLARGE: "ml.p4de.24xlarge",
+    ML_P4D_24XLARGE: "ml.p4d.24xlarge",
+    ML_P5_48XLARGE: "ml.p5.48xlarge",
+    ML_TRN1N_32XLARGE: "ml.trn1n.32xlarge",
+    ML_TRN1_2XLARGE: "ml.trn1.2xlarge",
+    ML_TRN1_32XLARGE: "ml.trn1.32xlarge",
+};
+export var OptimizationConfig;
+(function (OptimizationConfig) {
+    OptimizationConfig.visit = (value, visitor) => {
+        if (value.ModelQuantizationConfig !== undefined)
+            return visitor.ModelQuantizationConfig(value.ModelQuantizationConfig);
+        if (value.ModelCompilationConfig !== undefined)
+            return visitor.ModelCompilationConfig(value.ModelCompilationConfig);
+        if (value.ModelShardingConfig !== undefined)
+            return visitor.ModelShardingConfig(value.ModelShardingConfig);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(OptimizationConfig || (OptimizationConfig = {}));
+export const PartnerAppAuthType = {
+    IAM: "IAM",
+};
+export const PartnerAppType = {
+    COMET: "comet",
+    DEEPCHECKS_LLM_EVALUATION: "deepchecks-llm-evaluation",
+    FIDDLER: "fiddler",
+    LAKERA_GUARD: "lakera-guard",
+};
 export const DataDistributionType = {
     FULLYREPLICATED: "FullyReplicated",
     SHARDEDBYS3KEY: "ShardedByS3Key",
@@ -31,6 +96,8 @@ export var CustomFileSystem;
     CustomFileSystem.visit = (value, visitor) => {
         if (value.EFSFileSystem !== undefined)
             return visitor.EFSFileSystem(value.EFSFileSystem);
+        if (value.FSxLustreFileSystem !== undefined)
+            return visitor.FSxLustreFileSystem(value.FSxLustreFileSystem);
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(CustomFileSystem || (CustomFileSystem = {}));
@@ -199,62 +266,6 @@ export const HumanTaskUiStatus = {
     ACTIVE: "Active",
     DELETING: "Deleting",
 };
-export const TrainingJobStatus = {
-    COMPLETED: "Completed",
-    FAILED: "Failed",
-    IN_PROGRESS: "InProgress",
-    STOPPED: "Stopped",
-    STOPPING: "Stopping",
-};
-export const HyperParameterTuningJobStatus = {
-    COMPLETED: "Completed",
-    DELETE_FAILED: "DeleteFailed",
-    DELETING: "Deleting",
-    FAILED: "Failed",
-    IN_PROGRESS: "InProgress",
-    STOPPED: "Stopped",
-    STOPPING: "Stopping",
-};
-export const ImageStatus = {
-    CREATED: "CREATED",
-    CREATE_FAILED: "CREATE_FAILED",
-    CREATING: "CREATING",
-    DELETE_FAILED: "DELETE_FAILED",
-    DELETING: "DELETING",
-    UPDATE_FAILED: "UPDATE_FAILED",
-    UPDATING: "UPDATING",
-};
-export const ImageVersionStatus = {
-    CREATED: "CREATED",
-    CREATE_FAILED: "CREATE_FAILED",
-    CREATING: "CREATING",
-    DELETE_FAILED: "DELETE_FAILED",
-    DELETING: "DELETING",
-};
-export const InferenceComponentStatus = {
-    CREATING: "Creating",
-    DELETING: "Deleting",
-    FAILED: "Failed",
-    IN_SERVICE: "InService",
-    UPDATING: "Updating",
-};
-export const ModelVariantStatus = {
-    CREATING: "Creating",
-    DELETED: "Deleted",
-    DELETING: "Deleting",
-    IN_SERVICE: "InService",
-    UPDATING: "Updating",
-};
-export const InferenceExperimentStatus = {
-    CANCELLED: "Cancelled",
-    COMPLETED: "Completed",
-    CREATED: "Created",
-    CREATING: "Creating",
-    RUNNING: "Running",
-    STARTING: "Starting",
-    STOPPING: "Stopping",
-    UPDATING: "Updating",
-};
 export const OidcConfigFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.ClientSecret && { ClientSecret: SENSITIVE_STRING }),