File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change 2
2
CHANGELOG
3
3
=========
4
4
5
+ 1.16.2
6
+ ======
7
+ * feature: Add support for AugmentedManifestFile and ShuffleConfig
8
+
5
9
1.16.1.post1
6
10
============
7
11
Original file line number Diff line number Diff line change @@ -282,13 +282,13 @@ def test_augmented_manifest(sagemaker_session):
282
282
train_instance_count = INSTANCE_COUNT , train_instance_type = INSTANCE_TYPE ,
283
283
enable_cloudwatch_metrics = True )
284
284
fw .fit (inputs = s3_input ('s3://mybucket/train_manifest' , s3_data_type = 'AugmentedManifestFile' ,
285
- attribute_names = [" foo" , " bar" ]))
285
+ attribute_names = [' foo' , ' bar' ]))
286
286
287
287
_ , _ , train_kwargs = sagemaker_session .train .mock_calls [0 ]
288
288
s3_data_source = train_kwargs ['input_config' ][0 ]['DataSource' ]['S3DataSource' ]
289
289
assert s3_data_source ['S3Uri' ] == 's3://mybucket/train_manifest'
290
290
assert s3_data_source ['S3DataType' ] == 'AugmentedManifestFile'
291
- assert s3_data_source ['AttributeNames' ] == [" foo" , " bar" ]
291
+ assert s3_data_source ['AttributeNames' ] == [' foo' , ' bar' ]
292
292
293
293
294
294
def test_shuffle_config (sagemaker_session ):
You can’t perform that action at this time.
0 commit comments