feat(client-sagemaker): This release adds support for Amazon EMR Serv…

…erless applications in SageMaker Studio for running data processing jobs.
aws · Aug 1, 2024 · b43bb89 · b43bb89
1 parent d2d3328
commit b43bb89
Show file tree

Hide file tree

Showing 12 changed files with 254 additions and 52 deletions.
diff --git a/clients/client-sagemaker/src/commands/CreateDomainCommand.ts b/clients/client-sagemaker/src/commands/CreateDomainCommand.ts
@@ -222,6 +222,14 @@ export interface CreateDomainCommandOutput extends CreateDomainResponse, __Metad
  *           RepositoryUrl: "STRING_VALUE", // required
  *         },
  *       ],
+ *       EmrSettings: { // EmrSettings
+ *         AssumableRoleArns: [ // AssumableRoleArns
+ *           "STRING_VALUE",
+ *         ],
+ *         ExecutionRoleArns: [ // ExecutionRoleArns
+ *           "STRING_VALUE",
+ *         ],
+ *       },
  *     },
  *     SpaceStorageSettings: { // DefaultSpaceStorageSettings
  *       DefaultEbsStorageSettings: { // DefaultEbsStorageSettings
@@ -324,6 +332,14 @@ export interface CreateDomainCommandOutput extends CreateDomainResponse, __Metad
  *           RepositoryUrl: "STRING_VALUE", // required
  *         },
  *       ],
+ *       EmrSettings: {
+ *         AssumableRoleArns: [
+ *           "STRING_VALUE",
+ *         ],
+ *         ExecutionRoleArns: [
+ *           "STRING_VALUE",
+ *         ],
+ *       },
  *     },
  *     SpaceStorageSettings: {
  *       DefaultEbsStorageSettings: {

diff --git a/clients/client-sagemaker/src/commands/CreateUserProfileCommand.ts b/clients/client-sagemaker/src/commands/CreateUserProfileCommand.ts
@@ -193,6 +193,14 @@ export interface CreateUserProfileCommandOutput extends CreateUserProfileRespons
  *           RepositoryUrl: "STRING_VALUE", // required
  *         },
  *       ],
+ *       EmrSettings: { // EmrSettings
+ *         AssumableRoleArns: [ // AssumableRoleArns
+ *           "STRING_VALUE",
+ *         ],
+ *         ExecutionRoleArns: [ // ExecutionRoleArns
+ *           "STRING_VALUE",
+ *         ],
+ *       },
  *     },
  *     SpaceStorageSettings: { // DefaultSpaceStorageSettings
  *       DefaultEbsStorageSettings: { // DefaultEbsStorageSettings

diff --git a/clients/client-sagemaker/src/commands/DescribeDomainCommand.ts b/clients/client-sagemaker/src/commands/DescribeDomainCommand.ts
@@ -194,6 +194,14 @@ export interface DescribeDomainCommandOutput extends DescribeDomainResponse, __M
  * //           RepositoryUrl: "STRING_VALUE", // required
  * //         },
  * //       ],
+ * //       EmrSettings: { // EmrSettings
+ * //         AssumableRoleArns: [ // AssumableRoleArns
+ * //           "STRING_VALUE",
+ * //         ],
+ * //         ExecutionRoleArns: [ // ExecutionRoleArns
+ * //           "STRING_VALUE",
+ * //         ],
+ * //       },
  * //     },
  * //     SpaceStorageSettings: { // DefaultSpaceStorageSettings
  * //       DefaultEbsStorageSettings: { // DefaultEbsStorageSettings
@@ -291,6 +299,14 @@ export interface DescribeDomainCommandOutput extends DescribeDomainResponse, __M
  * //           RepositoryUrl: "STRING_VALUE", // required
  * //         },
  * //       ],
+ * //       EmrSettings: {
+ * //         AssumableRoleArns: [
+ * //           "STRING_VALUE",
+ * //         ],
+ * //         ExecutionRoleArns: [
+ * //           "STRING_VALUE",
+ * //         ],
+ * //       },
  * //     },
  * //     SpaceStorageSettings: {
  * //       DefaultEbsStorageSettings: {

diff --git a/clients/client-sagemaker/src/commands/DescribeUserProfileCommand.ts b/clients/client-sagemaker/src/commands/DescribeUserProfileCommand.ts
@@ -193,6 +193,14 @@ export interface DescribeUserProfileCommandOutput extends DescribeUserProfileRes
  * //           RepositoryUrl: "STRING_VALUE", // required
  * //         },
  * //       ],
+ * //       EmrSettings: { // EmrSettings
+ * //         AssumableRoleArns: [ // AssumableRoleArns
+ * //           "STRING_VALUE",
+ * //         ],
+ * //         ExecutionRoleArns: [ // ExecutionRoleArns
+ * //           "STRING_VALUE",
+ * //         ],
+ * //       },
  * //     },
  * //     SpaceStorageSettings: { // DefaultSpaceStorageSettings
  * //       DefaultEbsStorageSettings: { // DefaultEbsStorageSettings

diff --git a/clients/client-sagemaker/src/commands/UpdateDomainCommand.ts b/clients/client-sagemaker/src/commands/UpdateDomainCommand.ts
@@ -178,6 +178,14 @@ export interface UpdateDomainCommandOutput extends UpdateDomainResponse, __Metad
  *           RepositoryUrl: "STRING_VALUE", // required
  *         },
  *       ],
