Skip to content

Commit c908352

Browse files
author
awstools
committed
feat(client-sagemaker): This release does the following: 1.) Adds DurationHours as a required field to the SearchTrainingPlanOfferings action in the SageMaker AI API; 2.) Adds support for G6e instance types for SageMaker AI inference optimization jobs.
1 parent 8aac7ca commit c908352

File tree

7 files changed

+64
-6
lines changed

7 files changed

+64
-6
lines changed

‎clients/client-sagemaker/src/commands/CreateOptimizationJobCommand.ts‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ export interface CreateOptimizationJobCommandOutput extends CreateOptimizationJo
5151
* },
5252
* },
5353
* },
54-
* DeploymentInstanceType: "ml.p4d.24xlarge" || "ml.p4de.24xlarge" || "ml.p5.48xlarge" || "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.g6.xlarge" || "ml.g6.2xlarge" || "ml.g6.4xlarge" || "ml.g6.8xlarge" || "ml.g6.12xlarge" || "ml.g6.16xlarge" || "ml.g6.24xlarge" || "ml.g6.48xlarge" || "ml.inf2.xlarge" || "ml.inf2.8xlarge" || "ml.inf2.24xlarge" || "ml.inf2.48xlarge" || "ml.trn1.2xlarge" || "ml.trn1.32xlarge" || "ml.trn1n.32xlarge", // required
54+
* DeploymentInstanceType: "ml.p4d.24xlarge" || "ml.p4de.24xlarge" || "ml.p5.48xlarge" || "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.g6.xlarge" || "ml.g6.2xlarge" || "ml.g6.4xlarge" || "ml.g6.8xlarge" || "ml.g6.12xlarge" || "ml.g6.16xlarge" || "ml.g6.24xlarge" || "ml.g6.48xlarge" || "ml.g6e.xlarge" || "ml.g6e.2xlarge" || "ml.g6e.4xlarge" || "ml.g6e.8xlarge" || "ml.g6e.12xlarge" || "ml.g6e.16xlarge" || "ml.g6e.24xlarge" || "ml.g6e.48xlarge" || "ml.inf2.xlarge" || "ml.inf2.8xlarge" || "ml.inf2.24xlarge" || "ml.inf2.48xlarge" || "ml.trn1.2xlarge" || "ml.trn1.32xlarge" || "ml.trn1n.32xlarge", // required
5555
* OptimizationEnvironment: { // OptimizationJobEnvironmentVariables
5656
* "<keys>": "STRING_VALUE",
5757
* },

‎clients/client-sagemaker/src/commands/DescribeOptimizationJobCommand.ts‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ export interface DescribeOptimizationJobCommandOutput extends DescribeOptimizati
6060
* // OptimizationEnvironment: { // OptimizationJobEnvironmentVariables
6161
* // "<keys>": "STRING_VALUE",
6262
* // },
63-
* // DeploymentInstanceType: "ml.p4d.24xlarge" || "ml.p4de.24xlarge" || "ml.p5.48xlarge" || "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.g6.xlarge" || "ml.g6.2xlarge" || "ml.g6.4xlarge" || "ml.g6.8xlarge" || "ml.g6.12xlarge" || "ml.g6.16xlarge" || "ml.g6.24xlarge" || "ml.g6.48xlarge" || "ml.inf2.xlarge" || "ml.inf2.8xlarge" || "ml.inf2.24xlarge" || "ml.inf2.48xlarge" || "ml.trn1.2xlarge" || "ml.trn1.32xlarge" || "ml.trn1n.32xlarge", // required
63+
* // DeploymentInstanceType: "ml.p4d.24xlarge" || "ml.p4de.24xlarge" || "ml.p5.48xlarge" || "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.g6.xlarge" || "ml.g6.2xlarge" || "ml.g6.4xlarge" || "ml.g6.8xlarge" || "ml.g6.12xlarge" || "ml.g6.16xlarge" || "ml.g6.24xlarge" || "ml.g6.48xlarge" || "ml.g6e.xlarge" || "ml.g6e.2xlarge" || "ml.g6e.4xlarge" || "ml.g6e.8xlarge" || "ml.g6e.12xlarge" || "ml.g6e.16xlarge" || "ml.g6e.24xlarge" || "ml.g6e.48xlarge" || "ml.inf2.xlarge" || "ml.inf2.8xlarge" || "ml.inf2.24xlarge" || "ml.inf2.48xlarge" || "ml.trn1.2xlarge" || "ml.trn1.32xlarge" || "ml.trn1n.32xlarge", // required
6464
* // OptimizationConfigs: [ // OptimizationConfigs // required
6565
* // { // OptimizationConfig Union: only one key present
6666
* // ModelQuantizationConfig: { // ModelQuantizationConfig

