Skip to content

Commit 8adbbb7

Browse files
committed
Update changelog and fix minor nit in test
1 parent f817923 commit 8adbbb7

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

CHANGELOG.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,10 @@
22
CHANGELOG
33
=========
44

5+
1.16.2
6+
======
7+
* feature: Add support for AugmentedManifestFile and ShuffleConfig
8+
59
1.16.1.post1
610
============
711

tests/unit/test_estimator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -282,13 +282,13 @@ def test_augmented_manifest(sagemaker_session):
282282
train_instance_count=INSTANCE_COUNT, train_instance_type=INSTANCE_TYPE,
283283
enable_cloudwatch_metrics=True)
284284
fw.fit(inputs=s3_input('s3://mybucket/train_manifest', s3_data_type='AugmentedManifestFile',
285-
attribute_names=["foo", "bar"]))
285+
attribute_names=['foo', 'bar']))
286286

287287
_, _, train_kwargs = sagemaker_session.train.mock_calls[0]
288288
s3_data_source = train_kwargs['input_config'][0]['DataSource']['S3DataSource']
289289
assert s3_data_source['S3Uri'] == 's3://mybucket/train_manifest'
290290
assert s3_data_source['S3DataType'] == 'AugmentedManifestFile'
291-
assert s3_data_source['AttributeNames'] == ["foo", "bar"]
291+
assert s3_data_source['AttributeNames'] == ['foo', 'bar']
292292

293293

294294
def test_shuffle_config(sagemaker_session):

0 commit comments

Comments
 (0)