Configure Amazon EMR CloudFormation templates in the Service Catalog - Amazon SageMaker

Configure Amazon EMR CloudFormation templates in the Service Catalog

This topic assumes administrators are familiar with AWS CloudFormation, portfolios and products in AWS Service Catalog, as well as Amazon EMR.

To simplify the creation of Amazon EMR clusters from Studio, administrators can register an Amazon EMR CloudFormation template as a product in an AWS Service Catalog portfolio. To make the template available to data scientists, they must associate the portfolio with the SageMaker execution role used in Studio or Studio Classic. Finally, to allow users to discover templates, provision clusters, and connect to Amazon EMR clusters from Studio or Studio Classic, administrators need to set appropriate access permissions.

The Amazon EMR AWS CloudFormation templates can allow end-users to customize various cluster aspects. For example, administrators can define an approved list of instance types that users can choose from when creating a cluster.

The following instructions use end-to-end CloudFormation stacks to setup a Studio or Studio Classic domain, a user profile, a Service Catalog portfolio, and populate an Amazon EMR launch template. The following steps highlight the specific settings that administrators must apply in their end-to-end stack to enable Studio or Studio Classic to access Service Catalog products and provision Amazon EMR clusters.

Note

The GitHub repository aws-samples/sagemaker-studio-emr contains example end-to-end CloudFormation stacks that deploy the necessary IAM roles, networking, SageMaker domain, user profile, Service Catalog portfolio, and add an Amazon EMR launch CloudFormation template. The templates provide different authentication options between Studio or Studio Classic and the Amazon EMR cluster. In these example templates, the parent CloudFormation stack passes SageMaker VPC, security group, and subnet parameters to the Amazon EMR cluster template.

The sagemaker-studio-emr/cloudformation/emr_servicecatalog_templates repository contains various sample Amazon EMR CloudFormation launch templates, including options for single account and cross-account deployments.

Refer to Connect to an Amazon EMR cluster from SageMaker Studio or Studio Classic for details on the authentication methods you can use to connect to an Amazon EMR cluster.

To let data scientists discover Amazon EMR CloudFormation templates and provision clusters from Studio or Studio Classic, follow these steps.

Step 0: Check your networking and prepare your CloudFormation stack

Before you start:

  • Ensure that you have reviewed the networking and security requirements in Configure network access for your Amazon EMR cluster.

  • You must have an existing end-to-end CloudFormation stack that supports the authentication method of your choice. You can find examples of such CloudFormation templates in the aws-samples/sagemaker-studio-emr GitHub repository. The following steps highlight the specific configurations in your end-to-end stack to enable the use of Amazon EMR templates within Studio or Studio Classic.

Step 1: Associate your Service Catalog portfolio with SageMaker

In your Service Catalog portfolio, associate your portfolio ID with the SageMaker execution role accessing your cluster.

To do so, add the following section (here in YAML format) to your stack. This grants the SageMaker execution role access to the specified Service Catalog portfolio containing products like Amazon EMR templates. It allows roles assumed by SageMaker to launch those products.

Replace SageMakerExecutionRole.Arn and SageMakerStudioEMRProductPortfolio.ID with their actual values.

SageMakerStudioEMRProductPortfolioPrincipalAssociation: Type: AWS::ServiceCatalog::PortfolioPrincipalAssociation Properties: PrincipalARN: SageMakerExecutionRole.Arn PortfolioId: SageMakerStudioEMRProductPortfolio.ID PrincipalType: IAM

For details on the required set of IAM permissions, see the permissions section.

Step 2: Reference an Amazon EMR template in a Service Catalog product

In a Service Catalog product of your portfolio, reference an Amazon EMR template resource and ensure its visibility in Studio or Studio Classic.

To do so, reference the Amazon EMR template resource in the Service Catalog product definition, and then add the following tag key "sagemaker:studio-visibility:emr" set to the value "true" (see the example in YAML format).

In the Service Catalog product definition, the AWS CloudFormation template of the cluster is referenced via URL. The additional tag set to true ensures the visibility of the Amazon EMR templates in Studio or Studio Classic.

Note

The Amazon EMR template referenced by the provided URL in the example does not enforce any authentication requirements when launched. This option is meant for demonstration and learning purposes. It is not recommended in a production environment.

SMStudioEMRNoAuthProduct: Type: AWS::ServiceCatalog::CloudFormationProduct Properties: Owner: AWS Name: SageMaker Studio Domain No Auth EMR ProvisioningArtifactParameters: - Name: SageMaker Studio Domain No Auth EMR Description: Provisions a SageMaker domain and No Auth EMR Cluster Info: LoadTemplateFromURL: Link to your CloudFormation template. For example, https://aws-blogs-artifacts-public.s3.amazonaws.com/artifacts/astra-m4-sagemaker/end-to-end/CFN-EMR-NoStudioNoAuthTemplate-v3.yaml Tags: - Key: "sagemaker:studio-visibility:emr" Value: "true"

