feat(client-sagemaker): Including DataCaptureConfig key in the Amazon…

… Sagemaker Search's transform job object
aws · Aug 4, 2023 · 126b1c6 · 126b1c6
1 parent 70146be
commit 126b1c6
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 0 deletions.
diff --git a/clients/client-sagemaker/src/commands/SearchCommand.ts b/clients/client-sagemaker/src/commands/SearchCommand.ts
@@ -831,6 +831,11 @@ export interface SearchCommandOutput extends SearchResponse, __MetadataBearer {}
  * //               RunName: "STRING_VALUE",
  * //             },
  * //             Tags: "<TagList>",
+ * //             DataCaptureConfig: { // BatchDataCaptureConfig
+ * //               DestinationS3Uri: "STRING_VALUE", // required
+ * //               KmsKeyId: "STRING_VALUE",
+ * //               GenerateInferenceId: true || false,
+ * //             },
  * //           },
  * //         },
  * //         LineageGroupArn: "STRING_VALUE",
@@ -1821,6 +1826,11 @@ export interface SearchCommandOutput extends SearchResponse, __MetadataBearer {}
  * //           },
  * //           ExperimentConfig: "<ExperimentConfig>",
  * //           Tags: "<TagList>",
+ * //           DataCaptureConfig: {
+ * //             DestinationS3Uri: "STRING_VALUE", // required
+ * //             KmsKeyId: "STRING_VALUE",
+ * //             GenerateInferenceId: true || false,
+ * //           },
  * //         },
  * //         MonitoringSchedules: [ // ModelDashboardMonitoringSchedules
  * //           { // ModelDashboardMonitoringSchedule

diff --git a/clients/client-sagemaker/src/models/models_3.ts b/clients/client-sagemaker/src/models/models_3.ts
@@ -21,6 +21,7 @@ import {
   AutoMLJobSummary,
   AutoMLSortBy,
   AutoMLSortOrder,
+  BatchDataCaptureConfig,
   BatchStrategy,
   CacheHitResult,
   CallbackStepMetadata,
@@ -9867,6 +9868,12 @@ export interface TransformJob {
    * <p>A list of tags associated with the transform job.</p>
    */
   Tags?: Tag[];
+
+  /**
+   * @public
+   * <p>Configuration to control how SageMaker captures inference data for batch transform jobs.</p>
+   */
+  DataCaptureConfig?: BatchDataCaptureConfig;
 }
 
 /**

diff --git a/clients/client-sagemaker/src/protocols/Aws_json1_1.ts b/clients/client-sagemaker/src/protocols/Aws_json1_1.ts
@@ -28110,6 +28110,7 @@ const de_TransformJob = (output: any, context: __SerdeContext): TransformJob =>
     AutoMLJobArn: __expectString,
     BatchStrategy: __expectString,
     CreationTime: (_: any) => __expectNonNull(__parseEpochTimestamp(__expectNumber(_))),
+    DataCaptureConfig: _json,
     DataProcessing: _json,
     Environment: _json,
     ExperimentConfig: _json,

diff --git a/codegen/sdk-codegen/aws-models/sagemaker.json b/codegen/sdk-codegen/aws-models/sagemaker.json
@@ -55952,6 +55952,9 @@
           "traits": {
             "smithy.api#documentation": "<p>A list of tags associated with the transform job.</p>"
           }
+        },
+        "DataCaptureConfig": {
+          "target": "com.amazonaws.sagemaker#BatchDataCaptureConfig"
         }
       },
       "traits": {