feat(client-sagemaker): This release adds support for Model Cards Mod…

…el Registry integration.
aws · Jun 28, 2023 · 75339d2 · 75339d2
1 parent bb74957
commit 75339d2
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 2 deletions.
diff --git a/clients/client-sagemaker/src/commands/SearchCommand.ts b/clients/client-sagemaker/src/commands/SearchCommand.ts
@@ -1976,6 +1976,7 @@ export interface SearchCommandOutput extends SearchResponse, __MetadataBearer {}
  * //         Tags: "<TagList>",
  * //         ModelId: "STRING_VALUE",
  * //         RiskRating: "STRING_VALUE",
+ * //         ModelPackageGroupName: "STRING_VALUE",
  * //       },
  * //     },
  * //   ],

diff --git a/clients/client-sagemaker/src/models/models_1.ts b/clients/client-sagemaker/src/models/models_1.ts
@@ -8110,7 +8110,8 @@ export interface CreatePipelineRequest {
   PipelineDisplayName?: string;
 
   /**
-   * <p>The JSON pipeline definition of the pipeline.</p>
+   * <p>The <a href="https://aws-sagemaker-mlops.github.io/sagemaker-model-building-pipeline-definition-JSON-schema/">JSON
+   *          pipeline definition</a> of the pipeline.</p>
    */
   PipelineDefinition?: string;
 

diff --git a/clients/client-sagemaker/src/models/models_3.ts b/clients/client-sagemaker/src/models/models_3.ts
@@ -7972,6 +7972,12 @@ export interface ModelCard {
    * <p>The risk rating of the model. Different organizations might have different criteria for model card risk ratings. For more information, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/model-cards-risk-rating.html">Risk ratings</a>.</p>
    */
   RiskRating?: string;
+
+  /**
+   * <p>The model package group that contains the model package. Only relevant for model cards created for model packages in the Amazon SageMaker Model Registry.
+   *       </p>
+   */
+  ModelPackageGroupName?: string;
 }
 
 /**

diff --git a/clients/client-sagemaker/src/protocols/Aws_json1_1.ts b/clients/client-sagemaker/src/protocols/Aws_json1_1.ts
@@ -26135,6 +26135,7 @@ const de_ModelCard = (output: any, context: __SerdeContext): ModelCard => {
     ModelCardStatus: __expectString,
     ModelCardVersion: __expectInt32,
     ModelId: __expectString,
+    ModelPackageGroupName: __expectString,
     RiskRating: __expectString,
     SecurityConfig: _json,
     Tags: _json,

diff --git a/codegen/sdk-codegen/aws-models/sagemaker.json b/codegen/sdk-codegen/aws-models/sagemaker.json
@@ -9833,7 +9833,7 @@
         "PipelineDefinition": {
           "target": "com.amazonaws.sagemaker#PipelineDefinition",
           "traits": {
-            "smithy.api#documentation": "<p>The JSON pipeline definition of the pipeline.</p>"
+            "smithy.api#documentation": "<p>The <a href=\"https://aws-sagemaker-mlops.github.io/sagemaker-model-building-pipeline-definition-JSON-schema/\">JSON \n         pipeline definition</a> of the pipeline.</p>"
           }
         },
         "PipelineDefinitionS3Location": {
@@ -37268,6 +37268,12 @@
           "traits": {
             "smithy.api#documentation": "<p>The risk rating of the model. Different organizations might have different criteria for model card risk ratings. For more information, see <a href=\"https://docs.aws.amazon.com/sagemaker/latest/dg/model-cards-risk-rating.html\">Risk ratings</a>.</p>"
           }
+        },
+        "ModelPackageGroupName": {
+          "target": "com.amazonaws.sagemaker#String",
+          "traits": {
+            "smithy.api#documentation": "<p>The model package group that contains the model package. Only relevant for model cards created for model packages in the Amazon SageMaker Model Registry.\n      </p>"
+          }
         }
       },
       "traits": {