New to Claude Skills? Learn how to install them →

Shuggingface on GitHub

SageMaker IAM Preflight

Free

Ensure a valid SageMaker execution role exists before deployment.

Get this skill

Free · Opens the source repo

What SageMaker IAM Preflight does

The SageMaker IAM Preflight skill is designed to streamline the process of preparing for SageMaker resource deployment by ensuring that a valid execution role exists. Many developers encounter deployment failures due to the absence of a usable IAM role, particularly when scripts attempt to create new roles without checking for existing ones. This skill provides a systematic approach to validate, discover, and create IAM roles only when necessary, thus preventing unnecessary errors and improving deployment efficiency.

When using this skill, users can run a series of Python scripts that first check if a specific role has been provided. If not, the skill will automatically discover existing roles that match common naming conventions associated with SageMaker execution roles. It ranks these roles by their last-used date to ensure that the most relevant and actively maintained role is selected. This process helps avoid the pitfalls of using outdated or incorrect IAM roles, which can lead to access issues during deployment.

If no suitable role is found, the skill will attempt to create a new execution role, provided the user has the necessary IAM permissions. In cases where the user is authenticated via SSO and lacks the permissions to create roles, the skill will clearly inform the user of the issue and provide actionable steps to resolve it. This level of transparency helps users quickly address permission-related obstacles, making the deployment process smoother.

This skill is particularly useful for data scientists and machine learning engineers who frequently work with AWS SageMaker and need to ensure that their execution roles are properly configured before starting training jobs or deploying models. By automating the role validation and creation process, users can focus more on their models and less on IAM configurations.

When to use it

Use this skill whenever you're about to create a SageMaker endpoint, model, training job, or any resource that requires an execution role.

When not to use it

Avoid this skill if you're already managing IAM roles effectively or if you have sufficient permissions to create roles without needing validation.

What you can build with it

Deploying a New Model

Before deploying a new model on SageMaker, use this skill to ensure that an appropriate execution role is available.

Creating a SageMaker Endpoint

When setting up a SageMaker endpoint, validate the execution role to avoid deployment errors related to IAM permissions.

Troubleshooting Access Denied Errors

If you encounter AccessDenied errors during deployment, run this skill to check for existing roles and permissions.

How to install SageMaker IAM Preflight

View source

1. Install with the skills CLI

npx skills add huggingface/skills/hf-cloud-sagemaker-iam-preflight --agent claude-code

2. Or install it manually

Download the skill folder and drop it into ~/.claude/skills/ for all projects, or .claude/skills/ to scope it to one repo. Restart Claude Code so it picks up the new skill.

Anthropic's agentic coding CLI, and the reference implementation of Agent Skills. Drop a skill folder into ~/.claude/skills and Claude Code loads it automatically whenever a task matches the skill's description. Claude Code docs

Inside SKILL.md

Written by huggingface

SageMaker IAM Preflight

Every SageMaker resource needs an execution role — the IAM role SageMaker assumes to read model artifacts from S3, pull serving containers from ECR, and write logs. Most deployments fail here because the script tried to create a new role without checking if a usable one already existed, then blew up because the caller is an SSO principal.

This skill encodes the right order: discover, validate, only create if necessary.

Running the helpers (cross-platform)

The helpers are Python so they run identically on Windows, macOS, and Linux:

python3 scripts/check_role.py        # macOS / Linux
python  scripts/check_role.py        # Windows (PowerShell / cmd)

Run them from the shell where the AWS CLI already works — i.e. wherever aws sts get-caller-identity succeeds. The script shells out to that same aws binary and inherits the shell's profile, region, SSO session, proxy, and credential chain.

Windows / WSL / Git Bash caveat. Do not invoke these through a Bash shim (WSL, Git Bash, MSYS) on Windows. Those Bash environments frequently do not share the Windows AWS config, credentials, SSO sessions, environment variables, or proxy settings — so aws sts get-caller-identity fails inside Bash even when it works natively in PowerShell. (This is exactly why the old .sh helpers failed on Windows and were replaced with Python.) If you're in PowerShell, run python ...\check_role.py directly in PowerShell. If the helper still can't see your identity, run the same discovery natively (see "Native AWS CLI equivalent" below) in the shell where aws sts get-caller-identity returns your ARN.

