Skip to content

Commit 169d8d1

Browse files
committed
change: update docstring for _upload_code
1 parent 613ba6a commit 169d8d1

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/sagemaker/model.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -406,7 +406,14 @@ def prepare_container_def(
406406
)
407407

408408
def _upload_code(self, key_prefix, repack=False):
409-
"""Placeholder Docstring"""
409+
"""Uploads code to S3 to be used with script mode with SageMaker inference.
410+
411+
Args:
412+
key_prefix (str): The S3 key associated with the ``code_location`` parameter of the
413+
``Model`` class.
414+
repack (bool): Optional. Set to ``True`` to indicate that the source code and model
415+
artifact should be repackaged into a new S3 object. (default: False).
416+
"""
410417
local_code = utils.get_config_value("local.local_code", self.sagemaker_session.config)
411418
if (self.sagemaker_session.local_mode and local_code) or self.entry_point is None:
412419
self.uploaded_code = None

0 commit comments

Comments
 (0)