feat(client-sagemaker): Adds Model Card information as a new componen…

…t to Model Package. Autopilot launches algorithm selection for TimeSeries modality to generate AutoML candidates per algorithm.
aws · May 30, 2024 · 393afc7 · 393afc7
1 parent c2f93e1
commit 393afc7
Show file tree

Hide file tree

Showing 19 changed files with 921 additions and 454 deletions.
diff --git a/clients/client-sagemaker/src/commands/CreateAutoMLJobCommand.ts b/clients/client-sagemaker/src/commands/CreateAutoMLJobCommand.ts
@@ -93,7 +93,7 @@ export interface CreateAutoMLJobCommandOutput extends CreateAutoMLJobResponse, _
  *       AlgorithmsConfig: [ // AutoMLAlgorithmsConfig
  *         { // AutoMLAlgorithmConfig
  *           AutoMLAlgorithms: [ // AutoMLAlgorithms // required
- *             "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai",
+ *             "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
  *           ],
  *         },
  *       ],

diff --git a/clients/client-sagemaker/src/commands/CreateAutoMLJobV2Command.ts b/clients/client-sagemaker/src/commands/CreateAutoMLJobV2Command.ts
@@ -120,13 +120,22 @@ export interface CreateAutoMLJobV2CommandOutput extends CreateAutoMLJobV2Respons
  *           CountryCode: "STRING_VALUE",
  *         },
  *       ],
- *     },
- *     TabularJobConfig: { // TabularJobConfig
  *       CandidateGenerationConfig: { // CandidateGenerationConfig
  *         AlgorithmsConfig: [ // AutoMLAlgorithmsConfig
  *           { // AutoMLAlgorithmConfig
  *             AutoMLAlgorithms: [ // AutoMLAlgorithms // required
- *               "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai",
+ *               "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
+ *             ],
+ *           },
+ *         ],
+ *       },
+ *     },
+ *     TabularJobConfig: { // TabularJobConfig
+ *       CandidateGenerationConfig: {
+ *         AlgorithmsConfig: [
+ *           {
+ *             AutoMLAlgorithms: [ // required
+ *               "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
  *             ],
  *           },
  *         ],

diff --git a/clients/client-sagemaker/src/commands/CreateModelPackageCommand.ts b/clients/client-sagemaker/src/commands/CreateModelPackageCommand.ts
@@ -5,7 +5,11 @@ import { Command as $Command } from "@smithy/smithy-client";
 import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { commonParams } from "../endpoint/EndpointParameters";
-import { CreateModelPackageInput, CreateModelPackageOutput } from "../models/models_1";
+import {
+  CreateModelPackageInput,
+  CreateModelPackageInputFilterSensitiveLog,
+  CreateModelPackageOutput,
+} from "../models/models_1";
 import { de_CreateModelPackageCommand, se_CreateModelPackageCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 
@@ -299,6 +303,13 @@ export interface CreateModelPackageCommandOutput extends CreateModelPackageOutpu
  *   ],
  *   SkipModelValidation: "All" || "None",
  *   SourceUri: "STRING_VALUE",
+ *   SecurityConfig: { // ModelPackageSecurityConfig
+ *     KmsKeyId: "STRING_VALUE", // required
+ *   },
+ *   ModelCard: { // ModelPackageModelCard
+ *     ModelCardContent: "STRING_VALUE",
+ *     ModelCardStatus: "Draft" || "PendingReview" || "Approved" || "Archived",
+ *   },
  * };
  * const command = new CreateModelPackageCommand(input);
  * const response = await client.send(command);
@@ -346,7 +357,7 @@ export class CreateModelPackageCommand extends $Command
   })
   .s("SageMaker", "CreateModelPackage", {})
   .n("SageMakerClient", "CreateModelPackageCommand")
-  .f(void 0, void 0)
+  .f(CreateModelPackageInputFilterSensitiveLog, void 0)
   .ser(se_CreateModelPackageCommand)
   .de(de_CreateModelPackageCommand)
   .build() {}
diff --git a/clients/client-sagemaker/src/commands/CreateStudioLifecycleConfigCommand.ts b/clients/client-sagemaker/src/commands/CreateStudioLifecycleConfigCommand.ts
@@ -5,8 +5,7 @@ import { Command as $Command } from "@smithy/smithy-client";
 import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { commonParams } from "../endpoint/EndpointParameters";
