feat(client-sagemaker): This release adds Selective Execution feature…

… that allows SageMaker Pipelines users to run selected steps in a pipeline.
aws · Jun 2, 2023 · 4d78994 · 4d78994
1 parent ac3fec7
commit 4d78994
Show file tree

Hide file tree

Showing 14 changed files with 317 additions and 118 deletions.
diff --git a/clients/client-sagemaker/src/commands/DescribePipelineExecutionCommand.ts b/clients/client-sagemaker/src/commands/DescribePipelineExecutionCommand.ts
@@ -84,6 +84,14 @@ export interface DescribePipelineExecutionCommandOutput extends DescribePipeline
  * //   ParallelismConfiguration: { // ParallelismConfiguration
  * //     MaxParallelExecutionSteps: Number("int"), // required
  * //   },
+ * //   SelectiveExecutionConfig: { // SelectiveExecutionConfig
+ * //     SourcePipelineExecutionArn: "STRING_VALUE", // required
+ * //     SelectedSteps: [ // SelectedStepList // required
+ * //       { // SelectedStep
+ * //         StepName: "STRING_VALUE", // required
+ * //       },
+ * //     ],
+ * //   },
  * // };
  *
  * ```

diff --git a/clients/client-sagemaker/src/commands/ImportHubContentCommand.ts b/clients/client-sagemaker/src/commands/ImportHubContentCommand.ts
@@ -13,7 +13,7 @@ import {
 import { HttpRequest as __HttpRequest, HttpResponse as __HttpResponse } from "@smithy/protocol-http";
 import { SerdeContext as __SerdeContext } from "@smithy/types";
 
-import { ImportHubContentRequest, ImportHubContentResponse } from "../models/models_2";
+import { ImportHubContentRequest, ImportHubContentResponse } from "../models/models_3";
 import { de_ImportHubContentCommand, se_ImportHubContentCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 

diff --git a/clients/client-sagemaker/src/commands/ListPipelineExecutionStepsCommand.ts b/clients/client-sagemaker/src/commands/ListPipelineExecutionStepsCommand.ts
@@ -141,6 +141,9 @@ export interface ListPipelineExecutionStepsCommandOutput extends ListPipelineExe
  * //           Arn: "STRING_VALUE",
  * //         },
  * //       },
+ * //       SelectiveExecutionResult: { // SelectiveExecutionResult
+ * //         SourcePipelineExecutionArn: "STRING_VALUE",
+ * //       },
  * //     },
  * //   ],
  * //   NextToken: "STRING_VALUE",

diff --git a/clients/client-sagemaker/src/commands/SearchCommand.ts b/clients/client-sagemaker/src/commands/SearchCommand.ts
@@ -1304,6 +1304,14 @@ export interface SearchCommandOutput extends SearchResponse, __MetadataBearer {}
  * //             Value: "STRING_VALUE", // required
  * //           },
  * //         ],
+ * //         SelectiveExecutionConfig: { // SelectiveExecutionConfig
+ * //           SourcePipelineExecutionArn: "STRING_VALUE", // required
+ * //           SelectedSteps: [ // SelectedStepList // required
+ * //             { // SelectedStep
+ * //               StepName: "STRING_VALUE", // required
+ * //             },
+ * //           ],
+ * //         },
  * //       },
  * //       FeatureGroup: { // FeatureGroup
  * //         FeatureGroupArn: "STRING_VALUE",

diff --git a/clients/client-sagemaker/src/commands/StartPipelineExecutionCommand.ts b/clients/client-sagemaker/src/commands/StartPipelineExecutionCommand.ts
@@ -57,6 +57,14 @@ export interface StartPipelineExecutionCommandOutput extends StartPipelineExecut
  *   ParallelismConfiguration: { // ParallelismConfiguration
  *     MaxParallelExecutionSteps: Number("int"), // required
  *   },
+ *   SelectiveExecutionConfig: { // SelectiveExecutionConfig
+ *     SourcePipelineExecutionArn: "STRING_VALUE", // required
+ *     SelectedSteps: [ // SelectedStepList // required
+ *       { // SelectedStep
+ *         StepName: "STRING_VALUE", // required
+ *       },
+ *     ],
+ *   },
  * };
  * const command = new StartPipelineExecutionCommand(input);
  * const response = await client.send(command);

diff --git a/clients/client-sagemaker/src/commands/StopInferenceRecommendationsJobCommand.ts b/clients/client-sagemaker/src/commands/StopInferenceRecommendationsJobCommand.ts
@@ -13,7 +13,7 @@ import {
 import { HttpRequest as __HttpRequest, HttpResponse as __HttpResponse } from "@smithy/protocol-http";
 import { SerdeContext as __SerdeContext } from "@smithy/types";
 
-import { StopInferenceRecommendationsJobRequest } from "../models/models_3";
+import { StopInferenceRecommendationsJobRequest } from "../models/models_4";
 import {
   de_StopInferenceRecommendationsJobCommand,
   se_StopInferenceRecommendationsJobCommand,

diff --git a/clients/client-sagemaker/src/commands/StopLabelingJobCommand.ts b/clients/client-sagemaker/src/commands/StopLabelingJobCommand.ts
@@ -13,7 +13,7 @@ import {
 import { HttpRequest as __HttpRequest, HttpResponse as __HttpResponse } from "@smithy/protocol-http";
 import { SerdeContext as __SerdeContext } from "@smithy/types";
 
-import { StopLabelingJobRequest } from "../models/models_3";
+import { StopLabelingJobRequest } from "../models/models_4";
 import { de_StopLabelingJobCommand, se_StopLabelingJobCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 

diff --git a/clients/client-sagemaker/src/commands/StopMonitoringScheduleCommand.ts b/clients/client-sagemaker/src/commands/StopMonitoringScheduleCommand.ts
@@ -13,7 +13,7 @@ import {
 import { HttpRequest as __HttpRequest, HttpResponse as __HttpResponse } from "@smithy/protocol-http";
 import { SerdeContext as __SerdeContext } from "@smithy/types";
 
-import { StopMonitoringScheduleRequest } from "../models/models_3";
+import { StopMonitoringScheduleRequest } from "../models/models_4";
 import { de_StopMonitoringScheduleCommand, se_StopMonitoringScheduleCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 

diff --git a/clients/client-sagemaker/src/models/models_0.ts b/clients/client-sagemaker/src/models/models_0.ts
@@ -5028,12 +5028,13 @@ export interface AutoMLJobChannel {
    *          content types for different problems:</p>
    *          <ul>
    *             <li>
-   *                <p>ImageClassification: <code>image/png</code>, <code>image/jpeg</code>, or
-   *                   <code>image/*</code>. The default value is <code>image/*</code>.</p>
+   *                <p>ImageClassification: <code>image/png</code>, <code>image/jpeg</code>,
+   *                   <code>image/*</code>
+   *                </p>
    *             </li>
    *             <li>
-   *                <p>TextClassification: <code>text/csv;header=present</code> or <code>x-application/vnd.amazon+parquet</code>.
-   *                The default value is <code>text/csv;header=present</code>.</p>
+   *                <p>TextClassification: <code>text/csv;header=present</code>
+   *                </p>
    *             </li>
    *          </ul>
    */

diff --git a/clients/client-sagemaker/src/models/models_2.ts b/clients/client-sagemaker/src/models/models_2.ts
@@ -6110,6 +6110,37 @@ export interface PipelineExperimentConfig {
   TrialName?: string;
 }
 
+/**
+ * @public
+ * <p>A step selected to run in selective execution mode.</p>
+ */
+export interface SelectedStep {
+  /**
+   * <p>The name of the pipeline step.</p>
+   */
+  StepName: string | undefined;
+}
+
+/**
+ * @public
+ * <p>The selective execution configuration applied to the pipeline run.</p>
+ */
+export interface SelectiveExecutionConfig {
+  /**
+   * <p>The ARN from a reference execution of the current pipeline.
+   *         Used to copy input collaterals needed for the selected steps to run.
+   *         The execution status of the pipeline can be either <code>Failed</code>
+   *         or <code>Success</code>.</p>
+   */
+  SourcePipelineExecutionArn: string | undefined;
+
+  /**
+   * <p>A list of pipeline steps to run. All step(s) in all path(s) between
+   *         two selected steps should be included.</p>
+   */
+  SelectedSteps: SelectedStep[] | undefined;
+}
+
 /**
  * @public
  */
@@ -6175,6 +6206,11 @@ export interface DescribePipelineExecutionResponse {
    * <p>The parallelism configuration applied to the pipeline.</p>
    */
   ParallelismConfiguration?: ParallelismConfiguration;
+
+  /**
+   * <p>The selective execution configuration applied to the pipeline run.</p>
+   */
+  SelectiveExecutionConfig?: SelectiveExecutionConfig;
 }
 
 /**
@@ -10432,81 +10468,6 @@ export const ImageVersionSortOrder = {
  */
 export type ImageVersionSortOrder = (typeof ImageVersionSortOrder)[keyof typeof ImageVersionSortOrder];
 
-/**
- * @public
- */
-export interface ImportHubContentRequest {
-  /**
-   * <p>The name of the hub content to import.</p>
-   */
-  HubContentName: string | undefined;
-
-  /**
-   * <p>The version of the hub content to import.</p>
-   */
-  HubContentVersion?: string;
-
-  /**
-   * <p>The type of hub content to import.</p>
-   */
-  HubContentType: HubContentType | string | undefined;
-
-  /**
-   * <p>The version of the hub content schema to import.</p>
-   */
-  DocumentSchemaVersion: string | undefined;
-
-  /**
-   * <p>The name of the hub to import content into.</p>
-   */
-  HubName: string | undefined;
-
-  /**
-   * <p>The display name of the hub content to import.</p>
-   */
-  HubContentDisplayName?: string;
-
-  /**
-   * <p>A description of the hub content to import.</p>
-   */
-  HubContentDescription?: string;
-
-  /**
-   * <p>A string that provides a description of the hub content. This string can include links, tables, and standard markdown formating.</p>
-   */
-  HubContentMarkdown?: string;
-
-  /**
-   * <p>The hub content document that describes information about the hub content such as type, associated containers, scripts, and more.</p>
-   */
-  HubContentDocument: string | undefined;
-
-  /**
-   * <p>The searchable keywords of the hub content.</p>
-   */
-  HubContentSearchKeywords?: string[];
-
-  /**
-   * <p>Any tags associated with the hub content.</p>
-   */
-  Tags?: Tag[];
-}
-
-/**
- * @public
- */
-export interface ImportHubContentResponse {
-  /**
-   * <p>The ARN of the hub that the content was imported into.</p>
-   */
-  HubArn: string | undefined;
-
-  /**
-   * <p>The ARN of the hub content that was imported.</p>
-   */
-  HubContentArn: string | undefined;
-}
-
 /**
  * @internal
  */