Tuesday, April 29, 2025

Entry your current information and sources by way of Amazon SageMaker Unified Studio, Half 1: AWS Glue Knowledge Catalog and Amazon Redshift

Amazon SageMaker Unified Studio offers a unified setting for information, analytics, machine studying (ML), and AI workloads. A part of the following technology of Amazon SageMaker, SageMaker Unified Studio permits you to uncover your information and put it to work utilizing acquainted AWS instruments to finish end-to-end growth workflows, together with information evaluation, information processing, mannequin coaching, generative AI app growth, and extra, in a single ruled setting. You may create or be a part of tasks to collaborate together with your groups, share AI and analytics artifacts securely, and uncover and use your information saved in numerous information sources by way of Amazon SageMaker Lakehouse.

This sequence of posts demonstrates how one can onboard and entry current AWS information sources utilizing SageMaker Unified Studio. This publish focuses on onboarding current AWS Glue Knowledge Catalog tables and database tables obtainable in Amazon Redshift. Half 2 discusses utilizing Amazon Easy Storage Service (Amazon S3), Amazon Relational Database Service (Amazon RDS), Amazon DynamoDB, and Amazon EMR.

This sequence primarily focuses on the UI expertise. If you happen to want script-based automation, discuss with Bringing current sources into Amazon SageMaker Unified Studio.

Entry administration with SageMaker Unified Studio

The SageMaker Unified Studio authorization mannequin is a hierarchical entry management listing (ACL) based mostly on the useful resource kind equivalent to a website or a mission. For instance, on the area degree, a consumer might need a website proprietor designation and on the mission degree, the consumer might be an proprietor or contributor. You may configure these profiles at AWS Id and Entry Administration (IAM) consumer, single sign-on (SSO) consumer, and SSO group degree.

Every mission has a mission function. When the consumer interacts with sources inside SageMaker Unified Studio, it generates IAM session credentials based mostly on the consumer’s efficient profile within the particular mission context, after which customers can use instruments equivalent to Amazon Athena or Amazon Redshift to question the related information. The mission proprietor can add or take away mission members for his or her mission, create publishing agreements with a website, and publish belongings to a website.

SageMaker Unified Studio might be accessed by IAM customers or SSO authenticated customers, and IAM roles can work together with the SageMaker Unified Studio by way of its APIs.

Answer overview

AWS Lake Formation allows you to outline fine-grained entry management on the Knowledge Catalog, the place you possibly can configure entry at database, desk, row, or column degree or outline permissions with tags. When establishing Lake Formation, you possibly can configure it with hybrid entry mode, the place you get flexibility to selectively allow Lake Formation permissions for particular databases and tables, and proceed utilizing IAM permissions for others. SageMaker Unified Studio helps Lake Formation hybrid mode.

Once you create a mission in SageMaker Unified Studio, an AWS Glue database is added by default as a part of the mission. Property revealed into that database don’t want any further permissions, however if you wish to publish or subscribe belongings from an current AWS Glue database, then it’s good to present specific permissions to SageMaker Unified Studio to have the ability to entry the database and tables. For extra particulars, see Configure Lake Formation permissions for Amazon SageMaker Unified Studio.

Let’s perceive how we will entry current datasets by way of SageMaker Unified Studio.

Stipulations

To run the instruction, you will need to full the next conditions:

  • An AWS account
  • A SageMaker Unified Studio area
  • A SageMaker Unified Studio mission with All capabilities mission profile

Within the SageMaker Unified Studio, choose the mission and navigate to the Mission overview web page. Copy the Mission function ARN as highlighted within the screenshot. This mission function will likely be used additional within the publish to supply permissions on current datasets and sources.

Use current AWS Glue tables

This part has following conditions:

One further prerequisite step is to revoke IAMAllowedPrincipals group permission on each database and desk to implement Lake Formation permission for entry. For detailed instruction see Revoking permission utilizing the Lake Formation console.

To entry current Knowledge Catalog tables in SageMaker Unified Studio, full the next steps:

  1. On the Lake Formation console utilizing the information lake administrator, select Knowledge lake places within the navigation pane and select Register location.
  2. Enter the S3 prefix for Amazon S3 path.
  3. For IAM function, select your Lake Formation information entry IAM function, which isn’t a service linked function.
  4. Choose Lake Formation for Permission mode and select Register location.

  1. On the Lake Formation console, underneath Knowledge Catalog within the navigation pane, select Databases.
  2. Choose the present Knowledge Catalog database.
  3. From the Actions menu, select Grant to grant permissions to the mission function.

  1. For IAM customers and roles, select the mission function.
  2. Choose Named Knowledge Catalog sources, and for Catalogs, select the default catalog.
  3. For Databases, select your current Knowledge Catalog database.

  1. For Database permissions, choose Describe and select Grant.

