diff --git a/CHANGELOG.rst b/CHANGELOG.rst index ed5a20d4fc..ff35358a77 100644 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -8,6 +8,7 @@ CHANGELOG * bug-fix: Append retry id to default Airflow job name to avoid name collisions in retry * bug-fix: Local Mode: No longer requires s3 permissions to run local entry point file * bug-fix: Local Mode: Move dependency on sagemaker_s3_output from rl.estimator to model +* doc-fix: Fix quotes in estimator.py and model.py 1.16.2 ====== diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py index 4a6d035edc..53b8bab914 100644 --- a/src/sagemaker/estimator.py +++ b/src/sagemaker/estimator.py @@ -246,7 +246,7 @@ def compile_model(self, target_instance_family, input_shape, output_path, framew example: ml_c5. Allowed strings are: ml_c5, ml_m5, ml_c4, ml_m4, jetsontx1, jetsontx2, ml_p2, ml_p3, deeplens, rasp3b input_shape (dict): Specifies the name and shape of the expected inputs for your trained model in json - dictionary form, for example: {‘data’:[1,3,1024,1024]}, or {‘var1’: [1,1,28,28], ‘var2’:[1,1,28,28]} + dictionary form, for example: {'data':[1,3,1024,1024]}, or {'var1': [1,1,28,28], 'var2':[1,1,28,28]} output_path (str): Specifies where to store the compiled model framework (str): The framework that is used to train the original model. Allowed values: 'mxnet', 'tensorflow', 'pytorch', 'onnx', 'xgboost' diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py index 9572b834c0..72e260aa6e 100644 --- a/src/sagemaker/model.py +++ b/src/sagemaker/model.py @@ -159,7 +159,7 @@ def compile(self, target_instance_family, input_shape, output_path, role, example: ml_c5. Allowed strings are: ml_c5, ml_m5, ml_c4, ml_m4, jetsontx1, jetsontx2, ml_p2, ml_p3, deeplens, rasp3b input_shape (dict): Specifies the name and shape of the expected inputs for your trained model in json - dictionary form, for example: {‘data’:[1,3,1024,1024]}, or {‘var1’: [1,1,28,28], ‘var2’:[1,1,28,28]} + dictionary form, for example: {'data':[1,3,1024,1024]}, or {'var1': [1,1,28,28], 'var2':[1,1,28,28]} output_path (str): Specifies where to store the compiled model role (str): Execution role tags (list[dict]): List of tags for labeling a compilation job. For more, see