+ *       EmrSettings: { // EmrSettings
+ *         AssumableRoleArns: [ // AssumableRoleArns
+ *           "STRING_VALUE",
+ *         ],
+ *         ExecutionRoleArns: [ // ExecutionRoleArns
+ *           "STRING_VALUE",
+ *         ],
+ *       },
  *     },
  *     SpaceStorageSettings: { // DefaultSpaceStorageSettings
  *       DefaultEbsStorageSettings: { // DefaultEbsStorageSettings
@@ -267,6 +275,14 @@ export interface UpdateDomainCommandOutput extends UpdateDomainResponse, __Metad
  *           RepositoryUrl: "STRING_VALUE", // required
  *         },
  *       ],
+ *       EmrSettings: {
+ *         AssumableRoleArns: [
+ *           "STRING_VALUE",
+ *         ],
+ *         ExecutionRoleArns: [
+ *           "STRING_VALUE",
+ *         ],
+ *       },
  *     },
  *     SpaceStorageSettings: {
  *       DefaultEbsStorageSettings: {

diff --git a/clients/client-sagemaker/src/commands/UpdateUserProfileCommand.ts b/clients/client-sagemaker/src/commands/UpdateUserProfileCommand.ts
@@ -179,6 +179,14 @@ export interface UpdateUserProfileCommandOutput extends UpdateUserProfileRespons
  *           RepositoryUrl: "STRING_VALUE", // required
  *         },
  *       ],