‎clients/client-sagemaker/src/commands/ListOptimizationJobsCommand.ts‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ export interface ListOptimizationJobsCommandOutput extends ListOptimizationJobsR
6060
* // OptimizationStartTime: new Date("TIMESTAMP"),
6161
* // OptimizationEndTime: new Date("TIMESTAMP"),
6262
* // LastModifiedTime: new Date("TIMESTAMP"),
63-
* // DeploymentInstanceType: "ml.p4d.24xlarge" || "ml.p4de.24xlarge" || "ml.p5.48xlarge" || "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.g6.xlarge" || "ml.g6.2xlarge" || "ml.g6.4xlarge" || "ml.g6.8xlarge" || "ml.g6.12xlarge" || "ml.g6.16xlarge" || "ml.g6.24xlarge" || "ml.g6.48xlarge" || "ml.inf2.xlarge" || "ml.inf2.8xlarge" || "ml.inf2.24xlarge" || "ml.inf2.48xlarge" || "ml.trn1.2xlarge" || "ml.trn1.32xlarge" || "ml.trn1n.32xlarge", // required
63+
* // DeploymentInstanceType: "ml.p4d.24xlarge" || "ml.p4de.24xlarge" || "ml.p5.48xlarge" || "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.g6.xlarge" || "ml.g6.2xlarge" || "ml.g6.4xlarge" || "ml.g6.8xlarge" || "ml.g6.12xlarge" || "ml.g6.16xlarge" || "ml.g6.24xlarge" || "ml.g6.48xlarge" || "ml.g6e.xlarge" || "ml.g6e.2xlarge" || "ml.g6e.4xlarge" || "ml.g6e.8xlarge" || "ml.g6e.12xlarge" || "ml.g6e.16xlarge" || "ml.g6e.24xlarge" || "ml.g6e.48xlarge" || "ml.inf2.xlarge" || "ml.inf2.8xlarge" || "ml.inf2.24xlarge" || "ml.inf2.48xlarge" || "ml.trn1.2xlarge" || "ml.trn1.32xlarge" || "ml.trn1n.32xlarge", // required
6464
* // OptimizationTypes: [ // OptimizationTypes // required
6565
* // "STRING_VALUE",
6666
* // ],

‎clients/client-sagemaker/src/commands/SearchTrainingPlanOfferingsCommand.ts‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ export interface SearchTrainingPlanOfferingsCommandOutput
5656
* InstanceCount: Number("int"), // required
5757
* StartTimeAfter: new Date("TIMESTAMP"),
5858
* EndTimeBefore: new Date("TIMESTAMP"),
59-
* DurationHours: Number("long"),
59+
* DurationHours: Number("long"), // required
6060
* TargetResources: [ // SageMakerResourceNames // required
6161
* "training-job" || "hyperpod-cluster",
6262
* ],