-import { CreateStudioLifecycleConfigRequest } from "../models/models_1";
-import { CreateStudioLifecycleConfigResponse } from "../models/models_2";
+import { CreateStudioLifecycleConfigRequest, CreateStudioLifecycleConfigResponse } from "../models/models_2";
 import { de_CreateStudioLifecycleConfigCommand, se_CreateStudioLifecycleConfigCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 

diff --git a/clients/client-sagemaker/src/commands/DescribeAutoMLJobCommand.ts b/clients/client-sagemaker/src/commands/DescribeAutoMLJobCommand.ts
@@ -93,7 +93,7 @@ export interface DescribeAutoMLJobCommandOutput extends DescribeAutoMLJobRespons
  * //       AlgorithmsConfig: [ // AutoMLAlgorithmsConfig
  * //         { // AutoMLAlgorithmConfig
  * //           AutoMLAlgorithms: [ // AutoMLAlgorithms // required
- * //             "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai",
+ * //             "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
  * //           ],
  * //         },
  * //       ],

diff --git a/clients/client-sagemaker/src/commands/DescribeAutoMLJobV2Command.ts b/clients/client-sagemaker/src/commands/DescribeAutoMLJobV2Command.ts
@@ -116,13 +116,22 @@ export interface DescribeAutoMLJobV2CommandOutput extends DescribeAutoMLJobV2Res
  * //           CountryCode: "STRING_VALUE",
  * //         },
  * //       ],
- * //     },
- * //     TabularJobConfig: { // TabularJobConfig
  * //       CandidateGenerationConfig: { // CandidateGenerationConfig
  * //         AlgorithmsConfig: [ // AutoMLAlgorithmsConfig
  * //           { // AutoMLAlgorithmConfig
  * //             AutoMLAlgorithms: [ // AutoMLAlgorithms // required
- * //               "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai",
+ * //               "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
+ * //             ],
+ * //           },
+ * //         ],
+ * //       },
+ * //     },
+ * //     TabularJobConfig: { // TabularJobConfig
+ * //       CandidateGenerationConfig: {
+ * //         AlgorithmsConfig: [
+ * //           {
+ * //             AutoMLAlgorithms: [ // required
+ * //               "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
  * //             ],
  * //           },
  * //         ],

diff --git a/clients/client-sagemaker/src/commands/DescribeModelPackageCommand.ts b/clients/client-sagemaker/src/commands/DescribeModelPackageCommand.ts
@@ -5,7 +5,11 @@ import { Command as $Command } from "@smithy/smithy-client";
 import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { commonParams } from "../endpoint/EndpointParameters";
-import { DescribeModelPackageInput, DescribeModelPackageOutput } from "../models/models_2";
+import {
+  DescribeModelPackageInput,
+  DescribeModelPackageOutput,
+  DescribeModelPackageOutputFilterSensitiveLog,
+} from "../models/models_2";
 import { de_DescribeModelPackageCommand, se_DescribeModelPackageCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 
@@ -29,6 +33,11 @@ export interface DescribeModelPackageCommandOutput extends DescribeModelPackageO
 /**
  * <p>Returns a description of the specified model package, which is used to create SageMaker
  *             models or list them on Amazon Web Services Marketplace.</p>
+ *          <important>
+ *             <p>If you provided a KMS Key ID when you created your model package,
+ *             you will see the <a href="https://docs.aws.amazon.com/kms/latest/APIReference/API_Decrypt.html">KMS
+ *                 Decrypt</a> API call in your CloudTrail logs when you use this API.</p>
+ *          </important>
  *          <p>To create models in SageMaker, buyers can subscribe to model packages listed on Amazon Web Services
  *             Marketplace.</p>
  * @example
@@ -323,6 +332,13 @@ export interface DescribeModelPackageCommandOutput extends DescribeModelPackageO
  * //   ],
  * //   SkipModelValidation: "All" || "None",
  * //   SourceUri: "STRING_VALUE",
+ * //   SecurityConfig: { // ModelPackageSecurityConfig
+ * //     KmsKeyId: "STRING_VALUE", // required
+ * //   },
+ * //   ModelCard: { // ModelPackageModelCard
+ * //     ModelCardContent: "STRING_VALUE",
+ * //     ModelCardStatus: "Draft" || "PendingReview" || "Approved" || "Archived",
+ * //   },
  * // };
  *
  * ```
@@ -357,7 +373,7 @@ export class DescribeModelPackageCommand extends $Command
   })
   .s("SageMaker", "DescribeModelPackage", {})
   .n("SageMakerClient", "DescribeModelPackageCommand")
-  .f(void 0, void 0)
+  .f(void 0, DescribeModelPackageOutputFilterSensitiveLog)
   .ser(se_DescribeModelPackageCommand)
   .de(de_DescribeModelPackageCommand)
   .build() {}
diff --git a/clients/client-sagemaker/src/commands/DescribeProcessingJobCommand.ts b/clients/client-sagemaker/src/commands/DescribeProcessingJobCommand.ts
@@ -5,7 +5,8 @@ import { Command as $Command } from "@smithy/smithy-client";
 import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { commonParams } from "../endpoint/EndpointParameters";
-import { DescribeProcessingJobRequest, DescribeProcessingJobResponse } from "../models/models_2";
+import { DescribeProcessingJobRequest } from "../models/models_2";
+import { DescribeProcessingJobResponse } from "../models/models_3";
 import { de_DescribeProcessingJobCommand, se_DescribeProcessingJobCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 

diff --git a/clients/client-sagemaker/src/commands/DescribeProjectCommand.ts b/clients/client-sagemaker/src/commands/DescribeProjectCommand.ts
@@ -5,8 +5,7 @@ import { Command as $Command } from "@smithy/smithy-client";
 import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { commonParams } from "../endpoint/EndpointParameters";
-import { DescribeProjectInput } from "../models/models_2";
-import { DescribeProjectOutput } from "../models/models_3";
+import { DescribeProjectInput, DescribeProjectOutput } from "../models/models_3";
 import { de_DescribeProjectCommand, se_DescribeProjectCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 

diff --git a/clients/client-sagemaker/src/commands/ListNotebookInstancesCommand.ts b/clients/client-sagemaker/src/commands/ListNotebookInstancesCommand.ts
@@ -5,7 +5,8 @@ import { Command as $Command } from "@smithy/smithy-client";
 import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { commonParams } from "../endpoint/EndpointParameters";
-import { ListNotebookInstancesInput, ListNotebookInstancesOutput } from "../models/models_3";
+import { ListNotebookInstancesInput } from "../models/models_3";
+import { ListNotebookInstancesOutput } from "../models/models_4";
 import { de_ListNotebookInstancesCommand, se_ListNotebookInstancesCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 

diff --git a/clients/client-sagemaker/src/commands/SearchCommand.ts b/clients/client-sagemaker/src/commands/SearchCommand.ts
@@ -1305,6 +1305,13 @@ export interface SearchCommandOutput extends SearchResponse, __MetadataBearer {}
  * //           },
  * //         ],
  * //         SourceUri: "STRING_VALUE",
+ * //         SecurityConfig: { // ModelPackageSecurityConfig
+ * //           KmsKeyId: "STRING_VALUE", // required
+ * //         },
+ * //         ModelCard: { // ModelPackageModelCard
+ * //           ModelCardContent: "STRING_VALUE",
+ * //           ModelCardStatus: "Draft" || "PendingReview" || "Approved" || "Archived",
+ * //         },
  * //         Tags: "<TagList>",
  * //         CustomerMetadataProperties: { // CustomerMetadataMap
  * //           "<keys>": "STRING_VALUE",

diff --git a/clients/client-sagemaker/src/commands/UpdateModelPackageCommand.ts b/clients/client-sagemaker/src/commands/UpdateModelPackageCommand.ts
@@ -5,7 +5,11 @@ import { Command as $Command } from "@smithy/smithy-client";
 import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { commonParams } from "../endpoint/EndpointParameters";
-import { UpdateModelPackageInput, UpdateModelPackageOutput } from "../models/models_4";
+import {
+  UpdateModelPackageInput,
+  UpdateModelPackageInputFilterSensitiveLog,
+  UpdateModelPackageOutput,
+} from "../models/models_4";
 import { de_UpdateModelPackageCommand, se_UpdateModelPackageCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 
@@ -143,6 +147,10 @@ export interface UpdateModelPackageCommandOutput extends UpdateModelPackageOutpu
  *     ],
  *   },
  *   SourceUri: "STRING_VALUE",
+ *   ModelCard: { // ModelPackageModelCard
+ *     ModelCardContent: "STRING_VALUE",
+ *     ModelCardStatus: "Draft" || "PendingReview" || "Approved" || "Archived",
+ *   },
  * };
  * const command = new UpdateModelPackageCommand(input);
  * const response = await client.send(command);
@@ -186,7 +194,7 @@ export class UpdateModelPackageCommand extends $Command
   })
   .s("SageMaker", "UpdateModelPackage", {})
   .n("SageMakerClient", "UpdateModelPackageCommand")
-  .f(void 0, void 0)
+  .f(UpdateModelPackageInputFilterSensitiveLog, void 0)
   .ser(se_UpdateModelPackageCommand)
   .de(de_UpdateModelPackageCommand)
   .build() {}