Skip to content

Commit 393afc7

Browse files
author
awstools
committedMay 30, 2024
feat(client-sagemaker): Adds Model Card information as a new component to Model Package. Autopilot launches algorithm selection for TimeSeries modality to generate AutoML candidates per algorithm.
1 parent c2f93e1 commit 393afc7

19 files changed

+921
-454
lines changed
 

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ export interface CreateAutoMLJobCommandOutput extends CreateAutoMLJobResponse, _
9393
* AlgorithmsConfig: [ // AutoMLAlgorithmsConfig
9494
* { // AutoMLAlgorithmConfig
9595
* AutoMLAlgorithms: [ // AutoMLAlgorithms // required
96-
* "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai",
96+
* "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
9797
* ],
9898
* },
9999
* ],

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

+12-3
Original file line numberDiff line numberDiff line change
@@ -120,13 +120,22 @@ export interface CreateAutoMLJobV2CommandOutput extends CreateAutoMLJobV2Respons
120120
* CountryCode: "STRING_VALUE",
121121
* },
122122
* ],
123-
* },
124-
* TabularJobConfig: { // TabularJobConfig
125123
* CandidateGenerationConfig: { // CandidateGenerationConfig
126124
* AlgorithmsConfig: [ // AutoMLAlgorithmsConfig
127125
* { // AutoMLAlgorithmConfig
128126
* AutoMLAlgorithms: [ // AutoMLAlgorithms // required
129-
* "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai",
127+
* "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
128+
* ],
129+
* },
130+
* ],
131+
* },
132+
* },
133+
* TabularJobConfig: { // TabularJobConfig
134+
* CandidateGenerationConfig: {
135+
* AlgorithmsConfig: [
136+
* {
137+
* AutoMLAlgorithms: [ // required
138+
* "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
130139
* ],
131140
* },
132141
* ],

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

+13-2
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,11 @@ import { Command as $Command } from "@smithy/smithy-client";
55
import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { commonParams } from "../endpoint/EndpointParameters";
8-
import { CreateModelPackageInput, CreateModelPackageOutput } from "../models/models_1";
8+
import {
9+
CreateModelPackageInput,
10+
CreateModelPackageInputFilterSensitiveLog,
11+
CreateModelPackageOutput,
12+
} from "../models/models_1";
913
import { de_CreateModelPackageCommand, se_CreateModelPackageCommand } from "../protocols/Aws_json1_1";
1014
import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
1115

@@ -299,6 +303,13 @@ export interface CreateModelPackageCommandOutput extends CreateModelPackageOutpu
299303
* ],
300304
* SkipModelValidation: "All" || "None",
301305
* SourceUri: "STRING_VALUE",
306+
* SecurityConfig: { // ModelPackageSecurityConfig
307+
* KmsKeyId: "STRING_VALUE", // required
308+
* },
309+
* ModelCard: { // ModelPackageModelCard
310+
* ModelCardContent: "STRING_VALUE",
311+
* ModelCardStatus: "Draft" || "PendingReview" || "Approved" || "Archived",
312+
* },
302313
* };
303314
* const command = new CreateModelPackageCommand(input);
304315
* const response = await client.send(command);
@@ -346,7 +357,7 @@ export class CreateModelPackageCommand extends $Command
346357
})
347358
.s("SageMaker", "CreateModelPackage", {})
348359
.n("SageMakerClient", "CreateModelPackageCommand")
349-
.f(void 0, void 0)
360+
.f(CreateModelPackageInputFilterSensitiveLog, void 0)
350361
.ser(se_CreateModelPackageCommand)
351362
.de(de_CreateModelPackageCommand)
352363
.build() {}

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

+1-2
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,7 @@ import { Command as $Command } from "@smithy/smithy-client";
55
import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { commonParams } from "../endpoint/EndpointParameters";
8-
import { CreateStudioLifecycleConfigRequest } from "../models/models_1";
9-
import { CreateStudioLifecycleConfigResponse } from "../models/models_2";
8+
import { CreateStudioLifecycleConfigRequest, CreateStudioLifecycleConfigResponse } from "../models/models_2";
109
import { de_CreateStudioLifecycleConfigCommand, se_CreateStudioLifecycleConfigCommand } from "../protocols/Aws_json1_1";
1110
import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
1211

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ export interface DescribeAutoMLJobCommandOutput extends DescribeAutoMLJobRespons
9393
* // AlgorithmsConfig: [ // AutoMLAlgorithmsConfig
9494
* // { // AutoMLAlgorithmConfig
9595
* // AutoMLAlgorithms: [ // AutoMLAlgorithms // required
96-
* // "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai",
96+
* // "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
9797
* // ],
9898
* // },
9999
* // ],

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

