feat(client-sagemaker): Added support for ml.trn1.32xlarge instance t…

…ype in Reserved Capacity Offering
aws · Jan 16, 2025 · c4798d7 · c4798d7
1 parent 22b9f01
commit c4798d7
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/clients/client-sagemaker/src/commands/DescribeTrainingPlanCommand.ts b/clients/client-sagemaker/src/commands/DescribeTrainingPlanCommand.ts
@@ -60,7 +60,7 @@ export interface DescribeTrainingPlanCommandOutput extends DescribeTrainingPlanR
  * //   ReservedCapacitySummaries: [ // ReservedCapacitySummaries
  * //     { // ReservedCapacitySummary
  * //       ReservedCapacityArn: "STRING_VALUE", // required
- * //       InstanceType: "ml.p4d.24xlarge" || "ml.p5.48xlarge" || "ml.p5e.48xlarge" || "ml.p5en.48xlarge" || "ml.trn2.48xlarge", // required
+ * //       InstanceType: "ml.p4d.24xlarge" || "ml.p5.48xlarge" || "ml.p5e.48xlarge" || "ml.p5en.48xlarge" || "ml.trn1.32xlarge" || "ml.trn2.48xlarge", // required
  * //       TotalInstanceCount: Number("int"), // required
  * //       Status: "Pending" || "Active" || "Scheduled" || "Expired" || "Failed", // required
  * //       AvailabilityZone: "STRING_VALUE",

diff --git a/clients/client-sagemaker/src/commands/ListTrainingPlansCommand.ts b/clients/client-sagemaker/src/commands/ListTrainingPlansCommand.ts
@@ -74,7 +74,7 @@ export interface ListTrainingPlansCommandOutput extends ListTrainingPlansRespons
  * //       ReservedCapacitySummaries: [ // ReservedCapacitySummaries
  * //         { // ReservedCapacitySummary
  * //           ReservedCapacityArn: "STRING_VALUE", // required
- * //           InstanceType: "ml.p4d.24xlarge" || "ml.p5.48xlarge" || "ml.p5e.48xlarge" || "ml.p5en.48xlarge" || "ml.trn2.48xlarge", // required
+ * //           InstanceType: "ml.p4d.24xlarge" || "ml.p5.48xlarge" || "ml.p5e.48xlarge" || "ml.p5en.48xlarge" || "ml.trn1.32xlarge" || "ml.trn2.48xlarge", // required
  * //           TotalInstanceCount: Number("int"), // required
  * //           Status: "Pending" || "Active" || "Scheduled" || "Expired" || "Failed", // required
  * //           AvailabilityZone: "STRING_VALUE",

diff --git a/clients/client-sagemaker/src/commands/SearchTrainingPlanOfferingsCommand.ts b/clients/client-sagemaker/src/commands/SearchTrainingPlanOfferingsCommand.ts
@@ -52,7 +52,7 @@ export interface SearchTrainingPlanOfferingsCommandOutput
  * // const { SageMakerClient, SearchTrainingPlanOfferingsCommand } = require("@aws-sdk/client-sagemaker"); // CommonJS import
  * const client = new SageMakerClient(config);
  * const input = { // SearchTrainingPlanOfferingsRequest
- *   InstanceType: "ml.p4d.24xlarge" || "ml.p5.48xlarge" || "ml.p5e.48xlarge" || "ml.p5en.48xlarge" || "ml.trn2.48xlarge", // required
+ *   InstanceType: "ml.p4d.24xlarge" || "ml.p5.48xlarge" || "ml.p5e.48xlarge" || "ml.p5en.48xlarge" || "ml.trn1.32xlarge" || "ml.trn2.48xlarge", // required
  *   InstanceCount: Number("int"), // required
  *   StartTimeAfter: new Date("TIMESTAMP"),
  *   EndTimeBefore: new Date("TIMESTAMP"),
@@ -78,7 +78,7 @@ export interface SearchTrainingPlanOfferingsCommandOutput
  * //       CurrencyCode: "STRING_VALUE",
  * //       ReservedCapacityOfferings: [ // ReservedCapacityOfferings
  * //         { // ReservedCapacityOffering
- * //           InstanceType: "ml.p4d.24xlarge" || "ml.p5.48xlarge" || "ml.p5e.48xlarge" || "ml.p5en.48xlarge" || "ml.trn2.48xlarge", // required
+ * //           InstanceType: "ml.p4d.24xlarge" || "ml.p5.48xlarge" || "ml.p5e.48xlarge" || "ml.p5en.48xlarge" || "ml.trn1.32xlarge" || "ml.trn2.48xlarge", // required
  * //           InstanceCount: Number("int"), // required
  * //           AvailabilityZone: "STRING_VALUE",
  * //           DurationHours: Number("long"),

diff --git a/clients/client-sagemaker/src/models/models_3.ts b/clients/client-sagemaker/src/models/models_3.ts
@@ -5719,6 +5719,7 @@ export const ReservedCapacityInstanceType = {
   ML_P5EN_48XLARGE: "ml.p5en.48xlarge",
   ML_P5E_48XLARGE: "ml.p5e.48xlarge",
   ML_P5_48XLARGE: "ml.p5.48xlarge",
+  ML_TRN1_32XLARGE: "ml.trn1.32xlarge",
   ML_TRN2_48XLARGE: "ml.trn2.48xlarge",
 } as const;