The subsequent step is to grant the permission on the tables to the mission function.

  1. On the Lake Formation console, underneath Knowledge Catalog within the navigation pane, select Databases.
  2. Choose the present Knowledge Catalog database.
  3. From the Actions menu, select Grant to grant permissions to the mission function.
  4. For IAM customers and roles, select the mission function.
  5. Choose Named Knowledge Catalog sources, and for Catalogs, select the default catalog.
  6. For Databases, select your Knowledge Catalog database.
  7. For Tables, choose the tables that it’s good to present permission to the mission function.

  1. For Desk permissions, choose Choose and Describe.
  2. For Grantable permissions, choose Choose and Describe.
  3. Select Grant.

You need to revoke any current permissions of IAMAllowedPrincipals on the databases and tables inside Lake Formation.

Now let’s confirm that we will entry the present AWS Glue desk from the SageMaker Unified Studio Question Editor.

  1. In SageMaker Unified Studio, navigate to your mission.
  2. On the mission web page, underneath Lakehouse, select Knowledge.
  3. Subsequent to the Knowledge Catalog desk, select the choices menu (three dots), and select Question with Athena.

SageMaker Unified Studio offers a unified JupyterLab expertise throughout completely different languages, together with SQL, PySpark, and Scala Spark. It additionally helps unified entry throughout completely different compute runtimes equivalent to Amazon Redshift and Athena for SQL, Amazon EMR Serverless, Amazon EMR on EC2, and AWS Glue for Spark. To entry the information by way of the unified JupyterLab expertise, full the next steps:

  1. On the SageMaker Unified Studio mission web page, on the highest menu, select Construct, and underneath IDE & APPLICATIONS, select
  2. Watch for the house to be prepared.
  3. Select the plus signal and for Pocket book, select Python 3.
  4. Within the pocket book, swap the connection kind to PySpark, select spark.fineGrained, and question the present Knowledge Catalog desk:
from pyspark.sql import SparkSession spark = SparkSession.builder.getOrCreate() df_sql = spark.sql(""" choose * from retaildb.salesorders """ ) df_sql.present()

Use current Redshift clusters

This part has following conditions:

To usher in current Redshift clusters, observe these steps:

  1. To make use of your provisioned Redshift cluster or a Redshift Serverless workgroup, add both of the next tags (key/worth) to the useful resource:
    1. Add AmazonDataZoneProject: if you wish to enable solely a selected SageMaker Unified Studio mission to entry the Amazon Redshift useful resource. Change with the ID of the mission created in SageMaker Unified Studio.
    2. Add for-use-with-all-datazone-projects: true if you wish to enable all SageMaker Unified Studio tasks to entry the Amazon Redshift useful resource.

  1. So as to add the compute connection in SageMaker Unified Studio, you possibly can authenticate the cluster utilizing both the consumer title and password of the database, IAM credentials, or AWS Secrets and techniques Supervisor. To supply the authentication utilizing Secrets and techniques Supervisor, add both of the next tags. It will allow the present secret to look on the dropdown menu, whereas defining the connection in SageMaker Unified Studio.
    1. AmazonDataZoneProject:
    2. for-use-with-all-datazone-projects: true

Within the following screenshot, you possibly can see the tag configuration part inside Secrets and techniques Supervisor settings for Redshift Serverless compute. To grasp methods to create a secret for a database in a Redshift cluster utilizing Secrets and techniques Supervisor, discuss with Managing Amazon Redshift admin passwords utilizing AWS Secrets and techniques Supervisor.

  1. After the tags are utilized, log in to SageMaker Unified Studio and select the mission.
  2. Go to the Compute part of your mission, and on the Knowledge warehouse tab, select Add compute.
  3. Choose Connect with current compute sources.
  4. Select the compute kind: Amazon Redshift Provisioned cluster or Amazon Redshift Serverless.
  5. Configure the parameters by choosing the present compute and authentication and select Add compute.

