Skip to content

Commit 01896bd

Browse files
chuyang-dengChuyang Denglaurenyu
authored
fix: address flake8 error (aws#1493)
Co-authored-by: Chuyang Deng <[email protected]> Co-authored-by: Lauren Yu <[email protected]>
1 parent af9c913 commit 01896bd

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

tests/unit/test_amazon_estimator.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -228,16 +228,16 @@ def test_fit_ndarray(time, sagemaker_session):
228228
labels = [99, 85, 87, 2]
229229
pca.fit(pca.record_set(np.array(train), np.array(labels)))
230230
mock_s3.Object.assert_any_call(
231-
BUCKET_NAME, "key-prefix/PCA-2017-11-06-14:14:15.671/matrix_0.pbr".format(TIMESTAMP)
231+
BUCKET_NAME, "key-prefix/PCA-2017-11-06-14:14:15.671/matrix_0.pbr"
232232
)
233233
mock_s3.Object.assert_any_call(
234-
BUCKET_NAME, "key-prefix/PCA-2017-11-06-14:14:15.671/matrix_1.pbr".format(TIMESTAMP)
234+
BUCKET_NAME, "key-prefix/PCA-2017-11-06-14:14:15.671/matrix_1.pbr"
235235
)
236236
mock_s3.Object.assert_any_call(
237-
BUCKET_NAME, "key-prefix/PCA-2017-11-06-14:14:15.671/matrix_2.pbr".format(TIMESTAMP)
237+
BUCKET_NAME, "key-prefix/PCA-2017-11-06-14:14:15.671/matrix_2.pbr"
238238
)
239239
mock_s3.Object.assert_any_call(
240-
BUCKET_NAME, "key-prefix/PCA-2017-11-06-14:14:15.671/.amazon.manifest".format(TIMESTAMP)
240+
BUCKET_NAME, "key-prefix/PCA-2017-11-06-14:14:15.671/.amazon.manifest"
241241
)
242242

243243
assert mock_object.put.call_count == 4

tests/unit/test_fw_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -735,7 +735,7 @@ def test_invalid_framework_accelerator():
735735

736736

737737
def test_invalid_framework_accelerator_with_neo():
738-
error_message = "Neo does not support Amazon Elastic Inference.".format(MOCK_FRAMEWORK)
738+
error_message = "Neo does not support Amazon Elastic Inference."
739739
# accelerator was chosen for unsupported framework
740740
with pytest.raises(ValueError) as error:
741741
fw_utils.create_image_uri(

0 commit comments

Comments
 (0)