sagemaker-python-sdk
sagemaker-python-sdk copied to clipboard
LocalSession class does not accept default_bucket argument
Describe the bug
LocalSession class does not accept default_bucket argument and it always uses sagemaker-{region}-{aws-account-id}
bucket
To reproduce
Creating a local session while passing default_bucket
fails:
>>> import sagemaker
>>> sagemaker.local.LocalSession(default_bucket='bucketname')
TypeError: __init__() got an unexpected keyword argument 'default_bucket'
Expected behavior
LocalSession
should accept default_bucket
, pass it to Session
and store artifacts in given bucket.
System information A description of your system. Please provide:
- SageMaker Python SDK version: 2.75.1
- Python version: 3.8.10
Additional context I've already opened a pull request for this: https://github.com/aws/sagemaker-python-sdk/pull/3171/files