+12-3
Original file line numberDiff line numberDiff line change
@@ -116,13 +116,22 @@ export interface DescribeAutoMLJobV2CommandOutput extends DescribeAutoMLJobV2Res
116116
* // CountryCode: "STRING_VALUE",
117117
* // },
118118
* // ],
119-
* // },
120-
* // TabularJobConfig: { // TabularJobConfig
121119
* // CandidateGenerationConfig: { // CandidateGenerationConfig
122120
* // AlgorithmsConfig: [ // AutoMLAlgorithmsConfig
123121
* // { // AutoMLAlgorithmConfig
124122
* // AutoMLAlgorithms: [ // AutoMLAlgorithms // required
125-
* // "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai",
123+
* // "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
124+
* // ],
125+
* // },
126+
* // ],
127+
* // },
128+
* // },
129+
* // TabularJobConfig: { // TabularJobConfig
130+
* // CandidateGenerationConfig: {
131+
* // AlgorithmsConfig: [
132+
* // {
133+
* // AutoMLAlgorithms: [ // required
134+
* // "xgboost" || "linear-learner" || "mlp" || "lightgbm" || "catboost" || "randomforest" || "extra-trees" || "nn-torch" || "fastai" || "cnn-qr" || "deepar" || "prophet" || "npts" || "arima" || "ets",
126135
* // ],
127136
* // },
128137
* // ],

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

+18-2
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,11 @@ import { Command as $Command } from "@smithy/smithy-client";
55
import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { commonParams } from "../endpoint/EndpointParameters";
8-
import { DescribeModelPackageInput, DescribeModelPackageOutput } from "../models/models_2";
8+
import {
9+
DescribeModelPackageInput,
10+
DescribeModelPackageOutput,
11+
DescribeModelPackageOutputFilterSensitiveLog,
12+
} from "../models/models_2";
913
import { de_DescribeModelPackageCommand, se_DescribeModelPackageCommand } from "../protocols/Aws_json1_1";
1014
import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
1115

