AWS, Cloud Computing

4 Mins Read

A Guide to Integrate Amazon CodeWhisperer with Amazon SageMaker Studio


In the dynamic landscape of machine learning and artificial intelligence, developers and data scientists often face challenges in optimizing and managing their machine learning workflows. Amazon Web Services (AWS) offers comprehensive tools to address these challenges. This blog post explores the integration of Amazon CodeWhisperer with Amazon SageMaker Studio, a powerful combination that enhances collaboration, productivity, and efficiency in machine learning development.


Amazon CodeWhisperer is a collaborative development environment that allows teams to work seamlessly on machine learning projects. On the other hand, Amazon SageMaker Studio is a fully integrated development environment for building, training, and deploying machine learning models.

Integrating Amazon CodeWhisperer with Amazon SageMaker Studio provides a unified platform for teams to collaborate on coding, experiment tracking, and model deployment.

Pioneers in Cloud Consulting & Migration Services

  • Reduced infrastructural costs
  • Accelerated application deployment
Get Started

Purpose of Amazon CodeWhisperer

Amazon CodeWhisperer serves as a central hub for collaborative coding and model development. It offers real-time collaboration, code sharing, and version control features, making it an ideal solution for teams working on machine learning projects. By integrating Amazon CodeWhisperer with Amazon SageMaker Studio, users can leverage the strengths of both platforms, streamlining the development lifecycle.


Before diving into the integration process, ensure that you have the following prerequisites in place:

  • Prepare for Amazon SageMaker Usage

To get started with Amazon SageMaker, you’ll need to take a few preliminary steps. Begin by setting up an AWS account and establishing an administrative user. Detailed instructions can be found in the “Set up Amazon SageMaker prerequisites” section of the Amazon SageMaker User Guide.

  • Establish an Amazon SageMaker Domain

For effective utilization of Amazon SageMaker Studio, it’s crucial to go through the onboarding process for the Amazon SageMaker Domain. This can be achieved through the Amazon SageMaker console or the AWS CLI. Refer to the “Onboard to Amazon SageMaker Domain” section for comprehensive guidance in the Amazon SageMaker User Guide.

  • Grant Amazon CodeWhisperer Permissions in Amazon SageMaker

Ensure seamless integration by adding the necessary permissions related to Amazon CodeWhisperer to your Amazon SageMaker execution role. Create an AWS IAM policy incorporating specific statements and then link this policy to your execution role using AWS IAM or your permission set through the AWS IAM Identity Center. This step is essential for a well-configured and secure collaborative environment.

Step-by-Step Guide

Follow these steps to integrate Amazon CodeWhisperer with Amazon SageMaker Studio:

Step 1: Login into the AWS account and go to the Amazon SageMaker console.

Step 2: Now click on domains


Step 3: Now launch the Amazon SageMaker Studio through the domain


Step 4: Once you have launched the Amazon SageMaker Studio, select the studio classic for the Amazon CodeWhisperer integration. Now run the Amazon SageMaker Studio instance.


Step 5: Add the Amazon CodeWhisperer IAM policy to the Amazon SageMaker Studio domain so that it can access it


Step 6: Once the Amazon SageMaker Studio classic comes in running status, click on open so that it will launch the Studio environment. Under the file section, select the terminal to install the Amazon CodeWhisperer.


Enable the Amazon CodeWhisperer extension in your Amazon SageMaker Studio domain.

Successfully integrate the Amazon CodeWhisperer with Amazon SageMaker Studio.


Step 7: Now refresh the browser and open a new notebook file so that we can test it. We can explore Amazon CodeWhisperer abilities.


Step 8: Example of Amazon CodeWhisperer

As we can see, I started to write a function to fetch the list of objects from an Amazon S3 bucket, and it started giving me the recommendation code.


We can see the output of the recommended program.



The integration of Amazon CodeWhisperer with Amazon SageMaker Studio brings several advantages to machine learning development teams:

  • Collaborative Coding: Teams can collaborate in real-time, share code, and work together on machine learning projects within CodeWhisperer.
  • Unified Environment: The integration provides a unified environment, allowing developers and data scientists to seamlessly switch between Amazon CodeWhisperer and Amazon SageMaker Studio.
  • Experiment Consistency: Experiment details and tracking are synchronized between Amazon CodeWhisperer and Amazon SageMaker Studio, ensuring consistency in the machine learning development process.
  • Efficient Deployment: Deploying models become more efficient with Amazon SageMaker Studio’s deployment tools, which are directly accessible from Amazon CodeWhisperer.


Integrating Amazon CodeWhisperer with Amazon SageMaker Studio offers a powerful collaborative machine learning development solution. With a unified environment, real-time collaboration, and streamlined workflows, development teams can enhance productivity and efficiency in building and deploying machine learning models.

Drop a query if you have any questions regarding Amazon CodeWhisperer or Amazon SageMaker Studio and we will get back to you quickly.

Making IT Networks Enterprise-ready – Cloud Management Services

  • Accelerated cloud migration
  • End-to-end view of the cloud environment
Get Started

About CloudThat

CloudThat is a leading provider of Cloud Training and Consulting services with a global presence in India, the USA, Asia, Europe, and Africa. Specializing in AWS, Microsoft Azure, GCP, VMware, Databricks, and more, the company serves mid-market and enterprise clients, offering comprehensive expertise in Cloud Migration, Data Platforms, DevOps, IoT, AI/ML, and more.

CloudThat is recognized as a top-tier partner with AWS and Microsoft, including the prestigious ‘Think Big’ partner award from AWS and the Microsoft Superstars FY 2023 award in Asia & India. Having trained 650k+ professionals in 500+ cloud certifications and completed 300+ consulting projects globally, CloudThat is an official AWS Advanced Consulting Partner, Microsoft Gold Partner, AWS Training PartnerAWS Migration PartnerAWS Data and Analytics PartnerAWS DevOps Competency PartnerAmazon QuickSight Service Delivery PartnerAmazon EKS Service Delivery PartnerAWS Microsoft Workload PartnersAmazon EC2 Service Delivery Partner, and many more.

To get started, go through our Consultancy page and Managed Services PackageCloudThat’s offerings.


1. Can I use Amazon CodeWhisperer with Amazon SageMaker Studio for non-machine learning projects?

ANS: – Yes, while the integration is particularly beneficial for machine learning projects, Amazon CodeWhisperer can be used for any collaborative coding project.

2. Is there an additional cost for integrating Amazon CodeWhisperer with Amazon SageMaker Studio?

ANS: – Depending on your AWS usage, there may be associated costs with using both Amazon CodeWhisperer and Amazon SageMaker Studio.

3. Can I integrate Amazon CodeWhisperer with Amazon SageMaker Studio for existing projects?

ANS: – Yes, you can enable the integration for existing Amazon CodeWhisperer projects by updating the project settings and linking your Amazon SageMaker Studio environment.

WRITTEN BY Rohit Kumar

Rohit Kumar works as a Research Associate (Infra, Migration, and Security Team) at CloudThat. He is focused on gaining knowledge of the Cloud environment. He has a keen interest in learning and researching emerging technologies.



    Click to Comment

Get The Most Out Of Us

Our support doesn't end here. We have monthly newsletters, study guides, practice questions, and more to assist you in upgrading your cloud career. Subscribe to get them all!