feat(client-sagemaker): Expose ProfilerConfig attribute in SageMaker …

…Search API response.
aws · Jul 27, 2023 · d6e75cf · d6e75cf
1 parent c705b97
commit d6e75cf
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 0 deletions.
diff --git a/clients/client-sagemaker/src/commands/SearchCommand.ts b/clients/client-sagemaker/src/commands/SearchCommand.ts
@@ -285,6 +285,14 @@ export interface SearchCommandOutput extends SearchResponse, __MetadataBearer {}
  * //             LastModifiedTime: new Date("TIMESTAMP"),
  * //           },
  * //         ],
+ * //         ProfilerConfig: { // ProfilerConfig
+ * //           S3OutputPath: "STRING_VALUE",
+ * //           ProfilingIntervalInMilliseconds: Number("long"),
+ * //           ProfilingParameters: { // ProfilingParameters
+ * //             "<keys>": "STRING_VALUE",
+ * //           },
+ * //           DisableProfiler: true || false,
+ * //         },
  * //         Environment: { // TrainingEnvironmentMap
  * //           "<keys>": "STRING_VALUE",
  * //         },
@@ -635,6 +643,14 @@ export interface SearchCommandOutput extends SearchResponse, __MetadataBearer {}
  * //                 LastModifiedTime: new Date("TIMESTAMP"),
  * //               },
  * //             ],
+ * //             ProfilerConfig: {
+ * //               S3OutputPath: "STRING_VALUE",
+ * //               ProfilingIntervalInMilliseconds: Number("long"),
+ * //               ProfilingParameters: {
+ * //                 "<keys>": "STRING_VALUE",
+ * //               },
+ * //               DisableProfiler: true || false,
+ * //             },
  * //             Environment: {
  * //               "<keys>": "STRING_VALUE",
  * //             },

diff --git a/clients/client-sagemaker/src/models/models_4.ts b/clients/client-sagemaker/src/models/models_4.ts
@@ -47,6 +47,7 @@ import {
   ParallelismConfiguration,
   PipelineDefinitionS3Location,
   Processor,
+  ProfilerConfig,
   ProfilerRuleConfiguration,
   ProvisioningParameter,
   RetryStrategy,
@@ -505,6 +506,12 @@ export interface TrainingJob {
    */
   DebugRuleEvaluationStatuses?: DebugRuleEvaluationStatus[];
 
+  /**
+   * <p>Configuration information for Amazon SageMaker Debugger system monitoring, framework profiling, and
+   *             storage paths.</p>
+   */
+  ProfilerConfig?: ProfilerConfig;
+
   /**
    * <p>The environment variables to set in the Docker container.</p>
    */

diff --git a/clients/client-sagemaker/src/protocols/Aws_json1_1.ts b/clients/client-sagemaker/src/protocols/Aws_json1_1.ts
@@ -27881,6 +27881,7 @@ const de_TrainingJob = (output: any, context: __SerdeContext): TrainingJob => {
     LastModifiedTime: (_: any) => __expectNonNull(__parseEpochTimestamp(__expectNumber(_))),
     ModelArtifacts: _json,
     OutputDataConfig: _json,
+    ProfilerConfig: _json,
     ResourceConfig: _json,
     RetryStrategy: _json,
     RoleArn: __expectString,

diff --git a/codegen/sdk-codegen/aws-models/sagemaker.json b/codegen/sdk-codegen/aws-models/sagemaker.json
@@ -54756,6 +54756,9 @@
             "smithy.api#documentation": "<p>Information about the evaluation status of the rules for the training job.</p>"
           }
         },
+        "ProfilerConfig": {
+          "target": "com.amazonaws.sagemaker#ProfilerConfig"
+        },
         "Environment": {
           "target": "com.amazonaws.sagemaker#TrainingEnvironmentMap",
           "traits": {