Order of operations

Step 1 — Did the user provide a role?

Validate that one specifically:

python3 scripts/check_role.py "<role-name-or-arn>"

On success it prints the ARN to stdout (exit 0). On failure it logs why on stderr. Don't try to silently fix a broken role — surface the problem.

Step 2 — Discover existing roles

python3 scripts/check_role.py

Lists roles matching common SageMaker patterns (AmazonSageMaker-ExecutionRole-*, SageMakerExecutionRole*, etc.), ranks by last-used date (most recent first), validates trust policy in that order, returns the first usable ARN. Most accounts that have used SageMaker before already have one.

Why rank by last-used: in accounts with multiple roles (auto-generated 2021 role + manual project role + etc.), the alphabetically-first one is rarely the actively-maintained one. The most-recently-used role is more likely to have current policies — including cross-account ECR pull. The script prints the ranking so you can see which got picked.

IAM frequently reports no RoleLastUsed at all (tracking only covers recent activity). When every candidate ties at "never used", the script falls back to newest creation date — a newer role is more likely to have current policies than a 2021 leftover.

Step 3 — Create, only if discovery found nothing

If the user can create (has IAM permissions):

python3 scripts/create_role.py "<role-name>" "<model-bucket>"

Second arg scopes S3 access to a specific bucket. Omit if unknown; script warns and the user can update the policy later.

If the user cannot create (SSO principal — hf-cloud-aws-context-discovery will have flagged this):

Stop and surface this clearly. Don't retry alternative IAM operations hoping one works:

I can't find an existing SageMaker execution role, and you're authenticated via SSO so you can't create one directly. Please either:

  • Ask your AWS admin for a SageMaker execution role ARN, or
  • Have them grant your SSO permission set iam:CreateRole, iam:AttachRolePolicy, iam:PutRolePolicy

Specific instructions get unblocked fast; vague "permission denied" messages don't.

What "validated" means

A role is usable when (1) it exists, (2) its trust policy allows sagemaker.amazonaws.com to sts:AssumeRole — see references/trust-policy.json for the canonical form.

check_role.py verifies these two. It does not deep-check permissions because comprehensive analysis is expensive (iam:SimulatePrincipalPolicy per action) and most existing SageMaker roles are over-permissioned via AmazonSageMakerFullAccess. If you suspect a permissions issue at deploy time, the deployment error will tell you which action was denied — fix it then, not preemptively.

Minimum permissions

references/minimum-permissions.json covers what SageMaker actually needs:

  • s3:GetObject + s3:ListBucket on the model artifact bucket
  • ECR pull permissions
  • CloudWatch logs and metrics

Layered on top of AmazonSageMakerFullAccess (attached by create_role.py). Replace REPLACE_WITH_MODEL_BUCKET in the template with the actual bucket name — create_role.py does this automatically when given a bucket as its second argument.

Native AWS CLI equivalent (fallback)

If the Python helper can't run or can't see your identity (rare — usually a broken PATH or running under a Bash shim that lacks AWS context), do the same preflight by hand in the shell where aws sts get-caller-identity works. The logic is just AWS CLI calls; the helper exists only to bundle and rank them.

PowerShell:

# 1. List candidate SageMaker roles
aws iam list-roles --query "Roles[?contains(RoleName,'SageMaker') || contains(RoleName,'sagemaker')]" --output json

# 2. For each candidate, confirm the trust policy allows sagemaker.amazonaws.com
aws iam get-role --role-name <role-name> --query "Role.AssumeRolePolicyDocument" --output json

# 3. Prefer the most-recently-used role with SageMaker-execution naming
#    (LastUsedDate is often None for every role — then prefer newest CreateDate)
aws iam get-role --role-name <role-name> --query "Role.[RoleLastUsed.LastUsedDate, CreateDate]" --output text

Pick the most-recently-used role whose trust policy contains sagemaker.amazonaws.com. Use the resulting ARN exactly as if check_role.py had returned it. Bash/macOS/Linux use the same commands.

Frequently asked questions about SageMaker IAM Preflight

Similar skills