‎clients/client-sagemaker/src/models/models_2.ts‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -429,6 +429,14 @@ export const OptimizationJobDeploymentInstanceType = {
429429
ML_G5_4XLARGE: "ml.g5.4xlarge",
430430
ML_G5_8XLARGE: "ml.g5.8xlarge",
431431
ML_G5_XLARGE: "ml.g5.xlarge",
432+
ML_G6E_12XLARGE: "ml.g6e.12xlarge",
433+
ML_G6E_16XLARGE: "ml.g6e.16xlarge",
434+
ML_G6E_24XLARGE: "ml.g6e.24xlarge",
435+
ML_G6E_2XLARGE: "ml.g6e.2xlarge",
436+
ML_G6E_48XLARGE: "ml.g6e.48xlarge",
437+
ML_G6E_4XLARGE: "ml.g6e.4xlarge",
438+
ML_G6E_8XLARGE: "ml.g6e.8xlarge",
439+
ML_G6E_XLARGE: "ml.g6e.xlarge",
432440
ML_G6_12XLARGE: "ml.g6.12xlarge",
433441
ML_G6_16XLARGE: "ml.g6.16xlarge",
434442
ML_G6_24XLARGE: "ml.g6.24xlarge",

‎clients/client-sagemaker/src/models/models_4.ts‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10637,7 +10637,7 @@ export interface SearchTrainingPlanOfferingsRequest {
1063710637
* <p>The desired duration in hours for the training plan offerings.</p>
1063810638
* @public
1063910639
*/
10640-
DurationHours?: number | undefined;
10640+
DurationHours: number | undefined;
1064110641

1064210642
/**
1064310643
* <p>The target resources (e.g., SageMaker Training Jobs, SageMaker HyperPod) to search for in the

‎codegen/sdk-codegen/aws-models/sagemaker.json‎

Lines changed: 51 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52714,6 +52714,54 @@
5271452714
"smithy.api#enumValue": "ml.g6.48xlarge"
5271552715
}
5271652716
},
52717+
"ML_G6E_XLARGE": {
52718+
"target": "smithy.api#Unit",
52719+
"traits": {
52720+
"smithy.api#enumValue": "ml.g6e.xlarge"
52721+
}
52722+
},
52723+
"ML_G6E_2XLARGE": {
52724+
"target": "smithy.api#Unit",
52725+
"traits": {
52726+
"smithy.api#enumValue": "ml.g6e.2xlarge"
52727+
}
52728+
},
52729+
"ML_G6E_4XLARGE": {
52730+
"target": "smithy.api#Unit",
52731+
"traits": {
52732+
"smithy.api#enumValue": "ml.g6e.4xlarge"
52733+
}
52734+
},
52735+
"ML_G6E_8XLARGE": {
52736+
"target": "smithy.api#Unit",
52737+
"traits": {
52738+
"smithy.api#enumValue": "ml.g6e.8xlarge"
52739+
}
52740+
},
52741+
"ML_G6E_12XLARGE": {
52742+
"target": "smithy.api#Unit",
52743+
"traits": {
52744+
"smithy.api#enumValue": "ml.g6e.12xlarge"
52745+
}
52746+
},
52747+
"ML_G6E_16XLARGE": {
52748+
"target": "smithy.api#Unit",
52749+
"traits": {
52750+
"smithy.api#enumValue": "ml.g6e.16xlarge"
52751+
}
52752+
},
52753+
"ML_G6E_24XLARGE": {
52754+
"target": "smithy.api#Unit",
52755+
"traits": {
52756+
"smithy.api#enumValue": "ml.g6e.24xlarge"
52757+
}
52758+
},
52759+
"ML_G6E_48XLARGE": {
52760+
"target": "smithy.api#Unit",
52761+
"traits": {
52762+
"smithy.api#enumValue": "ml.g6e.48xlarge"
52763+
}
52764+
},
5271752765
"ML_INF2_XLARGE": {
5271852766
"target": "smithy.api#Unit",
5271952767
"traits": {
@@ -64029,7 +64077,9 @@
6402964077
"DurationHours": {
6403064078
"target": "com.amazonaws.sagemaker#TrainingPlanDurationHoursInput",
6403164079
"traits": {
64032-
"smithy.api#documentation": "<p>The desired duration in hours for the training plan offerings.</p>"
64080+
"smithy.api#clientOptional": {},
64081+
"smithy.api#documentation": "<p>The desired duration in hours for the training plan offerings.</p>",
64082+
"smithy.api#required": {}
6403364083
}
6403464084
},
6403564085
"TargetResources": {

0 commit comments

Comments
 (0)