Step 3: Parameterize the Amazon EMR CloudFormation template

The CloudFormation template used to define the Amazon EMR cluster within the Service Catalog product allows administrators to specify configurable parameters. Administrators can define Default values and AllowedValues ranges for these parameters within the template's Parameters section. During the cluster launch process, data scientists can provide custom inputs or make selections from those predefined options to customize certain aspects of their Amazon EMR cluster.

The following example illustrates additional input parameters that administrators can set when creating an Amazon EMR template.

"Parameters": { "EmrClusterName": { "Type": "String", "Description": "EMR cluster Name." }, "MasterInstanceType": { "Type": "String", "Description": "Instance type of the EMR master node.", "Default": "m5.xlarge", "AllowedValues": [ "m5.xlarge", "m5.2xlarge", "m5.4xlarge" ] }, "CoreInstanceType": { "Type": "String", "Description": "Instance type of the EMR core nodes.", "Default": "m5.xlarge", "AllowedValues": [ "m5.xlarge", "m5.2xlarge", "m5.4xlarge", "m3.medium", "m3.large", "m3.xlarge", "m3.2xlarge" ] }, "CoreInstanceCount": { "Type": "String", "Description": "Number of core instances in the EMR cluster.", "Default": "2", "AllowedValues": [ "2", "5", "10" ] }, "EmrReleaseVersion": { "Type": "String", "Description": "The release version of EMR to launch.", "Default": "emr-5.33.1", "AllowedValues": [ "emr-5.33.1", "emr-6.4.0" ] } }

After administrators have made the Amazon EMR CloudFormation templates available within Studio, data scientists can use them to self-provision Amazon EMR clusters. The Parameters section defined in the template translates into input fields on the cluster creation form within Studio or Studio Classic. For each parameter, data scientists can either enter a custom value into the input box or select from the predefined options listed in a dropdown menu, which corresponds to the AllowedValues specified in the template.

The following illustration shows the dynamic form assembled from a CloudFormation Amazon EMR template to create an Amazon EMR cluster in Studio or Studio Classic.

Illustration of a dynamic form assembled from a CloudFormation Amazon EMR template to create an Amazon EMR cluster from Studio or Studio Classic.

Visit Launch an Amazon EMR cluster from Studio or Studio Classic to learn about how to launch a cluster from Studio or Studio Classic using those Amazon EMR templates.

Step 4: Set up the permissions to enable listing and launching Amazon EMR clusters from Studio

Last, attach the required IAM permissions to enable listing existing running Amazon EMR clusters and self-provisioning new clusters from Studio or Studio Classic.

The role(s) to which you must add those permissions depends on whether Studio or Studio Classic and Amazon EMR are deployed in the same account (choose Single Account) or in different accounts (choose Cross account).

Important

You can only discover and connect to Amazon EMR clusters for JupyterLab and Studio Classic applications that are launched from private spaces. Ensure that the Amazon EMR clusters are located in the same AWS region as your Studio environment.

If your Amazon EMR clusters and Studio or Studio Classic are deployed in the same AWS account, attach the following permissions to the SageMaker execution role accessing your cluster.

  1. Step 1: Retrieve the ARN of the SageMaker execution role used by your private space.

    For information on spaces and execution roles in SageMaker, see Understanding domain space permissions and execution roles.

    For more information about how to retrieve the ARN of SageMaker's execution role, see Get your execution role.

  2. Step 2: Attach the following permissions to the SageMaker execution role accessing your Amazon EMR clusters.

    1. Navigate to the IAM console.

    2. Choose Roles and then search for your execution role by name in the Search field. The role name is the last part of the ARN, after the last forward slash (/).

    3. Follow the link to your role.

    4. Choose Add permissions and then Create inline policy.

    5. In the JSON tab, add the Amazon EMR permissions allowing Amazon EMR access and operations. For details on the policy document, see List Amazon EMR policies in Reference policies. Replace the region, and accountID with their actual values before copying the list of statements to the inline policy of your role.

    6. Choose Next and then provide a Policy name.

    7. Choose Create policy.

    8. Repeat the Create inline policy step to add another policy granting the execution role the permissions to provision new Amazon EMR clusters using AWS CloudFormation templates. For details on the policy document, see Create Amazon EMRclusters policies in Reference policies. Replace the region and accountID with their actual values before copying the list of statements to the inline policy of your role.

Note

Users of role-based access control (RBAC) connectivity to Amazon EMR clusters should also refer to Configure runtime role authentication when your Amazon EMR cluster and Studio are in the same account.

