aws / aws/sagemaker-python-sdk

Which permissions are needed to allow us to use the NetworkConfig for processing Jobs?

Closed
#3,026 1 comment 0 reactions 1 assignee Claimed by @nargokul View on GitHub
component: processing type: documentation type: question
Dominant language
Python
Stars
2.3k
Forks
1.3k
Avg merge
1d 22h
Merged PRs (30d)
35

Description

Since there was no activity in the discussions, I move this here in the hopes of some input.

### Discussed in https://github.com/aws/sagemaker-python-sdk/discussions/2912

Originally posted by **JimFawkes** February 8, 2022
We would like to run a SM processing Job in our own VPC & Security Group. We assume that that is what the `NetworkConfig is setting is for. Unfortunately we are running into a rather unhelpful error:
```
UnexpectedStatusException: Error for Processing job test-job-2022-02-08-16-14-38-489: Failed. Reason: InternalServerError: We encountered an internal error. Please try again.
```

The following code produces the error:
```python
import boto3
from sagemaker import Session
from sagemaker.sklearn.processing import ScriptProcessor
from sagemaker.network import NetworkConfig

sm = boto3.client("sagemaker")

session = Session(default_bucket="our-s3-bucket-name", sagemaker_client=sm)

networking = NetworkConfig(
security_group_ids=["sg-12345"],
subnets=["subnet-123345"]
)

script_processor = ScriptProcessor(
command=["python3"],
image_uri="12345.dkr.ecr.eu-west-1.amazonaws.com/abc/efg",
role="arn:aws:iam::12345:role/custom-service-role",
instance_count=1,
instance_type="ml.m5.large",
base_job_name="test-job",
sagemaker_session=session,
env=env_vars,
network_config=networking
)

script_processor.run(
code=script_location,
)
```
We assumed that the error is due to a missing permission in the sagemaker service role we pass to the job but could not figure out which it is.

Any tips to help us debug the error are greatly appreciated.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.