From 126b1c6d83889830a94e30a8c56ec0e0ce1d5682 Mon Sep 17 00:00:00 2001
From: awstools <awstools@amazon.com>
Date: Fri, 4 Aug 2023 18:24:43 +0000
Subject: [PATCH] feat(client-sagemaker): Including DataCaptureConfig key in
 the Amazon Sagemaker Search's transform job object

---
 clients/client-sagemaker/src/commands/SearchCommand.ts | 10 ++++++++++
 clients/client-sagemaker/src/models/models_3.ts        |  7 +++++++
 clients/client-sagemaker/src/protocols/Aws_json1_1.ts  |  1 +
 codegen/sdk-codegen/aws-models/sagemaker.json          |  3 +++
 4 files changed, 21 insertions(+)

diff --git a/clients/client-sagemaker/src/commands/SearchCommand.ts b/clients/client-sagemaker/src/commands/SearchCommand.ts
index b2a74c3a4d71..a919c63f1906 100644
--- 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
index ab45788bab80..183c5c4f151b 100644
--- 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
index 6013e4082d9f..ceedcca97a51 100644
--- 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
index ea7ff330e562..0657215190f1 100644
--- 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": {