sagemaker-python-sdk icon indicating copy to clipboard operation
sagemaker-python-sdk copied to clipboard

LocalSession class does not accept default_bucket argument

Open hossein-jazayeri opened this issue 2 years ago • 0 comments

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

hossein-jazayeri avatar Jun 15 '22 10:06 hossein-jazayeri