aws / aws/sagemaker-python-sdk
SageMaker Local Mode: Error for network sagemaker-local
- Dominant language
- Python
- Stars
- 2.3k
- Forks
- 1.3k
- Avg merge
- 2d 2h
- Merged PRs (30d)
- 32
Description
**Describe the bug**
Running the ScriptProcessor in local mode on either a desktop or notebook instance results in a RuntimeError when it reaches the 'docker compose up' command.
**To reproduce**
First run commands:
docker network rm sagemaker-local
docker network create sagemaker-local
```import boto3
sess = boto3.Session()
sm = sess.client("sagemaker")
s3_resource = boto3.resource("s3")
sagemaker_session = LocalSession()
config = {
"local": {
"local_code": True,
}
}
sagemaker_session.config = config
sagemaker_session.s3_resource = s3_resource
sagemaker_session.s3_client = sm
network_config = NetworkConfig(
enable_network_isolation=False,
security_group_ids=security_group_ids, # security group ids
subnets=subnets, # subnets
encrypt_inter_container_traffic=True,
)
script_processor = ScriptProcessor(
command=["bash", "process.sh"],
image_uri=image, # custom Docker image
role=role,
instance_count=1,
instance_type="local",
sagemaker_session=sagemaker_session,
network_config=network_config,
).run(...)
```
Just running 'docker rm sagemaker-local' and then the code in local mode is a workaround for the issue as of now.
**Expected behavior**
For the script processing job to run and output properly.
**Screenshots or logs**


**System information**
A description of your system. Please provide:
- **SageMaker Python SDK version**:2.217.0
- **Python version**:3.11.4
- **Custom Docker image**:Y
**Additional context**
Add any other context about the problem here.
Contributor guide
Assessment
This issue has not been assessed yet.