The detailed walkthrough course of is illustrated within the following screenshot.

Use Redshift tables with current compute

This part has following conditions:

On this part, we illustrate steps to create a federated connection for an current Amazon Redshift information supply. You may register an current Redshift provisioned cluster in addition to Redshift Serverless with the Knowledge Catalog utilizing SageMaker Unified Studio. This creates a federated multi-level catalog and offers the flexibility to centrally handle permissions to the information with fine-grained entry management utilizing Lake Formation. By mounting Amazon Redshift information within the Knowledge Catalog, you possibly can question it utilizing your most well-liked instruments equivalent to Athena or AWS Glue extract, remodel, and cargo (ETL) with out having to repeat or transfer the information.

Create an Amazon Redshift managed VPC endpoint for Amazon Redshift

Amazon Redshift managed digital non-public cloud (VPC) endpoints use AWS PrivateLink to permit one VPC to privately entry sources in one other VPC as in the event that they had been native to the identical VPC. With an Amazon Redshift managed VPC endpoint, you possibly can hook up with your non-public Redshift cluster with the RA3 occasion kind or Redshift Serverless inside your VPC.

On this part, we clarify methods to create an Amazon Redshift managed VPC endpoint for each Redshift Serverless and an Amazon Redshift provisioned cluster in a single account. The managed VPC endpoint must be created provided that your Redshift provisioned or Redshift Serverless cluster is in a unique VPC than the SageMaker Unified Studio area VPC.

If the SageMaker Unified Studio area account is in a unique account, enable the extra AWS accounts to create cluster endpoints. For steps to authorize your Amazon Redshift provisioned or Redshift Serverless cluster to deploy endpoints in further accounts and grant entry to the cross-account VPC, discuss with Granting entry to a VPC.

Redshift Serverless

For Redshift Serverless, observe these directions.

The frequent follow is to permit port 5439 (Amazon Redshift connectivity port) to the safety group or CIDR vary wherein your consumption workloads run.

  1. Within the safety group related to the Redshift cluster, add an inbound rule with Kind as Redshift, Protocol as TCP, Port vary as 5439 (Amazon Redshift connectivity port), and Supply because the CIDR vary wherein your consumption workloads run.

  1. On the Amazon Redshift console of the workgroup, go to Redshift-managed VPC endpoints.
  2. Select Create endpoint.
  3. Within the Endpoint settings part, select the VPC, related non-public subnet, and safety group created for the SageMaker Unified Studio area account to deploy the endpoint in opposition to.

The next screenshot reveals the Amazon Redshift managed VPC endpoint created for Redshift Serverless.

Redshift provisioned

For Amazon Redshift provisioned, observe these directions:

  1. To implement an Amazon Redshift managed VPC endpoint for a provisioned cluster, it’s good to allow cluster relocation and create subnet teams. Within the cluster subnet group, select the VPC and subnets of the SageMaker Unified Studio area account.
  2. On the Amazon Redshift console, select Configurations within the navigation pane.
  3. Present the endpoint particulars, then select Create endpoint.

Create a federated connection for Amazon Redshift

Full the next steps to create a federated catalog within the Knowledge Catalog to question the information utilizing numerous most well-liked analytics instruments equivalent to Athena, visible ETL in SageMaker Unified Studio, Amazon EMR, and extra:

  1. On the SageMaker Unified Studio console, select your mission.
  2. Select Knowledge within the navigation pane.
  3. Within the information explorer, select the plus signal so as to add a knowledge supply.
  4. Beneath Add a knowledge supply, select Add connection, then select Amazon Redshift.
  5. Enter the next parameters within the connection particulars, and select Add information.
    1. Title: Enter the connection title.
    2. Host: Enter the Amazon Redshift managed VPC endpoint.
    3. Port: Enter the port quantity (Amazon Redshift makes use of 5439 because the default port).
    4. Database: Enter the database title.
    5. Authentication: Select both the database consumer title and password credentials or Secrets and techniques Supervisor.

After the connection is established, you will note that the federated catalog is created, as proven within the following screenshot. This catalog makes use of the AWS Glue connection to connect with Amazon Redshift. The databases, tables, and views are routinely cataloged within the Knowledge Catalog and registered with Lake Formation.

With Athena, information analysts can run federated SQL queries to scan information from a number of information sources in-place with out creating complicated information pipelines or information replication.