Before you get started, retrieve the ARN of the SageMaker execution role used by your private space.

For information on spaces and execution roles in SageMaker, see Understanding domain space permissions and execution roles.

For more information about how to retrieve the ARN of SageMaker's execution role, see Get your execution role.

If your Amazon EMR clusters and Studio or Studio Classic are deployed in separate AWS accounts, you configure the permissions on both accounts.

Note

Users of role-based access control (RBAC) connectivity to Amazon EMR clusters should also refer to Configure runtime role authentication when your cluster and Studio are in different accounts.

On the Amazon EMR cluster account

Follow these steps to create the necessary roles and policies on the account where Amazon EMR is deployed, also referred to as the trusting account:

  1. Step 1: Retrieve the ARN of the service role of your Amazon EMR cluster.

    To learn about how to find the ARN of the service role of a cluster, see Configure IAM service roles for Amazon EMR permissions to AWS services and resources.

  2. Step 2: Create a custom IAM role named AssumableRole with the following configuration:

    • Permissions: Grant the necessary permissions to AssumableRole to allow accessing Amazon EMR resources. This role is also known as an Access role in scenarios involving cross-account access.

    • Trust relationship: Configure the trust policy for AssumableRole to allow assuming the execution role (The SageMakerExecutionRole in the cross-account diagram) from the Studio account that requires access.

    By assuming the role, Studio or Studio Classic can gain temporary access to the permissions it needs in Amazon EMR.

    For detailed instructions on how to create a new AssumableRole in your Amazon EMR AWS account, follow these steps:

    1. Navigate to the IAM console.

    2. In the left navigation pane, choose Policy, and then Create policy.

    3. In the JSON tab, add the Amazon EMR permissions allowing Amazon EMR access and operations. For details on the policy document, see List Amazon EMR policies in Reference policies. Replace the region, and accountID with their actual values before copying the list of statements to the inline policy of your role.

    4. Choose Next and then provide a Policy name.

    5. Choose Create policy.

    6. In the left navigation pane, choose Roles and then Create role.

    7. On the Create role page, choose Custom trust policy as the trusted entity.

    8. Paste in the following JSON document in the Custom trust policy section and then choose Next.

      For users of Studio and JupyterLab

      Replace studio-account with the Studio account ID, and AmazonSageMaker-ExecutionRole with the execution role used by your JupyterLab space.

      { "Version": "2012-10-17", "Statement": [ { "Effect": "Allow", "Principal": { "AWS": "arn:aws:iam::studio-account:role/service-role/AmazonSageMaker-ExecutionRole" }, "Action": "sts:AssumeRole" } ] }
      For users of Studio Classic

      Replace studio-account with the Studio Classic account ID.

      { "Version": "2012-10-17", "Statement": [ { "Effect": "Allow", "Principal": { "AWS": "arn:aws:iam::studio-account:root" }, "Action": "sts:AssumeRole" } ] }
    9. In the Add permissions page, add the permission you just created and then choose Next.

    10. On the Review page, enter a name for the role such as AssumableRole and an optional description.

    11. Review the role details and choose Create role.

    For more information about creating a role on an AWS account, see Creating an IAM role (console).

On the Studio account