+ *       EmrSettings: { // EmrSettings
+ *         AssumableRoleArns: [ // AssumableRoleArns
+ *           "STRING_VALUE",
+ *         ],
+ *         ExecutionRoleArns: [ // ExecutionRoleArns
+ *           "STRING_VALUE",
+ *         ],
+ *       },
  *     },
  *     SpaceStorageSettings: { // DefaultSpaceStorageSettings
  *       DefaultEbsStorageSettings: { // DefaultEbsStorageSettings

diff --git a/clients/client-sagemaker/src/models/models_1.ts b/clients/client-sagemaker/src/models/models_1.ts
@@ -1778,6 +1778,37 @@ export interface CustomPosixUserConfig {
   Gid: number | undefined;
 }
 
+/**
+ * <p>The configuration parameters that specify the IAM roles assumed by the execution role of
+ *             SageMaker (assumable roles) and the cluster instances or job execution environments
+ *             (execution roles or runtime roles) to manage and access resources required for running Amazon EMR
+ *             clusters or Amazon EMR Serverless applications.</p>
+ * @public
+ */
+export interface EmrSettings {
+  /**
+   * <p>An array of Amazon Resource Names (ARNs) of the IAM roles that the execution role of
+   *             SageMaker can assume for performing operations or tasks related to Amazon EMR clusters or Amazon EMR
+   *             Serverless applications. These roles define the permissions and access policies required
+   *             when performing Amazon EMR-related operations, such as listing, connecting to, or terminating
+   *             Amazon EMR clusters or Amazon EMR Serverless applications. They are typically used in
+   *             cross-account access scenarios, where the Amazon EMR resources (clusters or serverless
+   *             applications) are located in a different Amazon Web Services account than the SageMaker
+   *             domain.</p>
+   * @public
+   */
+  AssumableRoleArns?: string[];
+
+  /**
+   * <p>An array of Amazon Resource Names (ARNs) of the IAM roles used by the Amazon EMR cluster instances
+   *             or job execution environments to access other Amazon Web Services services and resources needed during the
+   *             runtime of your Amazon EMR or Amazon EMR Serverless workloads, such as Amazon S3 for data access, Amazon CloudWatch for logging, or other
+   *             Amazon Web Services services based on the particular workload requirements.</p>
+   * @public
+   */
+  ExecutionRoleArns?: string[];
+}
+
 /**
  * <p>The settings for the JupyterLab application.</p>
  * @public
@@ -1807,6 +1838,15 @@ export interface JupyterLabAppSettings {
    * @public
    */
   CodeRepositories?: CodeRepository[];
+
+  /**
+   * <p>The configuration parameters that specify the IAM roles assumed by the execution role of
+   *             SageMaker (assumable roles) and the cluster instances or job execution environments
+   *             (execution roles or runtime roles) to manage and access resources required for running Amazon EMR
+   *             clusters or Amazon EMR Serverless applications.</p>
+   * @public
+   */
+  EmrSettings?: EmrSettings;
 }
 
 /**
@@ -12706,20 +12746,6 @@ export const ProcessingS3CompressionType = {
 export type ProcessingS3CompressionType =
   (typeof ProcessingS3CompressionType)[keyof typeof ProcessingS3CompressionType];
 
-/**
- * @public
- * @enum
- */
-export const ProcessingS3DataType = {
-  MANIFEST_FILE: "ManifestFile",
-  S3_PREFIX: "S3Prefix",
-} as const;
-
-/**
- * @public
- */
-export type ProcessingS3DataType = (typeof ProcessingS3DataType)[keyof typeof ProcessingS3DataType];
-
 /**
  * @internal
  */

diff --git a/clients/client-sagemaker/src/models/models_2.ts b/clients/client-sagemaker/src/models/models_2.ts
@@ -133,7 +133,6 @@ import {
   OutputConfig,
   ProcessingInstanceType,
   ProcessingS3CompressionType,
-  ProcessingS3DataType,
   ProcessingS3UploadMode,
   Processor,
   ProductionVariant,
@@ -152,6 +151,20 @@ import {
   VendorGuidance,
 } from "./models_1";
 
+/**
+ * @public
+ * @enum
+ */
+export const ProcessingS3DataType = {
+  MANIFEST_FILE: "ManifestFile",
+  S3_PREFIX: "S3Prefix",
+} as const;
+
+/**
+ * @public
+ */
+export type ProcessingS3DataType = (typeof ProcessingS3DataType)[keyof typeof ProcessingS3DataType];
+
 /**
  * <p>Configuration for downloading input data from Amazon S3 into the processing container.</p>
  * @public
@@ -9458,21 +9471,6 @@ export interface ModelCardExportArtifacts {
   S3ExportArtifacts: string | undefined;
 }
 
-/**
- * @public
- * @enum
- */
-export const ModelCardExportJobStatus = {
-  COMPLETED: "Completed",
-  FAILED: "Failed",
-  IN_PROGRESS: "InProgress",
-} as const;
-
-/**
- * @public
- */
-export type ModelCardExportJobStatus = (typeof ModelCardExportJobStatus)[keyof typeof ModelCardExportJobStatus];
-
 /**
  * @internal
  */

diff --git a/clients/client-sagemaker/src/models/models_3.ts b/clients/client-sagemaker/src/models/models_3.ts
@@ -147,7 +147,6 @@ import {
   MemberDefinition,
   ModelArtifacts,
   ModelCardExportArtifacts,
-  ModelCardExportJobStatus,
   ModelClientConfig,
   ModelConfiguration,
   NotificationConfiguration,
@@ -181,6 +180,21 @@ import {
   WorkerAccessConfiguration,
 } from "./models_2";
 
+/**
+ * @public
+ * @enum
+ */
+export const ModelCardExportJobStatus = {
+  COMPLETED: "Completed",
+  FAILED: "Failed",
+  IN_PROGRESS: "InProgress",
+} as const;
+
+/**
+ * @public
+ */
+export type ModelCardExportJobStatus = (typeof ModelCardExportJobStatus)[keyof typeof ModelCardExportJobStatus];
+
 /**
  * @public
  */
@@ -11574,25 +11588,6 @@ export const ModelMetadataFilterType = {
  */
 export type ModelMetadataFilterType = (typeof ModelMetadataFilterType)[keyof typeof ModelMetadataFilterType];
 
-/**
- * <p>Part of the search expression. You can specify the name and value
- *           (domain, task, framework, framework version, task, and model).</p>
- * @public
- */
-export interface ModelMetadataFilter {
-  /**
-   * <p>The name of the of the model to filter by.</p>
-   * @public
-   */
-  Name: ModelMetadataFilterType | undefined;
-
-  /**
-   * <p>The value to filter the model metadata.</p>
-   * @public
-   */
-  Value: string | undefined;
-}
-
 /**
  * @internal
  */

diff --git a/clients/client-sagemaker/src/models/models_4.ts b/clients/client-sagemaker/src/models/models_4.ts
@@ -145,7 +145,7 @@ import {
   LambdaStepMetadata,
   LineageType,
   MetricData,
-  ModelMetadataFilter,
+  ModelMetadataFilterType,
   ModelPackageGroupStatus,
   ModelPackageStatusDetails,
   MonitoringExecutionSummary,
@@ -180,6 +180,25 @@ import {
   Workteam,
 } from "./models_3";
 
+/**
+ * <p>Part of the search expression. You can specify the name and value
+ *           (domain, task, framework, framework version, task, and model).</p>
+ * @public
+ */
+export interface ModelMetadataFilter {
+  /**
+   * <p>The name of the of the model to filter by.</p>
+   * @public
+   */
+  Name: ModelMetadataFilterType | undefined;
+
+  /**
+   * <p>The value to filter the model metadata.</p>
+   * @public
+   */
+  Value: string | undefined;
+}
+
 /**
  * <p>One or more filters that searches for the specified resource or resources in
  *           a search. All resource objects that satisfy the expression's condition are