Use current Knowledge Catalog tables and Amazon Redshift belongings within the SageMaker Unified Studio enterprise information catalog

You need to use the SageMaker Unified Studio enterprise information catalog to catalog the information throughout your group with enterprise context. To make use of Amazon SageMaker Catalog, you will need to carry your current information belongings into the stock of your mission. Comply with the directions on this part to carry your current Knowledge Catalogs and Amazon Redshift belongings into the mission stock.

Add an current Knowledge Catalog to the mission stock

To complement the asset with enterprise context and share your belongings outdoors your personal mission, you will need to first carry the metadata to SageMaker Catalog. To import the metadata of the belongings into the mission’s stock, it’s good to create a knowledge supply within the mission catalog.

  1. In SageMaker Unified Studio, navigate to the Mission catalog web page throughout the mission.
  2. Select Knowledge sources.
  3. Select CREATE DATA SOURCE.
  4. For Title, present the title of the information supply.
  5. Select AWS Glue (Lakehouse) for Knowledge supply kind.
  6. For Knowledge choice, select the Database title and select Subsequent.
  7. Preserve the remaining as default and select CREATE.
  8. Select RUN to import the metadata.

After the information supply efficiently completes its run, metadata of all the information belongings will get added to the mission’s stock.

Add current Redshift tables and views to the mission stock

Create a knowledge supply to herald the present Redshift tables and views so as to add to the mission’s stock:

  1. In SageMaker Unified Studio, navigate to the Mission catalog throughout the mission.
  2. Select Knowledge sources.
  3. Select CREATE DATA SOURCE.
  4. For Title, present the title of the information supply.
  5. Select Amazon Redshift for Knowledge supply kind.
  6. For Connection, select the title of the Redshift connection.
  7. For Database title, select dev and for Schema, enter public.
  8. Preserve the remaining as default and select CREATE.
  9. Select RUN to import the metadata.

After the information supply efficiently completes its run, metadata of all the information belongings will get added to the mission’s stock.

Conclusion

This publish defined how one can entry current information and sources obtainable within the Knowledge Catalog and Amazon Redshift utilizing SageMaker Unified Studio. SageMaker Unified Studio offers an built-in setting for analytics and AI. Having the ability to entry current datasets obtainable in your AWS account helps cut back operational overhead as a result of customers of your group can entry a standard interface, collaborate, and share datasets. It additionally brings in effectivity for directors as a result of they will handle permissions for domains and tasks in a standard place.

Within the subsequent publish, we are going to exhibit how one can onboard and entry different current information sources equivalent to Amazon S3, Amazon RDS, DynamoDB, and Amazon EMR.


In regards to the Authors

Lakshmi Nair is a Senior Analytics Specialist Options Architect at AWS. She focuses on designing superior analytics techniques throughout industries. She focuses on crafting cloud-based information platforms, enabling real-time streaming, huge information processing, and sturdy information governance. She might be reached through LinkedIn.

Noritaka Sekiyama is a Principal Large Knowledge Architect on the AWS Glue workforce. He’s additionally the writer of the ebook Serverless ETL and Analytics with AWS Glue. He’s liable for constructing software program artifacts to assist clients. In his spare time, he enjoys biking along with his highway bike.

Sakti Mishra is a Principal Knowledge and AI Options Architect at AWS, the place he helps clients modernize their information structure and outline end-to end-data methods, together with information safety, accessibility, governance, and extra. He’s additionally the writer of Simplify Large Knowledge Analytics with Amazon EMR and AWS Licensed Knowledge Engineer Examine Information. Exterior of labor, Sakti enjoys studying new applied sciences, watching films, and visiting locations with household. He might be reached through LinkedIn.

Daiyan Alamgir is a Principal Frontend Engineer on the Amazon SageMaker Unified Studio workforce based mostly in New York.

Vipin Mohan is a Principal Product Supervisor at AWS, main the launch of generative AI capabilities in Amazon SageMaker Unified Studio. He’s dedicated to shaping impactful merchandise by working backward from buyer insights, championing user-focused options, and delivering scalable outcomes.

Chanu Damarla is a Principal Product Supervisor on the Amazon SageMaker Unified Studio workforce. He works with clients across the globe to translate enterprise and technical necessities into merchandise that delight clients and allow them to be extra productive with their information, analytics, and AI.

Related Articles

LEAVE A REPLY

Please enter your comment!
Please enter your name here

Latest Articles