On the account where Studio is deployed, also referred to as the trusted account, update the SageMaker execution role accessing your clusters with the required permissions to access resources in the trusting account.

  1. Step 1: Retrieve the ARN of the SageMaker execution role used by your private space.

    For information on spaces and execution roles in SageMaker, see Understanding domain space permissions and execution roles.

    For more information about how to retrieve the ARN of SageMaker's execution role, see Get your execution role.

  2. Step 2: Attach the following permissions to the SageMaker execution role accessing your Amazon EMR clusters.

    1. Navigate to the IAM console.

    2. Choose Roles and then search for your execution role by name in the Search field. The role name is the last part of the ARN, after the last forward slash (/).

    3. Follow the link to your role.

    4. Choose Add permissions and then Create inline policy.

    5. In the JSON tab, add the inline policy granting the role permissions to update the domains, user profiles, and spaces. For details on the policy document, see Domain, user profile, and space update actions policy in Reference policies. Replace the region and accountID with their actual values before copying the list of statements to the inline policy of your role.

    6. Choose Next and then provide a Policy name.

    7. Choose Create policy.

    8. Repeat the Create inline policy step to add another policy granting the execution role the permissions to assume the AssumableRole and then perform actions permitted by the role's access policy. Replace emr-account with the Amazon EMR account ID, and AssumableRole with the name of the assumable role created in the Amazon EMR account.

      { "Version": "2012-10-17", "Statement": [ { "Sid": "AllowRoleAssumptionForCrossAccountDiscovery", "Effect": "Allow", "Action": "sts:AssumeRole", "Resource": ["arn:aws:iam::emr-account:role/AssumableRole" ] }] }
    9. Repeat the Create inline policy step to add another policy granting the execution role the permissions to provision new Amazon EMR clusters using AWS CloudFormation templates. For details on the policy document, see Create Amazon EMRclusters policies in Reference policies. Replace the region and accountID with their actual values before copying the list of statements to the inline policy of your role.

    10. (Optional) To allow listing Amazon EMR clusters deployed in the same account as Studio, add an additional inline policy to your Studio execution role as defined in List Amazon EMR policies in Reference policies.

  3. Step 3: Associate your assumable role(s) (access role) with your domain or user profile. JupyterLab users in Studio can use the SageMaker console or the provided script.

    Choose the tab that corresponds to your use case.

    Associate your assumable roles in JupyterLab using the SageMaker console

    To associate your assumable roles with your user profile or domain using the SageMaker console:

    1. Navigate to the SageMaker console at https://console.aws.amazon.com/sagemaker/.

    2. In the left navigation pane, choose domain, and then select the domain using the SageMaker execution role whose permissions you updated.

      • To add your assumable role(s) (access role) to your domain: In the App Configurations tab of the Domain details page, navigate to the JupyterLab section.

      • To add your assumable role(s) (access role) to your user profile: On the Domain details page, chose the User profiles tab, select the user profile using the SageMaker execution role whose permissions you updated. In the App Configurations tab, navigate to the JupyterLab section.

    3. Choose Edit and add the ARNs of your assumable role (access role).

    4. Choose Submit.

    Associate your assumable roles in JupyterLab using a Python script

    In a JupyterLab application started from a space using the SageMaker execution role whose permissions you updated, run the following command in a terminal. Replace the domainID, user-profile-name, emr-accountID, and AssumableRole ( EMRServiceRole for RBAC runtime roles) with their proper values. This code snippet updates the user profile settings for a specific user profile (use client.update_userprofile) or domain settings (use client.update_domain) within a SageMaker domain. Specifically, it allows the JupyterLab application to assume a particular IAM role (AssumableRole) for running Amazon EMR clusters within the Amazon EMR account.

    import botocore.session import json sess = botocore.session.get_session() client = sess.create_client('sagemaker') client.update_userprofile( DomainId="domainID", UserProfileName="user-profile-name", DefaultUserSettings={ 'JupyterLabAppSettings': { 'EmrSettings': { 'AssumableRoleArns': ["arn:aws:iam::emr-accountID:role/AssumableRole"], 'ExecutionRoleArns': ["arn:aws:iam::emr-accountID:role/EMRServiceRole", "arn:aws:iam::emr-accountID:role/AnotherServiceRole"] } } }) resp = client.describe_user_profile(DomainId="domainID", UserProfileName=user-profile-name") resp['CreationTime'] = str(resp['CreationTime']) resp['LastModifiedTime'] = str(resp['LastModifiedTime']) print(json.dumps(resp, indent=2))
    For users of Studio Classic

    Provide the ARN of the AssumableRole to your Studio Classic execution role. The ARN is loaded by the Jupyter server at launch. The execution role used by Studio assumes that cross-account role to discover and connect to Amazon EMR clusters in the trusting account.

    You can specify this information by using Lifecycle Configuration (LCC) scripts. You can attach the LCC to your domain or a specific user profile. The LCC script that you use must be a JupyterServer configuration. For more information on how to create an LCC script, see Use Lifecycle Configurations with Studio Classic.

    The following is an example LCC script. To modify the script, replace AssumableRole and emr-account with their respective values. The number of cross-accounts is limited to five.

    # This script creates the file that informs Studio Classic that the role "arn:aws:iam::emr-account:role/AssumableRole" in remote account "emr-account" must be assumed to list and describe Amazon EMR clusters in the remote account. #!/bin/bash set -eux FILE_DIRECTORY="/home/sagemaker-user/.cross-account-configuration-DO_NOT_DELETE" FILE_NAME="emr-discovery-iam-role-arns-DO_NOT_DELETE.json" FILE="$FILE_DIRECTORY/$FILE_NAME" mkdir -p $FILE_DIRECTORY cat > "$FILE" <<- "EOF" { emr-cross-account1: "arn:aws:iam::emr-cross-account1:role/AssumableRole", emr-cross-account2: "arn:aws:iam::emr-cross-account2:role/AssumableRole" } EOF

    After the LCC runs and the files are written, the server reads the file /home/sagemaker-user/.cross-account-configuration-DO_NOT_DELETE/emr-discovery-iam-role-arns-DO_NOT_DELETE.json and stores the cross-account ARN.