@@ -29,6 +33,11 @@ export interface DescribeModelPackageCommandOutput extends DescribeModelPackageO
2933
/**
3034
* <p>Returns a description of the specified model package, which is used to create SageMaker
3135
* models or list them on Amazon Web Services Marketplace.</p>
36+
* <important>
37+
* <p>If you provided a KMS Key ID when you created your model package,
38+
* you will see the <a href="https://docs.aws.amazon.com/kms/latest/APIReference/API_Decrypt.html">KMS
39+
* Decrypt</a> API call in your CloudTrail logs when you use this API.</p>
40+
* </important>
3241
* <p>To create models in SageMaker, buyers can subscribe to model packages listed on Amazon Web Services
3342
* Marketplace.</p>
3443
* @example
@@ -323,6 +332,13 @@ export interface DescribeModelPackageCommandOutput extends DescribeModelPackageO
323332
* // ],
324333
* // SkipModelValidation: "All" || "None",
325334
* // SourceUri: "STRING_VALUE",
335+
* // SecurityConfig: { // ModelPackageSecurityConfig
336+
* // KmsKeyId: "STRING_VALUE", // required
337+
* // },
338+
* // ModelCard: { // ModelPackageModelCard
339+
* // ModelCardContent: "STRING_VALUE",
340+
* // ModelCardStatus: "Draft" || "PendingReview" || "Approved" || "Archived",
341+
* // },
326342
* // };
327343
*
328344
* ```
@@ -357,7 +373,7 @@ export class DescribeModelPackageCommand extends $Command
357373
})
358374
.s("SageMaker", "DescribeModelPackage", {})
359375
.n("SageMakerClient", "DescribeModelPackageCommand")
360-
.f(void 0, void 0)
376+
.f(void 0, DescribeModelPackageOutputFilterSensitiveLog)
361377
.ser(se_DescribeModelPackageCommand)
362378
.de(de_DescribeModelPackageCommand)
363379
.build() {}

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

+2-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@ import { Command as $Command } from "@smithy/smithy-client";
55
import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { commonParams } from "../endpoint/EndpointParameters";
8-
import { DescribeProcessingJobRequest, DescribeProcessingJobResponse } from "../models/models_2";
8+
import { DescribeProcessingJobRequest } from "../models/models_2";
9+
import { DescribeProcessingJobResponse } from "../models/models_3";
910
import { de_DescribeProcessingJobCommand, se_DescribeProcessingJobCommand } from "../protocols/Aws_json1_1";
1011
import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
1112

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

+1-2
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,7 @@ import { Command as $Command } from "@smithy/smithy-client";
55
import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { commonParams } from "../endpoint/EndpointParameters";
8-
import { DescribeProjectInput } from "../models/models_2";
9-
import { DescribeProjectOutput } from "../models/models_3";
8+
import { DescribeProjectInput, DescribeProjectOutput } from "../models/models_3";
109
import { de_DescribeProjectCommand, se_DescribeProjectCommand } from "../protocols/Aws_json1_1";
1110
import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
1211

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

+2-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@ import { Command as $Command } from "@smithy/smithy-client";
55
import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { commonParams } from "../endpoint/EndpointParameters";
8-
import { ListNotebookInstancesInput, ListNotebookInstancesOutput } from "../models/models_3";
8+
import { ListNotebookInstancesInput } from "../models/models_3";
9+
import { ListNotebookInstancesOutput } from "../models/models_4";
910
import { de_ListNotebookInstancesCommand, se_ListNotebookInstancesCommand } from "../protocols/Aws_json1_1";
1011
import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
1112

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

+7
Original file line numberDiff line numberDiff line change
@@ -1305,6 +1305,13 @@ export interface SearchCommandOutput extends SearchResponse, __MetadataBearer {}
13051305
* // },
13061306
* // ],
13071307
* // SourceUri: "STRING_VALUE",
1308+
* // SecurityConfig: { // ModelPackageSecurityConfig
1309+
* // KmsKeyId: "STRING_VALUE", // required
1310+
* // },
1311+
* // ModelCard: { // ModelPackageModelCard
1312+
* // ModelCardContent: "STRING_VALUE",
1313+
* // ModelCardStatus: "Draft" || "PendingReview" || "Approved" || "Archived",
1314+
* // },
13081315
* // Tags: "<TagList>",
13091316
* // CustomerMetadataProperties: { // CustomerMetadataMap
13101317
* // "<keys>": "STRING_VALUE",

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

+10-2
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,11 @@ import { Command as $Command } from "@smithy/smithy-client";
55
import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { commonParams } from "../endpoint/EndpointParameters";
8-
import { UpdateModelPackageInput, UpdateModelPackageOutput } from "../models/models_4";
8+
import {
9+
UpdateModelPackageInput,
10+
UpdateModelPackageInputFilterSensitiveLog,
11+
UpdateModelPackageOutput,
12+
} from "../models/models_4";
913
import { de_UpdateModelPackageCommand, se_UpdateModelPackageCommand } from "../protocols/Aws_json1_1";
1014
import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
1115

@@ -143,6 +147,10 @@ export interface UpdateModelPackageCommandOutput extends UpdateModelPackageOutpu
143147
* ],
144148
* },
145149
* SourceUri: "STRING_VALUE",
150+
* ModelCard: { // ModelPackageModelCard
151+
* ModelCardContent: "STRING_VALUE",
152+
* ModelCardStatus: "Draft" || "PendingReview" || "Approved" || "Archived",
153+
* },
146154
* };
147155
* const command = new UpdateModelPackageCommand(input);
148156
* const response = await client.send(command);
@@ -186,7 +194,7 @@ export class UpdateModelPackageCommand extends $Command
186194
})
187195
.s("SageMaker", "UpdateModelPackage", {})
188196
.n("SageMakerClient", "UpdateModelPackageCommand")
189-
.f(void 0, void 0)
197+
.f(UpdateModelPackageInputFilterSensitiveLog, void 0)
190198
.ser(se_UpdateModelPackageCommand)
191199
.de(de_UpdateModelPackageCommand)
192200
.build() {}

0 commit comments

Comments
 (0)
Please sign in to comment.