diff --git a/doc/model.rst b/doc/model.rst index e2ab85e134..2846c58d0b 100644 --- a/doc/model.rst +++ b/doc/model.rst @@ -5,3 +5,13 @@ Model :members: :undoc-members: :show-inheritance: + +.. autoclass:: sagemaker.model.FrameworkModel + :members: + :undoc-members: + :show-inheritance: + +.. autoclass:: sagemaker.model.ModelPackage + :members: + :undoc-members: + :show-inheritance: diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py index 3d518aa299..8e60d53ddb 100644 --- a/src/sagemaker/model.py +++ b/src/sagemaker/model.py @@ -375,7 +375,7 @@ def __init__(self, model_data, image, role, entry_point, source_dir=None, predic entry_point (str): Path (absolute or relative) to the Python source file which should be executed as the entry point to model hosting. This should be compatible with either Python 2.7 or Python 3.5. source_dir (str): Path (absolute or relative) to a directory with any other training - source code dependencies aside from tne entry point file (default: None). Structure within this + source code dependencies aside from the entry point file (default: None). Structure within this directory will be preserved when training on SageMaker. If the directory points to S3, no code will be uploaded and the S3 location will be used instead. dependencies (list[str]): A list of paths to directories (absolute or relative) with