Skip to content

Commit da271ef

Browse files
ChoiByungWookjesterhazy
authored andcommitted
fix quotes (aws#562)
1 parent 04f8abf commit da271ef

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

CHANGELOG.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ CHANGELOG
88
* bug-fix: Append retry id to default Airflow job name to avoid name collisions in retry
99
* bug-fix: Local Mode: No longer requires s3 permissions to run local entry point file
1010
* bug-fix: Local Mode: Move dependency on sagemaker_s3_output from rl.estimator to model
11+
* doc-fix: Fix quotes in estimator.py and model.py
1112

1213
1.16.2
1314
======

src/sagemaker/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,7 @@ def compile_model(self, target_instance_family, input_shape, output_path, framew
246246
example: ml_c5. Allowed strings are: ml_c5, ml_m5, ml_c4, ml_m4, jetsontx1, jetsontx2, ml_p2, ml_p3,
247247
deeplens, rasp3b
248248
input_shape (dict): Specifies the name and shape of the expected inputs for your trained model in json
249-
dictionary form, for example: {data:[1,3,1024,1024]}, or {var1: [1,1,28,28], var2:[1,1,28,28]}
249+
dictionary form, for example: {'data':[1,3,1024,1024]}, or {'var1': [1,1,28,28], 'var2':[1,1,28,28]}
250250
output_path (str): Specifies where to store the compiled model
251251
framework (str): The framework that is used to train the original model. Allowed values: 'mxnet',
252252
'tensorflow', 'pytorch', 'onnx', 'xgboost'

src/sagemaker/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ def compile(self, target_instance_family, input_shape, output_path, role,
159159
example: ml_c5. Allowed strings are: ml_c5, ml_m5, ml_c4, ml_m4, jetsontx1, jetsontx2, ml_p2, ml_p3,
160160
deeplens, rasp3b
161161
input_shape (dict): Specifies the name and shape of the expected inputs for your trained model in json
162-
dictionary form, for example: {data:[1,3,1024,1024]}, or {var1: [1,1,28,28], var2:[1,1,28,28]}
162+
dictionary form, for example: {'data':[1,3,1024,1024]}, or {'var1': [1,1,28,28], 'var2':[1,1,28,28]}
163163
output_path (str): Specifies where to store the compiled model
164164
role (str): Execution role
165165
tags (list[dict]): List of tags for labeling a compilation job. For more, see

0 commit comments

Comments
 (0)