-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Make code_location to be S3 URI instead of bucket in training_config() #501
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Codecov Report
@@ Coverage Diff @@
## master #501 +/- ##
==========================================
+ Coverage 94.29% 94.29% +<.01%
==========================================
Files 59 59
Lines 4610 4613 +3
==========================================
+ Hits 4347 4350 +3
Misses 263 263
Continue to review full report at Codecov.
|
src/sagemaker/estimator.py
Outdated
If not specified, default bucket created by ``sagemaker.session.Session`` is used. | ||
The default S3 path is default_bucket/job-name/source/. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
including s3 protocol makes this clearer (s3://default_bucket/...)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Updated.
src/sagemaker/estimator.py
Outdated
@@ -670,8 +670,9 @@ def __init__(self, entry_point, source_dir=None, hyperparameters=None, enable_cl | |||
training jobs. This will be ignored for now and removed in a further release. | |||
container_log_level (int): Log level to use within the container (default: logging.INFO). | |||
Valid values are defined in the Python logging module. | |||
code_location (str): Name of the S3 bucket where custom code is uploaded (default: None). | |||
code_location (str): The S3 URI where custom code is uploaded (default: None). |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
let's fix this to make it clear we expect a prefix and will append /source/sourcedir.tar.gz
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Updated.
Issue #, if available:
Description of changes:
The docstring for code_location is not correct. code_location can be a S3 URI, not just a bucket. And I need to change related Airflow API because it treats code_location as a bucket.
Merge Checklist
Put an
x
in the boxes that apply. You can also fill these out after creating the PR. If you're unsure about any of them, don't hesitate to ask. We're here to help! This is simply a reminder of what we are going to look for before merging your pull request.By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.