feat(client-sagemaker): Support for batch transform input in Model da…

…shboard
aws · Nov 1, 2023 · 1725b9c · 1725b9c
1 parent 1ef476e
commit 1725b9c
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 6 deletions.
diff --git a/clients/client-sagemaker/src/commands/SearchCommand.ts b/clients/client-sagemaker/src/commands/SearchCommand.ts
@@ -1994,6 +1994,28 @@ export interface SearchCommandOutput extends SearchResponse, __MetadataBearer {}
  * //               MonitoringJobDefinitionName: "STRING_VALUE",
  * //               MonitoringType: "DataQuality" || "ModelQuality" || "ModelBias" || "ModelExplainability",
  * //             },
+ * //             BatchTransformInput: {
+ * //               DataCapturedDestinationS3Uri: "STRING_VALUE", // required
+ * //               DatasetFormat: {
+ * //                 Csv: {
+ * //                   Header: true || false,
+ * //                 },
+ * //                 Json: {
+ * //                   Line: true || false,
+ * //                 },
+ * //                 Parquet: {},
+ * //               },
+ * //               LocalPath: "STRING_VALUE", // required
+ * //               S3InputMode: "Pipe" || "File",
+ * //               S3DataDistributionType: "FullyReplicated" || "ShardedByS3Key",
+ * //               FeaturesAttribute: "STRING_VALUE",
+ * //               InferenceAttribute: "STRING_VALUE",
+ * //               ProbabilityAttribute: "STRING_VALUE",
+ * //               ProbabilityThresholdAttribute: Number("double"),
+ * //               StartTimeOffset: "STRING_VALUE",
+ * //               EndTimeOffset: "STRING_VALUE",
+ * //               ExcludeFeaturesAttribute: "STRING_VALUE",
+ * //             },
  * //           },
  * //         ],
  * //         ModelCard: { // ModelDashboardModelCard

diff --git a/clients/client-sagemaker/src/models/models_4.ts b/clients/client-sagemaker/src/models/models_4.ts
@@ -9,6 +9,7 @@ import {
   AppSpecification,
   BatchDataCaptureConfig,
   BatchStrategy,
+  BatchTransformInput,
   BooleanOperator,
   Channel,
   CheckpointConfig,
@@ -532,6 +533,12 @@ export interface ModelDashboardMonitoringSchedule {
    * <p>Summary of information about the last monitoring job to run.</p>
    */
   LastMonitoringExecutionSummary?: MonitoringExecutionSummary;
+
+  /**
+   * @public
+   * <p>Input object for the batch transform job.</p>
+   */
+  BatchTransformInput?: BatchTransformInput;
 }
 
 /**

diff --git a/clients/client-sagemaker/src/protocols/Aws_json1_1.ts b/clients/client-sagemaker/src/protocols/Aws_json1_1.ts
@@ -26591,6 +26591,7 @@ const de_ModelDashboardMonitoringSchedule = (
   context: __SerdeContext
 ): ModelDashboardMonitoringSchedule => {
   return take(output, {
+    BatchTransformInput: (_: any) => de_BatchTransformInput(_, context),
     CreationTime: (_: any) => __expectNonNull(__parseEpochTimestamp(__expectNumber(_))),
     EndpointName: __expectString,
     FailureReason: __expectString,

diff --git a/codegen/sdk-codegen/aws-models/sagemaker.json b/codegen/sdk-codegen/aws-models/sagemaker.json
@@ -542,7 +542,7 @@
           "min": 1,
           "max": 2048
         },
-        "smithy.api#pattern": "^arn:aws[a-z\\-]*:sagemaker:[a-z0-9\\-]*:[0-9]{12}:algorithm/"
+        "smithy.api#pattern": "^arn:aws(-cn|-us-gov)?:sagemaker:[a-z0-9\\-]{9,16}:[0-9]{12}:algorithm/[\\S]{1,2048}$"
       }
     },
     "com.amazonaws.sagemaker#AlgorithmImage": {
@@ -5763,7 +5763,7 @@
           "min": 1,
           "max": 2048
         },
-        "smithy.api#pattern": "^arn:aws[a-z\\-]*:sagemaker:[a-z0-9\\-]*:[0-9]{12}:code-repository/"
+        "smithy.api#pattern": "^arn:aws(-cn|-us-gov)?:sagemaker:[a-z0-9\\-]{9,16}:[0-9]{12}:code-repository/[\\S]{1,2048}$"
       }
     },
     "com.amazonaws.sagemaker#CodeRepositoryContains": {
@@ -26309,7 +26309,11 @@
     "com.amazonaws.sagemaker#GitConfigUrl": {
       "type": "string",
       "traits": {
-        "smithy.api#pattern": "^https://([^/]+)/?(.*)$"
+        "smithy.api#length": {
+          "min": 11,
+          "max": 1024
+        },
+        "smithy.api#pattern": "^https://([^/]+)/?.{3,1016}$"
       }
     },
     "com.amazonaws.sagemaker#Group": {
@@ -39877,6 +39881,9 @@
         },
         "LastMonitoringExecutionSummary": {
           "target": "com.amazonaws.sagemaker#MonitoringExecutionSummary"
+        },
+        "BatchTransformInput": {
+          "target": "com.amazonaws.sagemaker#BatchTransformInput"
         }
       },
       "traits": {
@@ -40490,7 +40497,7 @@
           "min": 1,
           "max": 2048
         },
-        "smithy.api#pattern": "^arn:aws[a-z\\-]*:sagemaker:[a-z0-9\\-]*:[0-9]{12}:model-package/"
+        "smithy.api#pattern": "^arn:aws(-cn|-us-gov)?:sagemaker:[a-z0-9\\-]{9,16}:[0-9]{12}:model-package/[\\S]{1,2048}$"
       }
     },
     "com.amazonaws.sagemaker#ModelPackageArnList": {
@@ -40657,7 +40664,7 @@
           "min": 1,
           "max": 2048
         },
-        "smithy.api#pattern": "^arn:aws[a-z\\-]*:sagemaker:[a-z0-9\\-]*:[0-9]{12}:model-package-group/"
+        "smithy.api#pattern": "^arn:aws(-cn|-us-gov)?:sagemaker:[a-z0-9\\-]{9,16}:[0-9]{12}:model-package-group/[\\S]{1,2048}$"
       }
     },
     "com.amazonaws.sagemaker#ModelPackageGroupSortBy": {
@@ -47472,7 +47479,7 @@
           "min": 1,
           "max": 2048
         },
-        "smithy.api#pattern": "^arn:aws[a-z\\-]*:sagemaker:[a-z0-9\\-]*:[0-9]{12}:project:"
+        "smithy.api#pattern": "^arn:aws(-cn|-us-gov)?:sagemaker:[a-z0-9\\-]{9,16}:[0-9]{12}:project/[\\S]{1,2048}$"
       }
     },
     "com.amazonaws.sagemaker#ProjectEntityName": {